7141 2023. 12. 31 13:00:51 xxx Túra (75 pont) cpp17 Elfogadva 75/75 56ms 16520 KiB
#include <bits/stdc++.h>
using namespace std;

void solve() {
    int n;
    cin >> n;
    vector<int> t(n+1);
    for(int i = 1; i <= n; i++) {
        cin >> t[i];
    }

    vector<vector<int> > adj(n+1);

    for(int i = 1; i < n; i++) {
        int x, y;
        cin >> x >> y;
        adj[x].push_back(y);
        adj[y].push_back(x);
    }

    for(int i = 1; i <= n; i++) {
        if (t[i] == -1) {
            for(auto u : adj[i]) {
                if (t[u] == -1) {
                    cout << "NEM\n";
                    cout << 2 << '\n' << i << ' ' << u << '\n';
                    return;
                } else {
                    for(auto v : adj[u]) {
                        if (t[v] == -1 && v != i) {
                            cout << "NEM\n";
                            cout << 3 << '\n' << i << ' ' << u << ' ' << v << '\n';
                            return;
                        }
                    }
                }
            }
        }
    }
    cout << "IGEN\n";

}

int main() {
	ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int tt;
	cin >> tt;
	while(tt--) {
        solve();
	}


	return 0;
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 75/75
1 Elfogadva 0/0 3ms 2108 KiB
2 Elfogadva 0/0 3ms 2244 KiB
3 Elfogadva 6/6 3ms 2280 KiB
4 Elfogadva 3/3 7ms 2556 KiB
5 Elfogadva 3/3 7ms 2840 KiB
6 Elfogadva 4/4 7ms 2872 KiB
7 Elfogadva 4/4 7ms 3044 KiB
8 Elfogadva 4/4 7ms 3092 KiB
9 Elfogadva 2/2 43ms 4140 KiB
10 Elfogadva 2/2 45ms 4020 KiB
11 Elfogadva 2/2 45ms 4272 KiB
12 Elfogadva 2/2 45ms 4292 KiB
13 Elfogadva 2/2 43ms 4220 KiB
14 Elfogadva 3/3 43ms 4480 KiB
15 Elfogadva 2/2 45ms 4504 KiB
16 Elfogadva 2/2 43ms 4692 KiB
17 Elfogadva 2/2 45ms 4972 KiB
18 Elfogadva 2/2 43ms 4984 KiB
19 Elfogadva 3/3 43ms 5104 KiB
20 Elfogadva 2/2 41ms 5232 KiB
21 Elfogadva 2/2 43ms 5336 KiB
22 Elfogadva 3/3 46ms 5244 KiB
23 Elfogadva 2/2 48ms 16520 KiB
24 Elfogadva 2/2 50ms 15868 KiB
25 Elfogadva 2/2 50ms 10176 KiB
26 Elfogadva 2/2 46ms 10772 KiB
27 Elfogadva 2/2 50ms 10264 KiB
28 Elfogadva 2/2 46ms 10892 KiB
29 Elfogadva 2/2 52ms 10264 KiB
30 Elfogadva 2/2 46ms 11000 KiB
31 Elfogadva 2/2 50ms 10776 KiB
32 Elfogadva 2/2 56ms 10472 KiB