uva101 The Blocks Problem

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#define LOCAL

#define MAXN 30

typedef struct node

{

    int data;

    struct node  *pre, *next;

}*Node;

Node link[MAXN] ;

int n;

void init();

Node search(int data, int *pos);

void freespace();

int main()

{

    char str1[5], str2[5];

    int src, dst;

    Node a, b, p, q;

    int flag;

    int i;

    int pos1, pos2;

    #ifdef LOCAL

        freopen("c://uva_in.txt", "r", stdin);

    #endif

    scanf("%d", &n);

    init();

    while (scanf("%s", str1) && (strcmp(str1, "quit") != 0))

    {

        scanf("%d%s%d", &src, str2, &dst);

        a = search(src, &pos1);

        b = search(dst, &pos2);

        if (strcmp(str1, "move") == 0 && strcmp(str2, "onto") == 0)

            flag = 1;

        else if (strcmp(str1, "move") == 0 && strcmp(str2, "over") == 0)

            flag = 2;

        else if (strcmp(str1, "pile") == 0 && strcmp(str2, "onto") == 0)

            flag = 3;

        else if (strcmp(str1, "pile") == 0 && strcmp(str2, "over") == 0)

            flag = 4;

        if (flag == 1 && pos1 != pos2)

        {

            p = a->next;

            a->next = NULL;

            while (p)

            {

                q = p->next;

                p->next = NULL;

                p->pre = link[p->data ];

                link[p->data ]->next = p;

                p = q;

            }

            p = b->next;

            b->next = NULL;

            while (p)

            {

                q = p->next;

                p->next = NULL;

                p->pre = link[p->data];

                link[p->data]->next = p;

                p = q;

            }

            a->pre->next = NULL;

            a->pre = b;

            b->next = a;

        } else if (flag == 2 && pos1 != pos2)

        {

            p = a->next;

            a->next = NULL;

            while (p)

            {

                q = p->next;

                p->next = NULL;

                p->pre = link[p->data];

                link[p->data]->next = p;

                p = q;

            }

            while (b->next)

                b = b->next;

            a->pre->next = NULL;

            a->pre = b;

            b->next = a;

        } else if (flag == 3 && pos1 != pos2)

        {

            p = b->next;

            b->next = NULL;

            while (p)

            {

                q = p->next;

                p->next = NULL;

                p->pre = link[p->data];

                link[p->data]->next = p;

                p = q;

            }

            a->pre->next = NULL;

            a->pre = b;

            b->next = a;

        } else if (flag == 4 && pos1 != pos2)

        {

            while (b->next)

                b = b->next;

            a->pre->next = NULL;

            a->pre = b;

            b->next = a;

        }

    }

    for (i = 0; i < n; i++)

    {

        p = link[i]->next;

        printf("%d:", i);

        while (p)

        {

            printf(" %d", p->data);

            p = p->next;

        }

        printf("/n");

    }

    return 0;

}

void init()

{

    int i;

    Node p;

    for (i = 0; i < n; i++)

    {

        p = (Node)malloc(sizeof(struct node));

        p->data = i;

        p->next = NULL;

        link[i] = (Node)malloc(sizeof(struct node));

        link[i]->next = p;

        p->pre = link[i];

    }

}

Node search(int data, int *pos)

{

    int i;

    Node p;

    for (i = 0; i < n; i++)

    {

        p = link[i]->next;

        while(p)

        {

            if (p->data == data)

            {

                *pos = i;

                return p;

            }

            p = p->next;

        }

    }

    return NULL;

}

void freespace()

{

    int i;

    Node p, q;

    for (i = 0; i < n; i++)

    {

        p = link[i]->next;

        while(p)

        {

            q = p->next;

            free(p);

        }

        free(link[i]);

    }

}