Description
维尼管辖的领土很大,我们可以抽象为n个城市,其中1号点为首都。这n个城市之有n条单向电缆,一条信息经过一条电缆进行传输所需时间会+1s,然而维尼并不能忍受时间白白被续,他要求从任何点发出的信息经过恰好ks后恰好出现在首都。
由于时间紧迫,维尼希望修改的电缆数量尽可能地少,他把这个任务交给了你。
Input
第一行两个整数n,k如题目所述
第二行n个正整数,其中第i个正整数a[i]表示有一条从i出发至a[i]的电缆。
题目保证在不做修改的情况下从任一点一定可以走到首都
Output
一行,一个整数,为修改的电缆条数
Sample Input
8 2
4 1 2 3 1 2 3 4
Sample Output
3
HINT
n<100000,k<1e9
补充样例:
3 1
2 3 1
输出:2
4 2
1 1 2 2
输出:0
Sol
模拟赛的题面。。。懒得改了qwq。
1连自环。
剩下的点dfs去寻找深度为k的子树并且找一个ans就++即可。然后删除这个子树。
Code
#include <bits/stdc++.h>
using namespace std;
vector<int>e[100005];int n,k,a[100005],cnt;
int dfs(int x,int dep)
{
int mx=dep;for(int i=0;i<e[x].size();i++) mx=max(mx,dfs(e[x][i],dep+1));
if(a[x]!=1&&mx-dep==k-1){cnt++;return 0;}else return mx;
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++) scanf("%d",&a[i]),(i!=1)?e[a[i]].push_back(i),0:cnt+=(a[i]!=1);
for(int i=0;i<e[1].size();i++) dfs(e[1][i],0);
printf("%d\n",cnt);
}