pat 1004 counting Leaves

之前一直WA，转换了个思路（用一个parent[]保存父结点，而不是老想用一个数组保存子结点），就1Y了。输入时进行处理，用一个parent[]保存父结点，标记出所有的非叶结点，再对每个叶结点计算它们的层数，相应的层数的叶结点数++即可。

AC代码：

#include<iostream>
using namespace std;
struct node
{
	node *left,*right;
	int value;
};
const int NUM=105;
int parent[NUM],level[NUM],isLeaf[NUM];
int maxLevel=1;
void findLevel(int x)
{
	int r=x;
	int temp=1;//temp是叶结点的层数
	while(parent[r]!=r){
		temp++;
		r=parent[r];
	}
	level[temp]++;//所在层数的叶结点++
	if(temp>=maxLevel)//求出最大层数
		maxLevel=temp;
}
int main()
{
	int n,m,i,id,k,j;
	cin>>n>>m;
	for(i=0;i<NUM;i++){
		parent[i]=i;
		isLeaf[i]=1;
	}
	for(i=0;i<m;i++){
		cin>>id>>k;
		while(k--){
			cin>>j;
			parent[j]=id;
		}
		isLeaf[id]=0;//将id标记为非叶结点

	}
	for(i=1;i<=n;i++){
		if(isLeaf[i]==1){//如果是叶结点，再进行层数计算
			findLevel(i);
		}

	}
	cout<<level[1];
	for(i=2;i<=maxLevel;i++){
		cout<<" "<<level[i];
	}
	return 0;


}