167972025-05-13 12:51:20BucsMateKülönböző katicákcpp17Hibás válasz 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;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
subtask10/0
1Hibás válasz1ms316 KiB
2Elfogadva1ms316 KiB
3Hibás válasz71ms5940 KiB
subtask210/10
4Elfogadva63ms5172 KiB
5Elfogadva68ms5684 KiB
6Elfogadva75ms6196 KiB
7Elfogadva81ms6708 KiB
subtask30/15
8Elfogadva71ms14376 KiB
9Elfogadva68ms14388 KiB
10Hibás válasz90ms15412 KiB
11Hibás válasz82ms14900 KiB
12Hibás válasz85ms14644 KiB
13Hibás válasz83ms15032 KiB
14Elfogadva82ms14388 KiB
subtask40/35
15Elfogadva2ms508 KiB
16Elfogadva2ms508 KiB
17Elfogadva2ms316 KiB
18Hibás válasz2ms508 KiB
19Hibás válasz2ms316 KiB
20Hibás válasz1ms316 KiB
21Hibás válasz2ms440 KiB
22Hibás válasz2ms456 KiB
subtask50/40
23Elfogadva75ms6188 KiB
24Elfogadva63ms5952 KiB
25Elfogadva64ms6304 KiB
26Hibás válasz75ms6076 KiB
27Hibás válasz75ms6184 KiB
28Hibás válasz81ms6452 KiB
29Hibás válasz79ms6452 KiB
30Hibás válasz75ms6196 KiB
31Hibás válasz76ms6296 KiB
32Hibás válasz78ms6364 KiB
33Hibás válasz82ms6708 KiB