10381 2024. 04. 01 16:51:58 MagyarKendeSZLG Túra (75 pont) cpp17 Elfogadva 75/75 35ms 4608 KiB
#include <bits/stdc++.h>
using namespace std;

void solve() {
    int N;
    cin >> N;
    vector<bool> ugly(N + 1);
    vector<int> uglyNeighbor(N + 1);

    for (int i = 1; i <= N; i++) {
        int A;
        cin >> A;
        ugly[i] = A == -1;
    }

    bool running = true;
    for (int i = 1; i < N; i++) {
        int U, V;
        cin >> U >> V;

        if (running) {
            running = 0;
            if (ugly[U] && ugly[V]) {
                cout << "NEM\n2\n" << U << " " << V << "\n";
            }
            else if (uglyNeighbor[U] && ugly[V]) {
                cout << "NEM\n3\n" << uglyNeighbor[U] << " " << U << " " << V << "\n";
            }
            else if (ugly[U] && uglyNeighbor[V]) {
                cout << "NEM\n3\n" << uglyNeighbor[V] << " " << V << " " << U << "\n";
            }
            else running = 1;
        }
        
        if (ugly[U]) {
            uglyNeighbor[V] = U;
        }
        if (ugly[V]) {
            uglyNeighbor[U] = V;
        }
    }

    if (running) cout << "IGEN\n";
}

int main() {
	cin.tie(0), ios::sync_with_stdio(0);
    int T;
    cin >> T;
    while (T--) solve();
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 75/75
1 Elfogadva 0/0 3ms 1696 KiB
2 Elfogadva 0/0 3ms 1932 KiB
3 Elfogadva 6/6 3ms 2144 KiB
4 Elfogadva 3/3 4ms 2508 KiB
5 Elfogadva 3/3 4ms 2432 KiB
6 Elfogadva 4/4 4ms 2436 KiB
7 Elfogadva 4/4 4ms 2564 KiB
8 Elfogadva 4/4 4ms 2628 KiB
9 Elfogadva 2/2 28ms 2912 KiB
10 Elfogadva 2/2 29ms 3128 KiB
11 Elfogadva 2/2 29ms 3296 KiB
12 Elfogadva 2/2 29ms 3508 KiB
13 Elfogadva 2/2 29ms 3820 KiB
14 Elfogadva 3/3 29ms 3932 KiB
15 Elfogadva 2/2 29ms 3892 KiB
16 Elfogadva 2/2 29ms 3944 KiB
17 Elfogadva 2/2 29ms 4008 KiB
18 Elfogadva 2/2 29ms 4000 KiB
19 Elfogadva 3/3 29ms 3948 KiB
20 Elfogadva 2/2 30ms 3884 KiB
21 Elfogadva 2/2 30ms 3940 KiB
22 Elfogadva 3/3 32ms 3888 KiB
23 Elfogadva 2/2 32ms 4536 KiB
24 Elfogadva 2/2 32ms 4536 KiB
25 Elfogadva 2/2 32ms 4224 KiB
26 Elfogadva 2/2 32ms 4224 KiB
27 Elfogadva 2/2 32ms 4224 KiB
28 Elfogadva 2/2 30ms 4288 KiB
29 Elfogadva 2/2 34ms 4300 KiB
30 Elfogadva 2/2 32ms 4372 KiB
31 Elfogadva 2/2 34ms 4292 KiB
32 Elfogadva 2/2 35ms 4608 KiB