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

题目

题目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;
}