7892 2024. 01. 11 18:15:36 Ablablabla Túra (75 pont) cpp17 Elfogadva 75/75 64ms 16348 KiB
#include <bits/stdc++.h>

using namespace std;

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);

    int t;
    cin >> t;

    while(t--){
        int n;
        cin >> n;

        vector<bool> jo(n, 0);
        for(int i = 0; i < n; i++){
            int a;
            cin >> a;

            jo[i] = (a == 1);
        }

        vector<vector<int>> csucsok(n, vector<int>());
        for(int i = 0; i < n - 1; i++){
            int a, b;
            cin >> a >> b;
            a--; b--;

            csucsok[a].push_back(b);
            csucsok[b].push_back(a);
        }

        bool rossz = 0;
        for(int i = 0; i < n; i++){
            if(jo[i]) continue;

            for(int x : csucsok[i]){
                if(!jo[x]){
                    cout << "NEM\n";
                    cout << "2\n";
                    cout << i + 1 << " " << x + 1 << "\n";
                    rossz = 1;
                } else{
                    for(int y : csucsok[x]){
                        if(!jo[y] && y != i){
                            cout << "NEM\n";
                            cout << "3\n";
                            cout << i + 1 << " " << x + 1 << " " << y + 1 << "\n";
                            rossz = 1;
                        }

                        if(rossz){
                            break;
                        }
                    }
                }

                if(rossz){
                    break;
                }
            }

            if(rossz){
                break;
            }
        }

        if(!rossz){
            cout << "IGEN\n";
        }
    }
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 75/75
1 Elfogadva 0/0 3ms 1828 KiB
2 Elfogadva 0/0 3ms 2064 KiB
3 Elfogadva 6/6 3ms 2232 KiB
4 Elfogadva 3/3 6ms 2508 KiB
5 Elfogadva 3/3 6ms 2732 KiB
6 Elfogadva 4/4 6ms 2688 KiB
7 Elfogadva 4/4 7ms 2944 KiB
8 Elfogadva 4/4 7ms 3152 KiB
9 Elfogadva 2/2 43ms 4388 KiB
10 Elfogadva 2/2 45ms 4448 KiB
11 Elfogadva 2/2 43ms 4704 KiB
12 Elfogadva 2/2 43ms 4656 KiB
13 Elfogadva 2/2 43ms 4976 KiB
14 Elfogadva 3/3 43ms 4968 KiB
15 Elfogadva 2/2 45ms 5020 KiB
16 Elfogadva 2/2 43ms 5080 KiB
17 Elfogadva 2/2 43ms 5292 KiB
18 Elfogadva 2/2 41ms 5484 KiB
19 Elfogadva 3/3 41ms 5436 KiB
20 Elfogadva 2/2 41ms 5440 KiB
21 Elfogadva 2/2 43ms 5432 KiB
22 Elfogadva 3/3 46ms 5400 KiB
23 Elfogadva 2/2 48ms 16348 KiB
24 Elfogadva 2/2 64ms 15296 KiB
25 Elfogadva 2/2 48ms 10076 KiB
26 Elfogadva 2/2 46ms 10704 KiB
27 Elfogadva 2/2 50ms 10476 KiB
28 Elfogadva 2/2 46ms 10796 KiB
29 Elfogadva 2/2 50ms 10288 KiB
30 Elfogadva 2/2 46ms 10796 KiB
31 Elfogadva 2/2 52ms 10624 KiB
32 Elfogadva 2/2 54ms 10284 KiB