九度笔记之 水平打印二叉树

题目

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

时间限制：1 秒

内存限制：128 兆

特殊判题：否

提交：18

解决：3

题目描述：

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

输入：

输入可能包含多个测试样例，输入以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

分析

可以用类似广度优先搜索的方法

不同的是这里用vector保存遍历到的节点编号

在搜索到某个节点后，将该节点的子节点push_back到vector

代码

#include <iostream>
#include <vector>
#include <string>
struct Node{
    int value;
    int lid;
    int rid;
};
std::vector<Node> vn;
std::vector<int> v; //保存访问的节点编号
void Bsearch(){
    v.push_back(1);
    std::vector<int>::size_type vid = 0;;
    while(vid!=v.size()){
        int nodeId = v.at(vid)-1;//得到节点编号-1
        std::cout<<vn.at(nodeId).value;
        if(vn.at(nodeId).lid){      
            v.push_back(vn.at(nodeId).lid);
        }
        if(vn.at(nodeId).rid){  
            v.push_back(vn.at(nodeId).rid);
        }
        if(vid<v.size()-1)
            std::cout<<" ";
         else
            std::cout<<std::endl;
        vid++;
    }
}
 
void judo(int n){
    if(n==0){
        std::cout<<"NULL"<<std::endl;
        return;
    }
    Node nd;
    nd.lid = 0;
    nd.rid = 0;
    for(int i = 0;i<n;i++){
        std::cin>>nd.value;
        vn.push_back(nd);
    }
 
    std::string ci;
    for(int i = 0;i<n;i++){
        std::cin>>ci;
        switch (ci[0]){
        case 'd':
            std::cin>>vn.at(i).lid>>vn.at(i).rid;
            break;
        case 'r':
            std::cin>>vn.at(i).rid;
            break;
        case 'l':
            std::cin>>vn.at(i).lid;
            break;
        case 'z':
            break;
        }
    }
    Bsearch();
 
}
 
int main() {
    int n;
    while(std::cin>>n){
        judo(n);
    }
    return 0;
}