Submission #1241417

#	Time	Username	Problem	Language	Result	Execution time	Memory
1241417		dosts	참나무 (IOI23_beechtree)	C++20	0 / 100	85 ms	25272 KiB

beechtree

#include "beechtree.h"
#include <bits/stdc++.h>
//#define int long long
#define pii pair<int,int> 
#define vi vector<int>
#define ff first
#define ss second
#define sp << " " << 
#define all(x) x.begin(),x.end()
using namespace std;

const int LIM = 2e5+1;
vector<pii> edges[LIM];
vi ans(LIM,0),C,P;

int N;
vi ctr(LIM,0);


bool check(int node) {
    for (auto it : edges[node]) {
        ctr[it.ss] = 0;
    }
    for (auto it : edges[node]) {
        if (ctr[it.ss]) return false;
        ctr[it.ss]++;
    }
    //hepsi farklı
    int cocu = 0;
    for (auto it : edges[node]) {
        if (edges[it.ff].size()) cocu++;
    }
    if (cocu > 1) return false;
    if (!cocu) return true;
    for (auto it : edges[node]) {
        for (auto itt : edges[it.ff]) ctr[itt.ss] =0 ;
    }
    for (auto it : edges[node]) ctr[it.ss]++;
    for (auto it : edges[node]) {
        for (auto itt : edges[it.ff]) {
            if (!ctr[itt.ss]) return false;
        }
    }
    return true;
}


int dfs(int node) {
    for (auto it : edges[node]) dfs(it.ff);
   return ans[node] = check(node);
}

std::vector<int> beechtree(int N_, int M, std::vector<int> P_, std::vector<int> C_)
{
    N = N_;
    P = P_,C = C_;
    ans.assign(N_,0);
    for (int i = 0;i<N;i++) edges[i].clear(); 
    vi v;
    for (auto it : C) v.push_back(it);
    sort(all(v));
    v.erase(unique(all(v)),v.end());
    for (auto& it : C) it = lower_bound(all(v),it)-v.begin();
    for (int i = 1;i<N;i++) {
        edges[P[i]].push_back({i,C[i]});
    }
    dfs(0);
    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