Submission #866214

---

Source Code Expand

#include <cstdio>
#include <vector>
#include <algorithm>

using namespace std;

int k, ans;
int a[100000];
int b[100000];
vector <int> v[100000];

int dfs(int x, int y) {
    int i;
    
    a[x] = y;
    
    for (i = 0; i < v[x].size(); i++) b[x] = max(b[x], dfs(v[x][i], y + 1) + 1);
    
    return b[x];
}

void dfs2(int x, int y) {
    int i;
    
    if (b[x] + y <= k) return;
    
    if (y > k) {
        ans++;
        
        for (i = 0; i < v[x].size(); i++) dfs2(v[x][i], 2);
    } else {
        for (i = 0; i < v[x].size(); i++) {
            if (b[v[x][i]] + y + 1 > k && b[v[x][i]] + 2 <= k) break;
        }
        
        if (i < v[x].size()) {
            ans++;
            
            for (i = 0; i < v[x].size(); i++) dfs2(v[x][i], 2);
        } else {
            for (i = 0; i < v[x].size(); i++) dfs2(v[x][i], y + 1);
        }
    }
}

int main() {
    int n, i;
    
    scanf("%d %d", &n, &k);
    
    for (i = 0; i < n; i++) {
        int x;
        
        scanf("%d", &x);
        
        x--;
        
        if (i == 0) {
            if (x != 0) ans++;
        } else {
            v[x].push_back(i);
        }
    }
    
    dfs(0, 0);
    
    for (i = 0; i < v[0].size(); i++) dfs2(v[0][i], 1);
    
    printf("%d\n", ans);
    
    return 0;
}

Submission Info

Compile Error

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

Judge Result