제출 #1211386

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1211386		omsincoconut	참나무 (IOI23_beechtree)	C++17	0 / 100	1 ms	328 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);
    
    vector<int> maxdeg(N), subsz(N);
    vector<vector<int>> have(N, vector<int>(2, 0));
    for (int i = N-1; i >= 0; i--) {
        ret[i] = true;
        subsz[i] = 1;

        for (int v : child[i]) {
            maxdeg[i] = max(maxdeg[i], maxdeg[v]);
            have[i][0] += have[v][0];
            have[i][1] += have[v][1];
            subsz[i] += subsz[v];
        }

        if (maxdeg[i] >= 2 || child[i].size() > 2) ret[i] = false;

        maxdeg[i] = max(maxdeg[i], (int)child[i].size());
        
        if (child[i].size() == 2 && subsz[child[i][0]] >= 2 && subsz[child[i][1]] >= 2) {
            ret[i] = false;
        }

        for (int v : child[i]) {
            if (have[v][0] > 0 && have[v][1] > 0) ret[i] = false;
        }

        if (i != 0) have[i][C[i]-1]++;
    }

    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