Submission #1242540

#	Time	Username	Problem	Language	Result	Execution time	Memory
1242540		urosk	참나무 (IOI23_beechtree)	C++20	0 / 100	2094 ms	18620 KiB

beechtree

#include "beechtree.h"
#include "bits/stdc++.h"
#define pb push_back
#define popb pop_back
#define fi first
#define sc second
#define si(a_) (ll)(a_.size())
#define all(a_) a_.begin(),a_.end()
#define here cerr<<"========================================\n"
#define dbg(X_) cerr<<#X_<<": "<<X_<<endl;
#define ceri(a_,l_,r_) {for(ll i_=l_;i_<=r_;i_++) cerr<<a_[i_]<<" ";cerr<<endl;}
using namespace std;

using ll = int;
using pll = pair<ll,ll>;

const ll maxn = 200005;
ll n,m;
vector<pll> g[maxn];
bool oki[maxn];
vector<int> ans;
ll up[maxn];
ll c[maxn];
ll cnt[maxn];
bool oksub(ll i){
    bool oke = 1;
    for(pll p : g[i]){
        ll j = p.fi,col = p.sc;
        if(cnt[col]) oke = 0;
        cnt[col] = 1;
    }
    for(pll p : g[i]){
        ll j = p.fi,col = p.sc;
        cnt[col] = 0;
    }
    return oke;
}
ll dub[maxn];
ll rut;
vector<ll> v;
queue<ll> q;
std::vector<int> beechtree(int N, int M, std::vector<int> P, std::vector<int> C)
{
    n = N,m = M;
    for(ll i = 2;i<=n;i++){
        ll par = P[i-1]+1;
        g[par].pb({i,C[i-1]-1});
        up[i] = par;
        c[i] = C[i-1]-1;
    }
    ans.resize(n);
    for(ll i = 0;i<n;i++) ans[i] = 0;
    for(ll i = 1;i<=n;i++){
        rut = i;
        v.clear();
        cnt[0] = cnt[1] = 0;
        q.push(i);
        bool oke = 1;
        while(si(q)){
            ll u = q.front(); q.pop();
            if(u!=i){
                if(v[cnt[c[u]]]!=up[u]) oke = 0;
                cnt[c[u]]++;
            }
            v.pb(u);
            for(pll p : g[u]){
                ll s = p.fi;
                q.push(s);
            }
        }
        ans[i-1] = oke;
    }
    return ans;
}

• 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