Submission #1692479

---

Source Code Expand

//2017-10-18
//miaomiao
//
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

#define pb push_back
#define For(i, a, b) for(int i = (a); i <= (int)(b); ++i)

#define N (100000+5)

int fa[N], rk, ans;
vector<int> G[N];

int Dfs(int now){
	int D = 0;
	For(i, 0, G[now].size()-1) D = max(D, Dfs(G[now][i])+1);

	if(D == rk-1){
		if(fa[now] != 1) ++ans;
		return -1;
	}
	return D;
}

int main(){
	int n;

	scanf("%d%d", &n, &rk);
	For(i, 1, n){
		scanf("%d", &fa[i]);
		if(i == 1) ans += fa[i] != 1, fa[i] = 0;
		if(i != 1) G[fa[i]].pb(i);
	}

	For(i, 0, G[1].size()-1) Dfs(G[1][i]);
	printf("%d\n", ans);

	return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:36:24: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d%d", &n, &rk);
                        ^
./Main.cpp:38:22: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d", &fa[i]);
                      ^

Judge Result