152712025-02-17 19:28:54TakacsAndrasTúra (75 pont)cpp17Elfogadva 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";
    }
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base75/75
1Elfogadva0/01ms316 KiB
2Elfogadva0/01ms508 KiB
3Elfogadva6/61ms316 KiB
4Elfogadva3/38ms316 KiB
5Elfogadva3/38ms316 KiB
6Elfogadva4/48ms316 KiB
7Elfogadva4/48ms316 KiB
8Elfogadva4/48ms316 KiB
9Elfogadva2/290ms828 KiB
10Elfogadva2/292ms820 KiB
11Elfogadva2/292ms924 KiB
12Elfogadva2/290ms1024 KiB
13Elfogadva2/292ms1000 KiB
14Elfogadva3/390ms1004 KiB
15Elfogadva2/292ms828 KiB
16Elfogadva2/292ms948 KiB
17Elfogadva2/292ms1012 KiB
18Elfogadva2/290ms1228 KiB
19Elfogadva3/389ms828 KiB
20Elfogadva2/289ms828 KiB
21Elfogadva2/290ms824 KiB
22Elfogadva3/393ms828 KiB
23Elfogadva2/2108ms6436 KiB
24Elfogadva2/2111ms6116 KiB
25Elfogadva2/2104ms3284 KiB
26Elfogadva2/2105ms3580 KiB
27Elfogadva2/2104ms3276 KiB
28Elfogadva2/2104ms3512 KiB
29Elfogadva2/2104ms3280 KiB
30Elfogadva2/298ms3588 KiB
31Elfogadva2/2108ms3480 KiB
32Elfogadva2/2114ms3320 KiB