Submission #1046474

#	Time	Username	Problem	Language	Result	Execution time	Memory
1046474		Trustfulcomic	Beech Tree (IOI23_beechtree)	C++17	0 / 100	2092 ms	19272 KiB

beechtree

#include "beechtree.h"

#include<bits/stdc++.h>
using namespace std;

int n,m;
vector<int> p,c;
vector<vector<int>> g;

int good(int r){
    vector<int> ch;
    queue<int> q; q.push(r);
    while(!q.empty()){
        int n = q.front(); q.pop();
        ch.push_back(n);
        for (int i : g[n]) q.push(i);
    }

    if (ch.size() == 1) return 1;
    
    bool jupi = false;
    do {
        vector<int> count(m, 0);
        vector<int> is(ch.size(), -1);
        is[0] = r;
        bool good = true;

        for (int i = 0; i<ch.size() && good; i++){
            if (ch[i] == r) continue;
            if (p[ch[i]] != is[count[c[ch[i]]]]) good = false;
            count[c[ch[i]]]++;
            is[i+1]=ch[i];
        }

        if (good){
            jupi = true;
        }

    } while(next_permutation(ch.begin(), ch.end()));

    if (jupi) return 1;
    return 0;
}

std::vector<int> beechtree(int N, int M, std::vector<int> P, std::vector<int> C)
{   
    n = N;
    m = M;
    p = P;
    c = C;
    g.resize(n);
    for (int i = 1; i<n; i++){
        g[p[i]].push_back(i);
    }

    vector<int> res(n);
    for (int i = 0; i<n; i++){
        res[i] = good(i);
    }
    return res;
}

Compilation message (stderr)

beechtree.cpp: In function 'int good(int)':
beechtree.cpp:28:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   28 |         for (int i = 0; i<ch.size() && good; 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