oj.uz

Submission #704679

# Submission time Handle Problem Language Result Execution time Memory
704679 2023-03-02T15:27:07 Z Ronin13 Cat in a tree (BOI17_catinatree) C++14
0 / 100
 8 ms 12104 KB 
#include <bits/stdc++.h>
#define ll long  long
#define f first
#define s second
#define pii pair<int,int>
#define pll pair<ll,ll>
#define pb push_back
#define epb emplace_back

using namespace std;
const int nmax = 2e5 + 1;
vector <vector <pii> > cent(nmax);

vector <bool> marked(nmax);
vector <int> sz(nmax);
vector <int> mx(nmax);
vector <vector <int> > g(nmax);
vector <int> cur(nmax, 1e9);

void dfs(int v, int e = -1){
    sz[v] = 1;
    mx[v] = 0;
    for(int to : g[v]){
        if(marked[to]) continue;
        if(to == e) continue;
        dfs(to, v);
        sz[v] += sz[to];
        mx[v] = max(mx[v], sz[to]);
    }
}

int find_cen(int v, int s, int e = -1){
    if(mx[v] <= s / 2){
        if(sz[v] >= s / 2) return v;
    }
    for(int to : g[v]){
        if(to == e || marked[to]) continue;
       int x  = find_cen(to, s, v);
       if(x != -1) return x;
    }
    return -1;
}

void DFS(int v, int d, int st,  int e = -1){
    cent[v].pb({d, st});
    for(int to : g[v]){
        if(to == e || marked[to]) continue;
        DFS(to, d + 1, st, v);
    }
}

void rec(int v){
    dfs(v);
    int x = find_cen(v, sz[v]);
    DFS(x, 0, x);
    marked[x] = true;
    for(int to : g[x]){
        if(!marked[to])
            rec(to);
    }
}

int getmn(int v){
    int ans = 1e9;
    for(auto cc : cent[v]){
        int x = cc.s, y = cc.f;
        ans = min(ans, cur[x] + y);
    }
    return ans;
}

void mark(int v){
    for(auto cc : cent[v]){
        int x = cc.s, y = cc.f;
        cur[x] = min(cur[x], y);
    }
}

vector <pii> vv;

void stupidguy(int v, int d, int e = -1){
    vv.pb({d, v});
    for(int to : g[v]){
        if(to == e) continue;
        stupidguy(to, d + 1, v);
    }
}

int main(){
    ios_base::sync_with_stdio(false); cin.tie(0);
    int n; cin >> n;
    int D; cin >> D;
    for(int i = 1; i < n; i++){
        int x; cin >> x;
        g[x].pb(i);
        g[i].pb(x);
    }
    rec(0);
    sort(vv.begin(), vv.end());
    int ans = 0;
    stupidguy(0, 0);
    for(int i = vv.size() - 1; i >= 0; i--){
        int x = vv[i].s;
        if(getmn(x) >= D){
            ans++;
            mark(x);
        }
    }
    cout << ans;
    return 0;
}

/* - - - - - - - - - - - - - -
  |             ##            |
  |      #      ##      #     |
  |    #####    ##    #####   |
  |      #      ##      #     |
  |             ##            |
  | ##########################|
  |             ##            |
  |      #      ##      #     |
  |    #####    ##    #####   |
  |      #      ##      #     |
  |             ##            |
   - - - - - - - - - - - - - -
*/

Subtask #1
0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 12020 KB Output is correct
2 Correct 8 ms 12048 KB Output is correct
3 Correct 6 ms 11988 KB Output is correct
4 Correct 6 ms 11988 KB Output is correct
5 Correct 6 ms 11988 KB Output is correct
6 Correct 7 ms 11988 KB Output is correct
7 Correct 6 ms 12104 KB Output is correct
8 Correct 7 ms 11988 KB Output is correct
9 Correct 6 ms 11988 KB Output is correct
10 Incorrect 6 ms 11988 KB Output isn't correct
11 Halted 0 ms 0 KB -

Subtask #2
0 / 40.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 12020 KB Output is correct
2 Correct 8 ms 12048 KB Output is correct
3 Correct 6 ms 11988 KB Output is correct
4 Correct 6 ms 11988 KB Output is correct
5 Correct 6 ms 11988 KB Output is correct
6 Correct 7 ms 11988 KB Output is correct
7 Correct 6 ms 12104 KB Output is correct
8 Correct 7 ms 11988 KB Output is correct
9 Correct 6 ms 11988 KB Output is correct
10 Incorrect 6 ms 11988 KB Output isn't correct
11 Halted 0 ms 0 KB -

Subtask #3
0 / 49.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 12020 KB Output is correct
2 Correct 8 ms 12048 KB Output is correct
3 Correct 6 ms 11988 KB Output is correct
4 Correct 6 ms 11988 KB Output is correct
5 Correct 6 ms 11988 KB Output is correct
6 Correct 7 ms 11988 KB Output is correct
7 Correct 6 ms 12104 KB Output is correct
8 Correct 7 ms 11988 KB Output is correct
9 Correct 6 ms 11988 KB Output is correct
10 Incorrect 6 ms 11988 KB Output isn't correct
11 Halted 0 ms 0 KB -