Submission #1067711

#	Time	Username	Problem	Language	Result	Execution time	Memory
1067711		vjudge1	Beech Tree (IOI23_beechtree)	C++17	0 / 100	76 ms	78924 KiB

beechtree

#include "beechtree.h"
#include<bits/stdc++.h>
using namespace std;
map<int,int>mp[200100];
vector<int>par,col;
int dep[200100];
vector<int>sub[200100],adj[200100];
int check(vector<int>v){
    map<int,int>mp;
    for(int i=1;i<v.size();i++)
        if(v[mp[col[v[i]]]++]!=par[v[i]])
            return 0;
    return 1;
}
int checkdep2(vector<int>nods){
    sort(nods.begin(),nods.end(),[](int a,int b){
        return sub[a].size()>sub[b].size();
    });
    vector<int>otherstuf;
    while(sub[nods.back()].size()==1)
        otherstuf.push_back(nods.back()),
        nods.pop_back();
    for(int i=1;i<nods.size();i++)
        for(auto K:adj[nods[i]])
            if(sub[K].size()==1)
                nods.push_back(K);
    vector<int>v2{0};
    for(auto K:adj[0])
        if(sub[K].size()==1)
            v2.push_back(K);
    nods.erase(nods.begin());
    for(auto i:nods)
        v2.push_back(i);
    return check(v2);
}

vector<int> beechtree(int N, int M, vector<int> P, vector<int> C){
    P[0]=0;
    par=P;
    col=C;
    vector<int> V(N,1);
    for(int i=N;--i;) adj[par[i]].push_back(i),
        dep[par[i]]=max(dep[par[i]],++dep[i]);
    for(int i=N;i--;){
        sub[i].push_back(i);
        if(P[i]>=0&&dep[i]<=2)
            for(auto d:sub[i])
                sub[P[i]].push_back(d);
        sort(sub[i].begin(),sub[i].end());
        if(i)V[i]=check(sub[i]);
    }
    dep[0]++;
    assert(dep[0]<=3);
    vector<int>VV(N);
    iota(VV.begin(),VV.end(),0);
    V[0]=checkdep2(VV);
    return V;
}

Compilation message (stderr)

beechtree.cpp: In function 'int check(std::vector<int>)':
beechtree.cpp:10:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   10 |     for(int i=1;i<v.size();i++)
      |                 ~^~~~~~~~~
beechtree.cpp: In function 'int checkdep2(std::vector<int>)':
beechtree.cpp:23:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   23 |     for(int i=1;i<nods.size();i++)
      |                 ~^~~~~~~~~~~~

• 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