ツリーを上から下に階層化して印刷

2639 ワード

タイトルの説明:
ツリーの各ノードを上から下に印刷し、同レイヤノードを左から右に印刷します.
入力:
入力には複数のテストサンプルが含まれ、入力はEOFで終了します.各テストケースについて、入力する最初の行の整数n(1<=n<=1000、:nは入力するツリー要素の個数(ノード1から番号付け)を表します.次の行には、i番目のツリーノードの要素の値を表すn個の数字があります.次はn行で、各行に1文字Ciがあります.Ci=’d’は,i番目のノードに2人の子があり,次いで左の子番号と右の子番号であることを示す.Ci=’l’は、i番目のノードに左の子があり、左の子の番号が続くことを示す.Ci=’r’は、i番目のノードに右の子があり、右の子の番号が続くことを示す.Ci=’z’はi番目のノードに子供がいないことを示す.
出力:
各テストケースに対応して、ツリーノードの値を上下から左から右に印刷します.
サンプル入力:
7
8 6 5 7 10 9 11
d 2 5
d 3 4
z
z
d 6 7
z
z

サンプル出力:
8 6 10 5 7 9 11

ACコード(キューで実現):
#include<stdio.h>
#include<string.h>
struct Tree {
    int data;
    struct Tree *left;
    struct Tree *right;
} nodes[1001];

int is_empty();
void enqueue();
struct Tree *dequeue();

struct Tree *queue[1001];
int head = 0, tail = 0, n, l, r;
char str[3];

int main() {
    while (scanf("%d", &n) != EOF) {
        memset(nodes, 0, sizeof(struct Tree) * n);
        int i;
        for (i = 0; i < n; i++) {
            scanf("%d", &nodes[i].data);
        }
        for (i = 0; i < n; i++) {
            scanf("%s", str);
            if (str[0] == 'd') {
                scanf("%d %d", &l, &r);
                nodes[i].left = &nodes[l - 1];
                nodes[i].right = &nodes[r - 1];
            } else if (str[0] == 'l') {
                scanf("%d", &l);
                nodes[i].left = &nodes[l - 1];
            } else if (str[0] == 'r') {
                scanf("%d", &r);
                nodes[i].right = &nodes[r - 1];
            }
        }

        printf("%d", nodes[0].data);
        if(nodes[0].left) enqueue(nodes[0].left);
        if(nodes[0].right) enqueue(nodes[0].right);

        //enqueue(&nodes[0]);  , Presentation Error( )

        while (!is_empty()) {
            struct Tree *tmp = dequeue();
            printf(" %d", tmp->data);
            if (tmp->left) {
                enqueue(tmp->left);
            }
            if (tmp->right) {
                enqueue(tmp->right);
            }
        }
    }

    return 0;
}

int is_empty() {
    return head == tail;
}

void enqueue(struct Tree *node) {
    queue[tail++] = node;
}

struct Tree *dequeue() {
    return queue[head++];
}

/**************************************************************
    Problem: 1523
    User: wusuopuBUPT
    Language: C
    Result: Accepted
    Time:0 ms
    Memory:944 kb
****************************************************************/