【Atcoder Grand Contest 004】D - Teleporter——拓扑排序+贪心

题目链接

题目大意：给定n个点，告诉你第i个点的父亲是fa[i]，求最少需要改变多少个点的父亲使得所有点走k步均到达1。

 

---

 

 

分析：首先明确一点，如果要改变一个点的父亲就干脆把它的父亲改为1，这样一定是最优的。

　　   如果1的父亲是1本身，那么别的点走到1的步数只要<=k即可（1为自环）；

　　   而当1的父亲不是1时，需要用一次操作将1的父亲改成1。

　　   考虑一棵深度为k-1且根节点父亲不为1的子树，容易证明此时选取根节点是最优的（1次操作是最小代价）。

　　   这样的话就可以直接贪心了：从叶子节点向上拓扑排序，每次看该节点子树的深度是否为k-1即可。

 

 

---

 

代码：

 

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 const int N=1e5+10;
 5 int n,k,sum=0,count[N],q[N],h=1,t=0,fa[N],ll[N];
 6 int read(){
 7     int f=1,ans=0;char c=getchar();
 8     while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
 9     while(c>='0'&&c<='9'){ans=ans*10+c-48;c=getchar();}
10     return ans*f;
11 }
12 int main(){
13     n=read();k=read();
14     int a=read();if(a!=1)sum++;
15     for(int i=2;i<=n;i++){
16         a=read();fa[i]=a;count[a]++;
17     }
18     for(int i=2;i<=n;i++)if(!count[i])q[++t]=i;
19     while(h<=t){
20         int x=q[h];h++;
21         if(x==1)break;
22         if(fa[x]!=1&&ll[x]==k-1)sum++;
23         else ll[fa[x]]=std::max(ll[fa[x]],ll[x]+1);
24         count[fa[x]]--;
25         if(!count[fa[x]])q[++t]=fa[x];
26     }
27     printf("%d\n",sum);
28     return 0;
29 }

agc004 D

 

转载于:https://www.cnblogs.com/JKAI/p/7604127.html