Submission #1211247

#	Time	Username	Problem	Language	Result	Execution time	Memory
1211247		omsincoconut	Beech Tree (IOI23_beechtree)	C++17	0 / 100	2095 ms	62284 KiB

beechtree

#include "beechtree.h"
#include <bits/stdc++.h>

using namespace std;

vector<int> beechtree(int N, int M, vector<int> P, vector<int> C) {
    vector<int> ret(N);

    vector<vector<int>> child(N);
    for (int i = 1; i < N; i++) child[P[i]].push_back(i);

    for (int i = 0; i < N; i++) {
        map<int, vector<int>> P_C;

        stack<int> dt;
        dt.push(i);
        while (!dt.empty()) {
            int u = dt.top();
            dt.pop();
            
            if (u != i) P_C[P[u]].push_back(C[u]);
            for (int v : child[u]) dt.push(v);
        }

        bool can = true;
        vector<tuple<int, int, vector<int>>> sz_PC;
        for (auto [p, c] : P_C) {
            sort(c.begin(), c.end());
            for (int j = 0; j < (int)c.size()-1; j++) {
                if (c[j] == c[j+1]) can = false;
            }
            sz_PC.emplace_back(c.size(), p, c);
        }

        if (!can) {
            ret[i] = can;
            continue;
        }

        sort(sz_PC.begin(), sz_PC.end());
        reverse(sz_PC.begin(), sz_PC.end());
        map<int, int> cnt;
        int cur = 0;
        for (auto [sz, p, c] : sz_PC) {
            for (int j : c) {
                if (cnt[j] < cur) can = false;
                cnt[j]++;
            }
            cur++;
        }

        ret[i] = can;
    }

    return ret;
}

• Subtask 0: 0 / 0
• Subtask 1: 0 / 9
• Subtask 2: 0 / 5
• Subtask 3: 0 / 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