167972025-05-13 12:51:20BucsMateKülönböző katicákcpp17Wrong answer 10/10090ms15412 KiB
#include <iostream>
#include <vector>

using namespace std;

const int INF = 1000*1000*1000;

vector<vector<int>> adj;
vector<pair<int, int>> intervals;
vector<int> parity, depth;

void DFS1(int node, int parent)
{
    for(int i = 0; i < adj[node].size(); i++){
        int newNode = adj[node][i];
        depth[newNode] = depth[node] + 1;

        DFS1(newNode, node);

        intervals[node].first = max(intervals[node].first, intervals[newNode].first-1);
        intervals[node].second = min(intervals[node].second, intervals[newNode].second+1);
        if(intervals[node].first > intervals[node].second || (parity[node] != -1 && parity[node] == parity[newNode])){
            cout << "NEM";
            exit(0);
        }
        if(parity[newNode] != -1){
            parity[node] = 1 - parity[newNode];
        }
    }
}

void DFS2(int node, int parent)
{
    intervals[node].first = max(intervals[parent].first, intervals[parent].first-1);
    intervals[node].second = min(intervals[parent].second, intervals[parent].second+1);
    if(intervals[node].first > intervals[node].second || (parity[node] != -1 && parity[node] == parity[parent])){
        cout << "NEM";
        exit(0);
    }
    if(parity[parent] != -1){
        parity[node] = 1 - parity[parent];
    }
    for(int i = 0; i < adj[node].size(); i++){
        int newNode = adj[node][i];
        DFS2(newNode, node);
    }
}

int main()
{
    int N;
    cin >> N;
    adj.resize(N+1);
    for(int i = 1; i <= N; i++){
        int p;
        cin >> p;
        adj[p].push_back(i);
    }
    intervals.assign(N+1, {0, INF});
    parity.assign(N+1, -1);
    for(int i = 1; i <= N; i++){
        int val;
        cin >> val;
        if(val != -1){
            intervals[i] = {val, val};
            parity[i] = val%2;
        }
    }

    depth.assign(N+1, -1);
    depth[1] = 1;

    DFS1(1, 0);
    DFS2(1, 0);
    cout << "IGEN\n";
    if(intervals[1].first == 0 && intervals[1].second == INF){
        for(int i = 1; i <= N; i++){
            cout << depth[i] << " ";
        }
    }
    else{
        for(int i = 1; i <= N; i++){
            if(intervals[i].first == 0 && parity[i] == 1){
                cout << 1 << " ";
            }
            else{
                cout << intervals[i].first << " ";
            }
        }
    }
    return 0;
}
SubtaskSumTestVerdictTimeMemory
subtask10/0
1Wrong answer1ms316 KiB
2Accepted1ms316 KiB
3Wrong answer71ms5940 KiB
subtask210/10
4Accepted63ms5172 KiB
5Accepted68ms5684 KiB
6Accepted75ms6196 KiB
7Accepted81ms6708 KiB
subtask30/15
8Accepted71ms14376 KiB
9Accepted68ms14388 KiB
10Wrong answer90ms15412 KiB
11Wrong answer82ms14900 KiB
12Wrong answer85ms14644 KiB
13Wrong answer83ms15032 KiB
14Accepted82ms14388 KiB
subtask40/35
15Accepted2ms508 KiB
16Accepted2ms508 KiB
17Accepted2ms316 KiB
18Wrong answer2ms508 KiB
19Wrong answer2ms316 KiB
20Wrong answer1ms316 KiB
21Wrong answer2ms440 KiB
22Wrong answer2ms456 KiB
subtask50/40
23Accepted75ms6188 KiB
24Accepted63ms5952 KiB
25Accepted64ms6304 KiB
26Wrong answer75ms6076 KiB
27Wrong answer75ms6184 KiB
28Wrong answer81ms6452 KiB
29Wrong answer79ms6452 KiB
30Wrong answer75ms6196 KiB
31Wrong answer76ms6296 KiB
32Wrong answer78ms6364 KiB
33Wrong answer82ms6708 KiB