2020CCPC(秦皇岛) - Kingdom‘s Power(树形dp+贪心)

题目大意：给出一棵 n 个节点的有根树，点 1 为根节点，现在在根节点有无穷多个士兵，每一秒可以控制任意一个士兵向任意一个单位移动一步，士兵移动到的点会被永久占领，现在问最少需要经过多少秒，才能将所有的点都占领

题目分析：首先不难看出的两个小结论是：

1. 如果所有的叶子节点都被占领，那么沿途的所有点也会被占领，所以问题转换为了到每个叶子节点的最短时间
2. 第一步肯定是需要从根节点到其中一个叶子节点的，贪心去想，到深度最低的那个叶子节点一定是最优的

树形dp维护一下从根节点过来更优还是从相邻的叶子结点更优，然后更新答案即可，这里参考了网上的解法，形式参数维护一个变量表示到达当前节点时的最短距离，递归的返回值是自底向上的最短距离，也就是从相邻的叶子节点过来的距离，这个距离和深度取个最小值就是答案了

最后还有一点需要贪心去考虑，感觉这个就是本题的难点了，假设只有一个士兵，从根节点出发若想遍历整棵树的话，最短的路线一定是：对于每个子树而言，最长的链只遍历一次，其余的链都会被遍历两次

基于此，对于每个子树而言，我们只需要将其按照 最长链 进行排序即可，因为最后去遍历的话一定只需要走一遍

代码：

//#pragma GCC optimize(2)
//#pragma GCC optimize("Ofast","inline","-ffast-math")
//#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
#include<bitset>
#include<unordered_map>
using namespace std;
     
typedef long long LL;
     
typedef unsigned long long ull;
     
const int inf=0x3f3f3f3f;

const int N=1e6+100;

vector<pair<int,int>>node[N];

int deep[N],val[N];

int dfs1(int u,int dep)
{
	if(node[u].empty())
		return 1;
	deep[u]=dep;
	for(auto &it:node[u])
	{
		int v=it.second;
		it.first=max(it.first,dfs1(v,dep+1));
	}
	sort(node[u].begin(),node[u].end());
	return node[u].back().first+1;
}

int dfs2(int u,int dis)//返回值为自底向上的最短距离,dis储存的是到达当前点的最短距离 
{
	val[u]=dis;
	if(node[u].empty())
		return 1;
	int mmin=dis;//到当前节点的最近距离 
	for(auto it:node[u])
	{
		int v=it.second;
		mmin=min(deep[u],dfs2(v,mmin+1));
	}
	return mmin+1;
}

int main()
{
#ifndef ONLINE_JUDGE
//  freopen("data.in.txt","r",stdin);
//  freopen("data.out.txt","w",stdout);
#endif
//  ios::sync_with_stdio(false);
	int w;
	cin>>w;
	int kase=0;
	while(w--)
	{
		int n;
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
			node[i].clear();
		for(int i=2;i<=n;i++)
		{
			int fa;
			scanf("%d",&fa);
			node[fa].push_back(make_pair(0,i));
		}
		dfs1(1,0);
		dfs2(1,0);
		LL ans=0;
		for(int i=1;i<=n;i++)
			if(node[i].empty())
				ans+=val[i];
		printf("Case #%d: %lld\n",++kase,ans);
	}



















    return 0;
}