103812024-04-01 16:51:58MagyarKendeSZLGTúra (75 pont)cpp17Accepted 75/7535ms4608 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();
}
SubtaskSumTestVerdictTimeMemory
base75/75
1Accepted0/03ms1696 KiB
2Accepted0/03ms1932 KiB
3Accepted6/63ms2144 KiB
4Accepted3/34ms2508 KiB
5Accepted3/34ms2432 KiB
6Accepted4/44ms2436 KiB
7Accepted4/44ms2564 KiB
8Accepted4/44ms2628 KiB
9Accepted2/228ms2912 KiB
10Accepted2/229ms3128 KiB
11Accepted2/229ms3296 KiB
12Accepted2/229ms3508 KiB
13Accepted2/229ms3820 KiB
14Accepted3/329ms3932 KiB
15Accepted2/229ms3892 KiB
16Accepted2/229ms3944 KiB
17Accepted2/229ms4008 KiB
18Accepted2/229ms4000 KiB
19Accepted3/329ms3948 KiB
20Accepted2/230ms3884 KiB
21Accepted2/230ms3940 KiB
22Accepted3/332ms3888 KiB
23Accepted2/232ms4536 KiB
24Accepted2/232ms4536 KiB
25Accepted2/232ms4224 KiB
26Accepted2/232ms4224 KiB
27Accepted2/232ms4224 KiB
28Accepted2/230ms4288 KiB
29Accepted2/234ms4300 KiB
30Accepted2/232ms4372 KiB
31Accepted2/234ms4292 KiB
32Accepted2/235ms4608 KiB