【AGC004D】Teleporter 贪心+DFS

原题走这里

又是一道奇妙的题

首先，为满足要求，首都必须指向自己，将首都指向自己后，整张图可以构成一棵树

问题就转化成了，对于一颗有根树，如何将其分割成多棵树，使各棵树满足：

树根是1时，深度不大于K

树根不是1时，深度不大于K-1

且使得切割的边数最少

做法是贪心，做一个DFS，在回溯过程中，计算某子树在切割后的最大深度

如果符合条件，立刻将该子树切割开

最后如果首都没有指向自己就再加上1即可

复杂度O(n)

代码如下

#include <bits/stdc++.h>
using namespace std;
struct edge
{
	int v,p;
}e[100010];
int n,k,a[100010],top,head[100010],cnt;
void addedge(int u,int v)
{
	e[++top]=(edge){v,head[u]};
	head[u]=top;	
}
int dfs(int u,int dep)
{
	int ret=dep;
	for(int i=head[u];i;i=e[i].p)
	{
		ret=max(ret,dfs(e[i].v,dep+1));
	}
	if(a[u]!=1&&ret-dep==k-1)
	{
		cnt++;
		return 0;
	}
	return ret;
}
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",a+i);
	}
	if(a[1]!=1)cnt++;
	a[1]=1;
	for(int i=2;i<=n;i++)
	{
		addedge(a[i],i);
	}
	dfs(1,0);
	cout<<cnt<<endl;
	return 0;
}