[AGC004D] Teleporter [贪心]

题面：

传送门

思路：

分析可知，这道题中的图是一个环套内向树，首都在环上

首先有一个结论：当首都的出边指向首都时，一定最优（不然首都出发可能无法按时到达首都）（可以按时到达的情况也一定有到不了的）

这种情况下首都构成自环，其他边构成一棵指向首都的树

因此可以这样贪心地解决问题：

从所有的叶节点开始拓扑排序，记录每一个节点下属的深度最大的叶节点的深度

如果有一个节点的距离（上述深度）到达了k-1，那么就把这个节点拆下来接到首都上面，剩下的节点继续作为新叶节点拓扑排序

这样所有1的子树的深度都在k以内，到达首都以后可以走自环达到k

Code：

     #include<iostream>
     #include<cstdio>
     #include<cstring>
     #include<algorithm>
     using namespace std;
     inline int read(){
         int re=,flag=;char ch=getchar();
         while(ch>''||ch<''){
             if(ch=='-') flag=-;
             ch=getchar();
         }
         while(ch>=''&&ch<='') re=(re<<)+(re<<)+ch-'',ch=getchar();
         return re*flag;
     }
     int n,K,a[],dep[];
     int q[],head=,tail=,cnt[];
     int main(){
         int i,u,ans=;
         n=read();K=read();
         a[]=read();
         if(a[]!=) ans++,a[]=;
         for(i=;i<=n;i++) a[i]=read(),cnt[a[i]]++;
     //    for(i=1;i<=n;i++) cout<<cnt[i]<<ends;cout<<endl;
         for(i=;i<=n;i++) if(!cnt[i]) q[tail++]=i;
         while(head<tail){
             u=q[head++];
     //        cout<<"topo "<<u<<ends<<dep[u]<<endl;
             if(u==) break;
             if(a[u]!=&&dep[u]==K-) ans++;
             else dep[a[u]]=max(dep[a[u]],dep[u]+);
             cnt[a[u]]--;
             if(cnt[a[u]]==) q[tail++]=a[u];
         }
         printf("%d\n",ans);
     }