面试题23 从上往下打印二叉树

地址：http://ac.jobdu.com/problem.php?pid=1523

题目1523：从上往下打印二叉树

题目描述：

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

输入：

输入可能包含多个测试样例，输入以EOF结束。
对于每个测试案例，输入的第一行一个整数n(1<=n<=1000, ：n代表将要输入的二叉树元素的个数（节点从1开始编号）。接下来一行有n个数字,代表第i个二叉树节点的元素的值。接下来有n行，每行有一个字母Ci。
Ci=’d’表示第i个节点有两子孩子，紧接着是左孩子编号和右孩子编号。
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

解题思路：

1、层次遍历二叉树

2、使用队列作为辅助数据结构

<span style="font-size:14px;">#include <stdio.h>
#include <stdlib.h>
#include <queue>
using namespace std;
//第一列：节点元素的值，第二列：节点左孩子序号，第三列右孩子序号
//若无左（右）孩子，则用-1标记
//可容纳1000个节点 
int buf[1000][3];
 
struct treeNode {
    int data;
    treeNode *left;
    treeNode *right;
};
//创建二叉树
//参数i：节点元素的序号，从0开始 
treeNode* createTree(int i) {
    treeNode *root = (treeNode *)malloc(sizeof(treeNode));
    root->data = buf[i][0];
    if (buf[i][1] != -1)
        root->left = createTree(buf[i][1] - 1);
    else
        root->left = NULL;
    if (buf[i][2] != -1)
        root->right = createTree(buf[i][2] - 1);
    else
        root->right = NULL;
    return root;
}
//层次遍历二叉树 
void levelTraverse(treeNode *root) {
    if(root == NULL)
        return;
    queue<treeNode *> Q;
    Q.push(root);
    while(!Q.empty()) {
        treeNode *tmp = Q.front();
        Q.pop();
        if(tmp->left != NULL)
            Q.push(tmp->left);
        if(tmp->right != NULL)
            Q.push(tmp->right);
        if(tmp == root)
            printf("%d", tmp->data);
        else
            printf(" %d", tmp->data);
    }
}
 
int main(){
    int n;
    while(scanf("%d", &n) != EOF) {
        for(int i = 0; i < n; i ++) {
            scanf("%d", &buf[i][0]);
        }
        char c;
        for(int i = 0; i < n; i ++) {
            while((c = getchar()) != '\n');
            c = getchar();
            if(c == 'd') {
                scanf("%d", &buf[i][1]);
                scanf("%d", &buf[i][2]);
            }
            else if(c == 'l') {
                scanf("%d", &buf[i][1]);
                buf[i][2] = -1;
            }
            else if(c == 'r'){
                buf[i][1] = -1;
                scanf("%d", &buf[i][2]);
            }
            else { //c == 'z'
                buf[i][1] = -1;
                buf[i][2] = -1;
            }
        }
        treeNode *root = createTree(0);
        levelTraverse(root);
        printf("\n");
    }
    return 0;
}
/**************************************************************
    Problem: 1523
    Language: C++
    Result: Accepted
    Time:0 ms
    Memory:1064 kb
****************************************************************/</span>