任务:对于给定的二叉树,计算二叉树各结点自己为根时子树的结点数,并按前序输出
输入:第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;
}