Submission #1058245

#	Time	Username	Problem	Language	Result	Execution time	Memory
1058245		Triseedot	Beech Tree (IOI23_beechtree)	C++17	9 / 100	2075 ms	15824 KiB

beechtree

// In honor of El Psy Congroo
#include <bits/stdc++.h>
using namespace std;

#ifdef LOCAL
#include <debug.h>
#else
#define debug(...)
#endif

using ll = long long;
#define all(v) v.begin(), v.end()
#define len(v) int(v.size())

vector<int> beechtree(int N, int M, vector<int> P, vector<int> C) {
    vector<vector<int>> g(N);
    for (int u = 1; u < N; u++) {
        int v = P[u];
        g[v].push_back(u);
    }
    
    auto compress = [] (vector<int> &a) {
        int n = len(a);
        vector<pair<int, int>> b(n);
        for (int i = 0; i < n; i++) {
            b[i] = {a[i], i};
        }
        sort(all(b));
        int curr = 0;
        for (int i = 0; i < n; i++) {
            if (i != 0 && b[i].first != b[i - 1].first) curr++;
            a[b[i].second] = curr;
        }
    };
    compress(C);
    
    vector<bool> bad(N, false);
    for (int v = 0; v < N; v++) {
        vector<int> cnt(N, 0);
        for (int u : g[v]) {
            if (cnt[C[u]]) bad[v] = true;
            cnt[C[u]]++;
        }
    }
    
    vector<int> cnt(N, 0), cnt_leaf(N, 0);
    vector<vector<int>> q(N + 1);
    auto calc = [&] (int root) {
        int curr = 0;
        function<void(int)> calc_cnt = [&] (int v) {
            if (len(g[v]) == 0) {
                cnt_leaf[C[v]]++;
            }
            for (int u : g[v]) {
                if (cnt[C[u]] == 0) curr++;
                cnt[C[u]]++;
                calc_cnt(u);
            }
        };
        function<void(int)> reset = [&] (int v) {
            cnt[C[v]] = 0;
            cnt_leaf[C[v]] = 0;
            q[len(g[v])].clear();
            for (int u : g[v]) {
                reset(u);
            }
        };
        calc_cnt(root);
        
        q[len(g[root])].push_back(root);
        while (!q[curr].empty() && curr > 0) {
            int v = q[curr].back();
            q[curr].pop_back();
            if (bad[v]) continue;
            
            for (int u : g[v]) {
                cnt[C[u]]--;
                if (cnt[C[u]] == 0) {
                    curr--;
                }
                q[len(g[u])].push_back(u);
            }
        }
        
        reset(root);
        return curr == 0;
    };
    
    vector<int> res(N);
    for (int i = 0; i < N; i++) res[i] = calc(i);
    return res;
}

• Subtask 0: 0 / 0
• Subtask 1: 0 / 9
• Subtask 2: 0 / 5
• Subtask 3: 9 / 9
• Subtask 4: 0 / 8
• Subtask 5: 0 / 14
• Subtask 6: 0 / 14
• Subtask 7: 0 / 12
• Subtask 8: 0 / 17
• Subtask 9: 0 / 12