253942026-02-19 18:53:06Rudika11Túra (75 pont)cpp17Elfogadva 75/75114ms6452 KiB
#include <bits/stdc++.h>
using namespace std;

int main() {
    int t; cin >> t;
    while(t--) {
        int n; cin >> n;
        vector<int> v(n+1);
        for (int i = 1; i <= n; i++) cin >> v[i];
        vector<vector<int>> szomszedsagi(n+1);
        for(int i = 0; i < n - 1; i++) {
            int a, b; cin >> a >> b;
            szomszedsagi[a].push_back(b);
            szomszedsagi[b].push_back(a);
        } bool b = false; 
        for(int i = 1; i <= n; i++) {
            for(int y : szomszedsagi[i]) {
                if(v[y] == -1 && v[i] == -1) {                    
                    b = true;
                    cout << "NEM" << "\n" << 2 << "\n" << i << " " << y << "\n";
                    break;
                } else if(v[i] == -1 && v[y] == 1 && szomszedsagi[y].size() >= 2) {
                    for(int z : szomszedsagi[y]) {
                        if(v[z] == -1 && z != i) {                            
                            b = true;
                            cout << "NEM" << "\n" << 3 << "\n" << i << " " << y << " " << z << "\n";
                            break;
                        }
                    } if(b == true) break; 
                }
            }
            if(b == true) break;
        } if(b == false) cout << "IGEN" << "\n";
    }
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base75/75
1Elfogadva0/01ms316 KiB
2Elfogadva0/01ms316 KiB
3Elfogadva6/61ms316 KiB
4Elfogadva3/38ms316 KiB
5Elfogadva3/38ms316 KiB
6Elfogadva4/48ms316 KiB
7Elfogadva4/48ms316 KiB
8Elfogadva4/48ms316 KiB
9Elfogadva2/292ms880 KiB
10Elfogadva2/292ms828 KiB
11Elfogadva2/292ms828 KiB
12Elfogadva2/292ms948 KiB
13Elfogadva2/292ms844 KiB
14Elfogadva3/392ms828 KiB
15Elfogadva2/293ms1136 KiB
16Elfogadva2/292ms832 KiB
17Elfogadva2/292ms828 KiB
18Elfogadva2/290ms1032 KiB
19Elfogadva3/389ms828 KiB
20Elfogadva2/290ms828 KiB
21Elfogadva2/290ms1040 KiB
22Elfogadva3/393ms1024 KiB
23Elfogadva2/2104ms6452 KiB
24Elfogadva2/2108ms6172 KiB
25Elfogadva2/2107ms3280 KiB
26Elfogadva2/2108ms3552 KiB
27Elfogadva2/2104ms3280 KiB
28Elfogadva2/2109ms3452 KiB
29Elfogadva2/2105ms3276 KiB
30Elfogadva2/2111ms3516 KiB
31Elfogadva2/2104ms3420 KiB
32Elfogadva2/2114ms3276 KiB