# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
1046569 | 0npata | 참나무 (IOI23_beechtree) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "beechtree.h"
#include<bits/stdc++.h>
using namespace std;
int n, m;
vector<int> p, c;
vector<vector<int>> g;
int good(int r){
vector<int> ch;
queue<int> q; q.push(r);
while(!q.empty()){
int n = q.front(); q.pop();
ch.push_back(node);
for (int i : g[node]) q.push(i);
}
if (ch.size() == 1) return 1;
bool jupi = false;
do {
vector<int> count(m, 0);
vector<int> is(ch.size(), -1);
is[0] = r;
bool good = true;
for (int i = 0; i < ch.size() && good; i++){
if (ch[i] == r) continue;
if (count[c[ch[i]]] >= is.size() || p[ch[i]] != is[count[c[ch[i]]]]) good = false;
count[c[ch[i]]]++;
if (i+1 < is.size()) is[i+1] = ch[i];
}
if (good){
jupi = true;
}
} while(next_permutation(ch.begin(), ch.end()));
if (jupi) return 1;
return 0;
}
std::vector<int> beechtree(int N, int M, std::vector<int> P, std::vector<int> C)
{
n = N;
m = M;
p = P;
c = C;
g.resize(n);
for (int i = 1; i < n; i++){
if (p[i] < g.size()) {
g[p[i]].push_back(i);
}
}
vector<int> res(n);
for (int i = 0; i < n; i++){
res[i] = good(i);
}
return res;
}