147462025-01-31 17:02:40miszorimarciTĂşra (75 pont)cpp17Accepted 75/7585ms2100 KiB
#include <bits/stdc++.h>
using namespace std;

void solve() {
    int N;
    cin >> N;
    vector<bool> ocsortány(N + 1);
    vector<int> ocsortányszomszed(N + 1);
    for (int i = 1; i <= N; i++) {
        int A;
        cin >> A;
        ocsortány[i] = A == -1;
    }
    bool b = true;
    for (int i = 1; i < N; i++) {
        int U, V;
        cin >> U >> V;

        if (b) {
            b = 0;
            if (ocsortány[U] && ocsortány[V]) {
                cout << "NEM\n2\n" << U << " " << V << "\n";
            }
            else if (ocsortányszomszed[U] && ocsortány[V]) {
                cout << "NEM\n3\n" << ocsortányszomszed[U] << " " << U << " " << V << "\n";
            }
            else if (ocsortány[U] && ocsortányszomszed[V]) {
                cout << "NEM\n3\n" << ocsortányszomszed[V] << " " << V << " " << U << "\n";
            }
            else b = 1;
        }
        if (ocsortány[U]) {
            ocsortányszomszed[V] = U;
        }
        if (ocsortány[V]) {
            ocsortányszomszed[U] = V;
        }
    }
    if (b) cout << "IGEN\n";
}

int main() {
    int T;
    cin >> T;
    while (T--) solve();
}
SubtaskSumTestVerdictTimeMemory
base75/75
1Accepted0/01ms500 KiB
2Accepted0/01ms316 KiB
3Accepted6/61ms316 KiB
4Accepted3/37ms316 KiB
5Accepted3/37ms316 KiB
6Accepted4/47ms316 KiB
7Accepted4/48ms316 KiB
8Accepted4/48ms316 KiB
9Accepted2/274ms1588 KiB
10Accepted2/274ms1584 KiB
11Accepted2/274ms1600 KiB
12Accepted2/274ms1580 KiB
13Accepted2/274ms1588 KiB
14Accepted3/374ms1780 KiB
15Accepted2/274ms1588 KiB
16Accepted2/274ms1580 KiB
17Accepted2/275ms1592 KiB
18Accepted2/274ms1600 KiB
19Accepted3/371ms1588 KiB
20Accepted2/272ms1632 KiB
21Accepted2/275ms1580 KiB
22Accepted3/376ms1596 KiB
23Accepted2/282ms2100 KiB
24Accepted2/282ms2100 KiB
25Accepted2/281ms1928 KiB
26Accepted2/281ms2084 KiB
27Accepted2/282ms1888 KiB
28Accepted2/282ms1788 KiB
29Accepted2/282ms1924 KiB
30Accepted2/282ms1796 KiB
31Accepted2/282ms1948 KiB
32Accepted2/285ms1888 KiB