7890 2024. 01. 11 18:11:29 Ablablabla Túra (75 pont) cpp17 Hibás válasz 0/75 111ms 18872 KiB
#include <bits/stdc++.h>

using namespace std;

int main()
{
    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 << "NO\n";
                    cout << "2\n";
                    cout << i + 1 << " " << x << "\n";
                    rossz = 1;
                } else{
                    for(int y : csucsok[x]){
                        if(!jo[y] && y != i){
                            cout << "NO\n";
                            cout << "3\n";
                            cout << i + 1 << " " << x << " " << y << "\n";
                            rossz = 1;
                        }

                        if(rossz){
                            break;
                        }
                    }
                }

                if(rossz){
                    break;
                }
            }

            if(rossz){
                break;
            }
        }

        if(!rossz){
            cout << "YES\n";
        }
    }
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 0/75
1 Hibás válasz 0/0 3ms 1996 KiB
2 Hibás válasz 0/0 3ms 2016 KiB
3 Hibás válasz 0/6 3ms 2260 KiB
4 Hibás válasz 0/3 10ms 2660 KiB
5 Hibás válasz 0/3 10ms 2920 KiB
6 Hibás válasz 0/4 10ms 3132 KiB
7 Hibás válasz 0/4 10ms 3312 KiB
8 Hibás válasz 0/4 10ms 3620 KiB
9 Hibás válasz 0/2 90ms 6120 KiB
10 Hibás válasz 0/2 92ms 6924 KiB
11 Hibás válasz 0/2 92ms 7780 KiB
12 Hibás válasz 0/2 90ms 7848 KiB
13 Hibás válasz 0/2 92ms 8088 KiB
14 Hibás válasz 0/3 90ms 8092 KiB
15 Hibás válasz 0/2 90ms 8352 KiB
16 Hibás válasz 0/2 92ms 8300 KiB
17 Hibás válasz 0/2 90ms 8384 KiB
18 Hibás válasz 0/2 89ms 8652 KiB
19 Hibás válasz 0/3 87ms 8552 KiB
20 Hibás válasz 0/2 87ms 8516 KiB
21 Hibás válasz 0/2 90ms 8552 KiB
22 Hibás válasz 0/3 93ms 8540 KiB
23 Hibás válasz 0/2 100ms 18872 KiB
24 Hibás válasz 0/2 104ms 18140 KiB
25 Hibás válasz 0/2 103ms 12884 KiB
26 Hibás válasz 0/2 98ms 13340 KiB
27 Hibás válasz 0/2 101ms 12808 KiB
28 Hibás válasz 0/2 97ms 13440 KiB
29 Hibás válasz 0/2 104ms 13132 KiB
30 Hibás válasz 0/2 98ms 13592 KiB
31 Hibás válasz 0/2 104ms 13520 KiB
32 Hibás válasz 0/2 111ms 13432 KiB