제출 #1233994

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1233994		mariza	참나무 (IOI23_beechtree)	C++20	9 / 100	26 ms	3400 KiB

beechtree

#include "beechtree.h"
#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const ll N=8;

vector<ll> sub;
vector<ll> t[N];
void dfs(ll curr){
    sub.push_back(curr);
    for(auto nxt:t[curr]){
        dfs(nxt);
    }
}

vector<int> beechtree(int n, int m, vector<int> p, vector<int> c){
    for(ll i=1; i<n; i++){
        t[p[i]].push_back(i);
    }

    vector<int> ans;
    for(ll i=0; i<n; i++){
        sub.clear();
        dfs(i);
        
        ans.push_back(0);
        do{
            // cout<<i<<endl;
            ll f[n], x[m+1]={};
            for(auto j:sub){
                if(j==i) continue;
                f[j]=x[c[j]];
                x[c[j]]++;
            }

            bool ok=1;
            for(auto j:sub){
                if(j==i) continue;
                if(sub[f[j]]!=p[j]) ok=0;
            }

            ans[i]|=ok;
        } while(next_permutation(sub.begin()+1,sub.end()));
    }

    return ans;
}

• Subtask 0: 0 / 0
• Subtask 1: 9 / 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