子树问题

任务：对于给定的二叉树，计算二叉树各结点自己为根时子树的结点数，并按前序输出

输入：第1行有1正整数n，表示给定的二叉树有n个顶点，编号为1,2,3,...n。接下来n行中，每行有3个整数a,b,c，分别表示编号为a的结点的左儿子编号为b，右儿子编号为c，为0表示没有儿子。各结点信息按照层序列表的顺序给出

输出：将计算出的以二叉树各结点为根的子树的结点数按照前序列表输出

 

输入样例：

5

1 4 2

4 3 0

2 5 0

3 0 0

5 0 0

输出样例：

5 2 1 2 1

 

样例与层序列表问题一致。这个题的思想与计算树高的算法比较相似，只需将叶节点的data为1，非叶节点的data设为 左子树data+右子树data+1 即可

#include <iostream>
using namespace std;

typedef struct{
	int data;
	int left, right;
}TreeNode;

void Get_NodeNum(TreeNode& root, TreeNode *node)
{	
	int lnum = 0, rnum = 0;	
	if( root.left != 0 ){
		Get_NodeNum(node[root.left], node);
		lnum = node[root.left].data;
	}
	if( root.right!= 0 ){
		Get_NodeNum(node[root.right],node);
		rnum = node[root.right].data;
	}
	root.data = lnum + rnum + 1;			
} 

void PreOrder(TreeNode& root, TreeNode *node)
{
	cout << root.data << " ";	//visit()
	if( root.left != 0 )
		PreOrder(node[root.left], node);
	if( root.right!= 0 )
		PreOrder(node[root.right],node);		
}

int main()
{	
	int i, n;
	while( cin >> n )
	{
		int a, b, c;
		TreeNode node[n+1];
		for(i = 1; i <= n; ++i){
			cin >> a >> b >> c;
			node[a].data  = 0;
			node[a].left  = b;
			node[a].right = c;
		}
		
		Get_NodeNum(node[1], node);
		PreOrder(node[1], node);		 
		cout << "-----END-----" << endl;	
	}
	
	return 0;
}