152712025-02-17 19:28:54TakacsAndrasTúra (75 pont)cpp17Accepted 75/75114ms6436 KiB
#include <bits/stdc++.h>
using namespace std;
int main() {
    int t; cin >> t;
    while (t--) {
        int n; cin >> n;
        vector<int> a(n+1);
        for (int i = 1; i <= n; i++) cin >> a[i];
        vector<vector<int>> g(n+1);
        for (int i = 0; i < n-1; i++) {
            int a,b; cin >> a >> b;
            g[a].push_back(b);
            g[b].push_back(a);
        }
        bool f = false;
        for (int i = 1; i <= n; i++) {
            for (int x : g[i]) {
                if (a[i] == -1 && a[x] == -1) {
                    cout << "NEM\n2\n" << i << ' ' << x << '\n';
                    f = true;
                    break;
                }
                else if (g[x].size() >= 2 && a[i] == -1 && a[x] == 1) {
                    for (int y : g[x]) {
                        if (a[y] == -1 && y != i) {
                            cout << "NEM\n3\n" << i << ' ' << x << ' ' << y << '\n';
                            f = true;
                            break;
                        }
                    }
                    if (f) break;
                }
            }
            if (f) break;
        }
        if (!f) cout << "IGEN\n";
    }
}
SubtaskSumTestVerdictTimeMemory
base75/75
1Accepted0/01ms316 KiB
2Accepted0/01ms508 KiB
3Accepted6/61ms316 KiB
4Accepted3/38ms316 KiB
5Accepted3/38ms316 KiB
6Accepted4/48ms316 KiB
7Accepted4/48ms316 KiB
8Accepted4/48ms316 KiB
9Accepted2/290ms828 KiB
10Accepted2/292ms820 KiB
11Accepted2/292ms924 KiB
12Accepted2/290ms1024 KiB
13Accepted2/292ms1000 KiB
14Accepted3/390ms1004 KiB
15Accepted2/292ms828 KiB
16Accepted2/292ms948 KiB
17Accepted2/292ms1012 KiB
18Accepted2/290ms1228 KiB
19Accepted3/389ms828 KiB
20Accepted2/289ms828 KiB
21Accepted2/290ms824 KiB
22Accepted3/393ms828 KiB
23Accepted2/2108ms6436 KiB
24Accepted2/2111ms6116 KiB
25Accepted2/2104ms3284 KiB
26Accepted2/2105ms3580 KiB
27Accepted2/2104ms3276 KiB
28Accepted2/2104ms3512 KiB
29Accepted2/2104ms3280 KiB
30Accepted2/298ms3588 KiB
31Accepted2/2108ms3480 KiB
32Accepted2/2114ms3320 KiB