方法一:深搜
对于节点 i,在其子节点中找出:令该节点作为根节点时可以使高度最大的节点 j
令节点 j 作为 i 的子节点中最后一个出现的节点,可使高度达到最大。
#include <bits/stdc++.h> using namespace std; const int MAXN = 100005; int n, tmp; vector<int> mp[MAXN]; int dfs(int idx) { int sz = mp[idx].size(); int ans = 0; for (int i = 0; i < sz; i++) { ans = max(ans, dfs(mp[idx][i])); } return ans + sz; } int main() { cin >> n; for (int i = 2; i <= n; i++) { cin >> tmp; mp[tmp].push_back(i); } cout << dfs(1) << endl; return 0; }
方法二:动态规划
用数组 s[i] 表示节点 i 的子节点个数,用数组 f[i] 表示节点 i 的父节点,dp[i] 表示当节点 i 作为根节点时的最大高度
可以推断出如下状态转移方程:
dp[f[i]] = max(dp[f[i]], s[f[i]] + dp[i]);
由于子节点一定比父节点的编号大,所以对节点编号逆向遍历即可。
#include <bits/stdc++.h> using namespace std; const int MAXN = 100005; int n, tmp; int s[MAXN], f[MAXN], dp[MAXN]; int main() { cin >> n; for (int i = 2; i <= n; i++) { cin >> tmp; f[i] = tmp; s[tmp]++; } for (int i = n; i > 1; i--) { int fa = f[i]; dp[fa] = max(dp[fa], s[fa] + dp[i]); } cout << dp[1] << endl; return 0; }
0.0分
13 人评分
C语言网提供由在职研发工程师或ACM蓝桥杯竞赛优秀选手录制的视频教程,并配有习题和答疑,点击了解:
一点编程也不会写的:零基础C语言学练课程
解决困扰你多年的C语言疑难杂症特性的C语言进阶课程
从零到写出一个爬虫的Python编程课程
只会语法写不出代码?手把手带你写100个编程真题的编程百练课程
信息学奥赛或C++选手的 必学C++课程
蓝桥杯ACM、信息学奥赛的必学课程:算法竞赛课入门课程
手把手讲解近五年真题的蓝桥杯辅导课程
发表评论 取消回复