L2-026. 小字辈

题给定一个庞大家族的家谱，要请你给出最小一辈的名单。

输入格式：

输入在第一行给出家族人口总数 N（不超过 100 000 的正整数） —— 简单起见，我们把家族成员从 1 到 N 编号。随后第二行给出 N 个编号，其中第 i 个编号对应第 i 位成员的父/母。家谱中辈分最高的老祖宗对应的父/母编号为 -1。一行中的数字间以空格分隔。

输出格式：

首先输出最小的辈分（老祖宗的辈分为 1，以下逐级递增）。然后在第二行按递增顺序输出辈分最小的成员的编号。编号间以一个空格分隔，行首尾不得有多余空格。

输入样例：

9
2 6 5 5 -1 5 6 4 7

输出样例：

4
1 9

一般方法

#include <bits/stdc++.h> 
using namespace std;  
int len=0;
int mmax=1;  
int dp[100010];  
vector<int>we[100005];  
void dfs(int x, int k){  
    if(k>mmax && we[x].size()== 0){  
        len=0;  
        dp[len]=x;  
        mmax=k;  
        len+=1;
        return;  
    }  
    if(k==mmax && we[x].size()==0){  
        dp[len]=x;  
        len+=1;
        return;  
    }  
    for(int i=0;i<we[x].size();i++){  
        dfs(we[x][i],k+1);  
    }  
}  
int main(){ 
	int n; 
	int k;
    scanf("%d",&n);  
    for(int i=1;i<=n;i++){  
        int a;  
        scanf("%d",&a);  
        we[a].push_back(i);  
        if(a==-1) 
		   k=i;  
    }  
    dfs(k, 1);  
    printf("%d\n",mmax);  
    int f=0; 
    for(int i=0;i<len;i++){ 
		if(f==0){
			printf("%d",dp[i]);
			f=1;
		}
		else 
        	printf(" %d",dp[i]);  
    }  
    return 0;  
}  

最近看了数位dp就试了下,过了，开心0.0

#include <bits/stdc++.h>
using namespace std;
int a[100010];
int dp[100010];
int dfs(int x){
	if(a[x]==0)
		return 0;
	if(dp[x]!=0)
		return dp[x];
	dp[a[x]]=dfs(a[x]);
	return 1+dp[a[x]];
}
int main(){
  int n;
  scanf("%d",&n);
  for(int i=1;i<=n;i++)
  	scanf("%d",&a[i]);
  int max=0;
  for(int i=1;i<=n;i++){
	dp[i]=dfs(i);
	if(dp[i]>max)
		max=dp[i];
  }
  printf("%d\n",max);
  int f=0;
  for(int i=1;i<=n;i++){
  	if(f==0 && dp[i]==max){
  		printf("%d",i);
  		f=1;
	  }
	else if(dp[i]==max && f==1)
	printf(" %d",i);
  } 
  printf("\n");
  return 0;
}