167952025-05-13 12:47:17BucsMateKülönböző katicákcpp17Wrong answer 10/10092ms16176 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, {1, 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 == 1 && intervals[1].second == INF){
        for(int i = 1; i <= N; i++){
            cout << depth[i] << " ";
        }
    }
    else{
        for(int i = 1; i <= N; i++){
            cout << intervals[i].first << " ";
        }
    }
    return 0;
}
SubtaskSumTestVerdictTimeMemory
subtask10/0
1Wrong answer1ms316 KiB
2Accepted1ms316 KiB
3Wrong answer75ms6524 KiB
subtask210/10
4Accepted64ms5940 KiB
5Accepted71ms6452 KiB
6Accepted76ms6964 KiB
7Accepted83ms7476 KiB
subtask30/15
8Accepted74ms15412 KiB
9Accepted74ms15156 KiB
10Wrong answer92ms16176 KiB
11Wrong answer86ms15716 KiB
12Wrong answer82ms15532 KiB
13Wrong answer90ms15924 KiB
14Accepted81ms15176 KiB
subtask40/35
15Accepted2ms316 KiB
16Accepted2ms316 KiB
17Accepted2ms508 KiB
18Wrong answer2ms316 KiB
19Wrong answer2ms508 KiB
20Wrong answer2ms408 KiB
21Wrong answer2ms340 KiB
22Wrong answer2ms508 KiB
subtask50/40
23Accepted78ms6928 KiB
24Accepted68ms6800 KiB
25Accepted67ms7064 KiB
26Wrong answer76ms6912 KiB
27Wrong answer78ms6828 KiB
28Wrong answer86ms7220 KiB
29Wrong answer81ms7220 KiB
30Wrong answer79ms7220 KiB
31Wrong answer78ms6956 KiB
32Wrong answer81ms7180 KiB
33Wrong answer87ms7444 KiB