Submission #1521463

---

Source Code Expand

#include<bits/stdc++.h>
#define rep(i,n)for(int i=0;i<(n);i++)
using namespace std;
typedef pair<int, int>P;

P p[100000];
int a[100000];
bool used[100000];
vector<int>E[100000];

void dfs(int u) {
	used[u] = true;
	for (int v : E[u]) {
		if (!used[v]) {
			p[v].first = p[u].first + 1;
			dfs(v);
		}
	}
}
void dfs2(int u) {
	used[u] = true;
	for (int v : E[u]) {
		if (!used[v])dfs(v);
	}
}
int main(){
	int n, k; scanf("%d%d", &n, &k);
	rep(i, n) {
		scanf("%d", &a[i]); a[i]--;
		if (i != 0)E[a[i]].push_back(i);
		p[i].second = i;
	}
	int cnt = (a[0] != 0);
	a[0] = 0;
	dfs(0);
	sort(p, p + n, greater<>());
	memset(used, 0, sizeof(used));
	rep(i, n) {
		if (p[i].first <= k)break;
		if (used[p[i].second])continue;
		cnt++;
		int x = p[i].second;
		rep(j, k - 1)x = a[x];
		dfs2(x);
	}
	printf("%d\n", cnt);
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:27:33: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  int n, k; scanf("%d%d", &n, &k);
                                 ^
./Main.cpp:29:21: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d", &a[i]); a[i]--;
                     ^

Judge Result