Submission #1088006

#	Time	Username	Problem	Language	Result	Execution time	Memory
1088006		Valaki2	Beech Tree (IOI23_beechtree)	C++17	5 / 100	61 ms	20132 KiB

beechtree

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

#define pb push_back

const int maxn = 2e5;

int n, k;
vector<int> g[maxn];
int par[maxn];
int col[maxn];
int ans[maxn];

void solve() {
    for(int i = n - 2; i >= 0; i--) {
        if(i != 0) if(col[i] != col[i + 1]) {
            ans[i - 1] = 0;
        }
        if(ans[i + 1] == 0) {
            ans[i] = 0;
        }
    }
}

vector<int> beechtree(int N, int M, vector<int> P, vector<int> C)
{
    n = N, k = M;
    for(int i = 0; i < n; i++) {
        par[i] = P[i];
        col[i] = C[i];
        if(i > 0) {
            g[par[i]].pb(i);
        }
        ans[i] = 1;
    }
    solve();
    vector<int> answer(n);
    for(int i = 0; i < n; i++) {
        answer[i] = ans[i];
    }
    return answer;
}

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