51062023-04-17 17:47:50rmlanKülönböző katicákcpp14Wrong answer 10/10083ms46192 KiB
#include<bits/stdc++.h>
using namespace std;

int n;
bool jo=1;
vector<int> p,m;
vector<pair<int, int> > l,e;
vector<vector<int>> g;

void dfs1(int u){
    for(int v:g[u]){
        dfs1(v);
        e.push_back({v,u});
        if(l[u].first == 0 && l[u].second == 1e6) continue;

        l[v].first = max(max(l[v].first, l[u].first-1),0);
        l[v].second = min(l[v].second, l[u].second+1);
        if(l[v].first%2 == l[u].first%2 || l[v].first > l[v].second) jo=0;
    }
}

int main(){
    pair<int, int> df = {0,1e6};
    cin >> n;
    p.resize(n+1);
    l.resize(n+1);
    g.resize(n+1);
    m.resize(n+1);


    for(int i = 1; i <= n; i++){
        l[i] = {0, 1e6};
        cin>>p[i];
        g[p[i]].push_back(i);

    }
    for(int i = 1; i <= n; i++){
        cin >> m[i];
        if(m[i] != -1) l[i]={m[i], m[i]};
    }
    dfs1(1);
    reverse(e.begin(), e.end());
    for(pair<int, int> pa:e){
        int fi=pa.first,se=pa.second;
        if(l[se]!=df && l[fi]!=df &&l[se].first%2 == l[fi].first%2) jo=0;
        l[se].first=max(max(l[se].first, l[fi].first-1),0);
        l[se].second=min(l[se].second, l[fi].second+1);
        if(l[se].first > l[se].second) jo = 0;
    }
    if(!jo){
        cout << "NEM";
        return 0;
    }
    cout << "IGEN\n";

    for(int i = 1; i <= n; i++){
        if(l[1] == df){
            cout << m[p[i]]+1 << " ";
            m[i] = m[p[i]]+1;
            continue;
        }
        cout << l[i].second << " ";
    }
}
SubtaskSumTestVerdictTimeMemory
subtask10/0
1Accepted3ms1876 KiB
2Accepted3ms1912 KiB
3Wrong answer71ms15628 KiB
subtask210/10
4Accepted61ms15100 KiB
5Accepted68ms17232 KiB
6Accepted74ms19128 KiB
7Accepted82ms21136 KiB
subtask30/15
8Accepted74ms41300 KiB
9Accepted75ms42056 KiB
10Wrong answer76ms44284 KiB
11Wrong answer82ms44512 KiB
12Wrong answer74ms44608 KiB
13Wrong answer75ms46192 KiB
14Accepted79ms45608 KiB
subtask40/35
15Accepted3ms14076 KiB
16Wrong answer3ms14128 KiB
17Wrong answer3ms14388 KiB
18Wrong answer3ms14364 KiB
19Wrong answer3ms14352 KiB
20Wrong answer3ms14304 KiB
21Wrong answer3ms14568 KiB
22Wrong answer3ms14548 KiB
subtask50/40
23Accepted74ms28420 KiB
24Accepted68ms29784 KiB
25Accepted70ms31216 KiB
26Wrong answer75ms31212 KiB
27Wrong answer74ms31696 KiB
28Wrong answer82ms33328 KiB
29Wrong answer79ms34336 KiB
30Wrong answer78ms34720 KiB
31Wrong answer67ms35344 KiB
32Wrong answer81ms36576 KiB
33Wrong answer83ms38100 KiB