#include "beechtree.h"
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef vector<int> vi;
typedef vector<ll> vl;
#define all(x) x.begin(), x.end()
#define sz(x) (int)x.size()
#define mk make_pair
#define pb push_back
#define fr first
#define sc second
vector<vi> edges;
vi vis;
vi beechtree(int N, int M, vi P, vi C) {
vi ans(N);
edges.clear();
edges.resize(N);
vis.clear();
vis.resize(N);
for(int i = 1; i <= N-1; i++)
edges[P[i]].pb(i), C[i]--;
vi color_qtd(M), in_sub(N);
for(int i = 0; i <= N-1; i++) { // check do i
vi sub;
queue<int> fila;
fila.push(i);
while(!fila.empty()) {
int x = fila.front();
fila.pop();
sub.pb(x);
for(auto viz : edges[x])
fila.push(viz), vis[viz] = vis[x] + 1;
}
sort(all(sub), [](int a, int b){
return mk(sz(edges[a]), -vis[a]) > mk(sz(edges[b]), -vis[b]);
});
ans[i] = 1;
for(int j = 1; j < sz(sub); j++) {
if(P[sub[j]] != sub[color_qtd[C[sub[j]]]]) ans[i] = 0;
color_qtd[C[sub[j]]]++;
}
for(auto it : sub)
in_sub[it] = color_qtd[C[it]] = vis[i] = 0;
}
return ans;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |