0
点赞
收藏
分享

微信扫一扫

试题H 左Children右兄弟

(吐槽:为什么不能使用‘孩’和‘子’这两个字)

题目链接:https://www.lanqiao.cn/courses/2786/learning/?id=280825

题目解析:典型的树形DP

记dp[i]是以i结点为根的树的最大高度,则状态转移方程:

dp[cur] = max(dp[cur],e[cur].size() + dp[next]);

dp[cur] = 1 (当该结点没有子结点时)

其中e[cur]记录了cur结点的所有子节点,next为其中的一个结点

又1结点高度为0

故答案为dp[1]-1

代码:

//#define local
#include<cstdio>
#include<iostream>
#include<vector>
#define MAXN 100005
typedef long long ll;
using namespace std;
int max(const int &a,const int &b){
	return a>b?a:b;
}
int n;
vector<int>e[MAXN];
int dp[MAXN]={0};
void dfs(int cur){
	if(!e[cur].size()){
		dp[cur]=1;
		return;
	}
	else{
		for(int i=0;i<e[cur].size();++i){
			int next=e[cur][i];
			dfs(next);
			dp[cur]=max(dp[cur],e[cur].size()+dp[next]);
		}
	}
}
int main(){
#ifdef local
freopen("data.in","rb",stdin);
//freopen("data.out","wb",stdout);
#endif
scanf("%d",&n);
for(int i=2;i<=n;++i){
	int fa;
	scanf("%d",&fa);
	e[fa].push_back(i);
}
dfs(1);
printf("%d",dp[1]-1);
return 0;
} 
举报

相关推荐

0 条评论