PAT (Advanced Level) Practice 1004 Counting Leaves(C语言）

不是很难的题目，recursive即可完成，从树的node开始向下递归，如果遇到leaf则那个leaf所在level+1，如果没遇到leaf则继续向下递归，判断是否为leaf则通过遍历整个数组实现。

#include<stdio.h>

int ret[100]={0};
int depth=0;

int dfs(int a[100][100], int num[100], int nonleafnodes, int nodes, int level);
int main(){

	int numofnodes,nonleafnodes=0;
	scanf("%d %d",&numofnodes,&nonleafnodes);
	
	int a[100][100]={0};
	int num[100]={0};
	int i,j=0;
	
	if(numofnodes!=0){
		if(nonleafnodes==0){
			printf("1");
		}else{
			for(i=0; i<nonleafnodes; i++){
				scanf("%02d %d",&a[i][0],&num[i]);
				for(j=0; j<num[i]; j++){
					scanf("%02d",&a[i][j+1]);
				}
			}
			dfs(a,num,nonleafnodes,1,0);
			printf("0");
			for(i=1; i<=depth; i++){
				printf(" %d",ret[i]);
			}
		}
		
	}
	return 0;
}

int dfs(int a[100][100], int num[100], int nonleafnodes, int nodes, int level){
	if(level>depth){
		depth=level;
	}
	for(int i=0; i<nonleafnodes; i++){
		if(nodes==a[i][0]){
			for(int j=0; j<num[i]; j++){
				dfs(a,num,nonleafnodes,a[i][j+1],level+1);
			}return 0;				//易错点，若不退出会重复加 
		}
	}
	ret[level]++;
}

最开始做这题的时候没有考虑到输入的值应该是无序的，把输入当成有序在做，直到有两组数据过不了judge才发现这个问题，当输入是无序的时候，就存在父级比son后输入的情况，导致出错无法搜索到父级。然后开始用递归（第一次写递归的代码，之前写的递归都是pseudo code),在这里把之前的代码也记录一下吧。

#include<stdio.h>

int main(){

	int numofnodes,nonleafnodes=0;
	scanf("%d %d",&numofnodes,&nonleafnodes);
	
	int level[100]={0};
	int a[100][100]={0};
	int i,j=0;
	int ID,k=0;
	int p,pp=0;
	
	level[0]=1;
	a[0][0]=1;

	if(numofnodes!=0){
		if(nonleafnodes==0){
			printf("1");
		}else{
			while(nonleafnodes!=0){
				scanf("%02d %d",&ID,&k);
				for(i=0; i<100; i++){
					for(j=0; j<100; j++){
						if(a[i][j]==0){
							break;
						}else if(a[i][j]==ID){
							p=i;
							break;
						}
					}if(a[i][j]==ID){
						break;
					}
				}
				level[p]--;
				level[p+1]+=k;
				while(k!=0){
					scanf("%02d",&ID);
					for(j=0; j<100; j++){
						if(a[p+1][j]==0){
							pp=j;
							break;
						}
					}
					a[p+1][pp]=ID;
					k--;
				}
				nonleafnodes--;
			}
			
			printf("0");
			for(i=1; i<=(p+1); i++){
				printf(" %d",level[i]);
			} 
		}
	}
	return 0;
}