Submission #1837196

---

Source Code Expand

#include <algorithm>
#include <climits>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <ctime>
#include <iostream>
#include <sstream>
#include <functional>
#include <map>
#include <string>
#include <cstring>
#include <vector>
#include <queue>
#include <stack>
#include <deque>
#include <set>
#include <list>
#include <numeric>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll> P;
const double PI = 3.14159265358979323846;
const double EPS = 1e-12;
const ll INF = 1LL<<29;
const ll mod = 1e9+7;
#define rep(i,n) for(int (i)=0;(i)<(ll)(n);++(i))
#define repd(i,n,d) for(ll (i)=0;(i)<(ll)(n);(i)+=(d))
#define all(v) (v).begin(), (v).end()
#define pb(x) push_back(x)
#define mp(x,y) make_pair((x),(y))
#define mset(m,v) memset((m),(v),sizeof(m))
#define chmin(X,Y) ((X)>(Y)?X=(Y),true:false)
#define chmax(X,Y) ((X)<(Y)?X=(Y),true:false)
#define fst first
#define snd second
#define UNIQUE(x) (x).erase(unique(all(x)),(x).end())
template<class T> ostream &operator<<(ostream &os, const vector<T> &v){int n=v.size();rep(i,n)os<<v[i]<<(i==n-1?"":" ");return os;}
#define N 100010
int n, k, a[N], d[N];
vector<int> g[N];
void dfs(int u, int x){
	d[u] = x;
	for(auto &&v: g[u]) dfs(v, x+1);
}
int main(){
	scanf("%d %d", &n, &k);
	rep(i, n) scanf("%d", a+i);
	rep(i, n) a[i]--;
	int res = 0;
	if(a[0]!=0){
		a[0] = 0;
		res++;
	}
	rep(i, n-1) g[a[i+1]].push_back(i+1);
	dfs(0, 0);
	priority_queue<P> q;
	rep(i, n) if(d[i]>k) q.push(P(d[i], i));
	while(!q.empty()){
		int u = q.top().snd; q.pop();
		int u2 = u;
		bool f = false;
		rep(_, k){
			if(d[u]==0){
				f = true;
				break;
			}
			u = a[u];
		}
		if(f) continue;
		res++;
		u = u2;
		rep(_, k){
			d[u] = 0;
			u = a[u];
		}
	}
	printf("%d\n", res);
	return 0;
}

Submission Info

Compile Error

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

Judge Result