首页 > ACM题库 > 九度OJ > 九度-1523-从上往下打印二叉树[解题代码]
2013
12-13

九度-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

cpp 代码如下:
#include <stdio.h>
#include <queue>
#include <string.h>
using namespace std;
struct Tree{
	int data;
	Tree * l,* r;
}nodes[1001];
int n, a,b;
char str[3];
queue<Tree *> q;
int main(){
	while(scanf("%d",&n) != EOF){
		memset(nodes, 0, sizeof(Tree) * (n));
		for(int i=0; i<n; i++)
			scanf("%d", &nodes[i].data);
		for(int i=0; i<n; i++){
			scanf("%s", str);
			if(str[0] == 'd'){
				scanf("%d %d", &a, &b);
				nodes[i].l = &nodes[a-1];
				nodes[i].r = &nodes[b-1];
			}else if(str[0] == 'l'){
scanf("%d", &a);nodes[i].l = &nodes[a-1];
			}else if(str[0] == 'r'){
	scanf("%d", &b);nodes[i].r = &nodes[b-1];}
		}
		printf("%d",nodes[0].data);
		if(nodes[0].l) q.push(nodes[0].l);
		if(nodes[0].r) q.push(nodes[0].r);
		while(q.size() > 0){
			Tree * tmp = q.front();
			q.pop();
			printf(" %d",tmp->data);
			if(tmp->l) q.push(tmp->l);
			if(tmp->r) q.push(tmp->r);
		}
	}
	return 0;
}
/**************************************************************
	Problem: 1523
	User: coder
	Language: C++
	Result: Accepted
	Time:0 ms
	Memory:1076 kb
****************************************************************/


  1. 老实说,这种方法就是穷举,复杂度是2^n,之所以能够AC是应为题目的测试数据有问题,要么数据量很小,要么能够得到k == t,否则即使n = 30,也要很久才能得出结果,本人亲测