[AGC004D] Teleporter(贪心+)

Description

维尼管辖的领土很大,我们可以抽象为n个城市,其中1号点为首都。这n个城市之有n条单向电缆,一条信息经过一条电缆进行传输所需时间会+1s,然而维尼并不能忍受时间白白被续,他要求从任何点发出的信息经过恰好ks后恰好出现在首都。

由于时间紧迫,维尼希望修改的电缆数量尽可能地少,他把这个任务交给了你。

Input

第一行两个整数n,k如题目所述

第二行n个正整数,其中第i个正整数a[i]表示有一条从i出发至a[i]的电缆。

题目保证在不做修改的情况下从任一点一定可以走到首都

Output

一行,一个整数,为修改的电缆条数

题解:

严重吐槽本题翻译

这是一道挺有趣的题……

他要让任一点都在 \(K\) 秒到首都,那首都肯定要自环,可以一直转 \(K\) 秒。

那么剩下只有 \(n-1\) 条边了,变成了一颗很漂亮的树。

那我们选一些电缆,把它换成连向1,也就是把原树分成几棵树,其中这几棵树的最大深度不能超过 \(K-1\)。(显然吧,如果距离短了可以在首都的自环上绕)

然后这就可以贪心了!……

CODE:

#include<iostream>
#include<cstdio>
using namespace std;

int n,k,ans=0;
int tot=0,h[100005],a[100005];
struct Edge{
	int x,next;
}e[200005];

inline void add_edge(int x,int y){
	e[++tot].x=y;
	e[tot].next=h[x],h[x]=tot;
}

int dfs(int x,int deep){
	int ret=deep;
	for(int i=h[x];i;i=e[i].next)
		ret=max(ret,dfs(e[i].x,deep+1));
	if(a[x]!=1&&ret-deep==k-1)return ans++,0;
	else return ret;
}

int main(){
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)scanf("%d",a+i);
	if(a[1]!=1)ans=a[1]=1;
	for(int i=2;i<=n;i++)add_edge(a[i],i);
	dfs(1,0),printf("%d",ans);
}
posted @ 2018-08-17 14:14  ezoiLZH  阅读(189)  评论(0编辑  收藏  举报