제출 #1211385

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

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

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

        maxdeg[i] = max(maxdeg[i], (int)child[i].size());

        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