253922026-02-19 18:47:27Rudika11Túra (75 pont)cpp17Wrong answer 4/75116ms6452 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" << 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" << i << " " << y << " " << z << "\n";
                            break;
                        }
                    } if(b == true) break; 
                }
            }
            if(b == true) break;
        } if(b == false) cout << "IGEN" << "\n";
    }
}
SubtaskSumTestVerdictTimeMemory
base4/75
1Wrong answer0/01ms316 KiB
2Wrong answer0/01ms316 KiB
3Wrong answer0/61ms512 KiB
4Wrong answer0/38ms316 KiB
5Wrong answer0/38ms316 KiB
6Wrong answer0/48ms452 KiB
7Wrong answer0/49ms316 KiB
8Wrong answer0/48ms316 KiB
9Wrong answer0/290ms1144 KiB
10Wrong answer0/292ms1068 KiB
11Wrong answer0/292ms828 KiB
12Wrong answer0/292ms828 KiB
13Wrong answer0/292ms828 KiB
14Wrong answer0/392ms820 KiB
15Wrong answer0/292ms820 KiB
16Wrong answer0/292ms956 KiB
17Wrong answer0/296ms824 KiB
18Wrong answer0/290ms1024 KiB
19Wrong answer0/389ms824 KiB
20Wrong answer0/290ms828 KiB
21Wrong answer0/290ms828 KiB
22Wrong answer0/393ms828 KiB
23Wrong answer0/2116ms6452 KiB
24Wrong answer0/2108ms6112 KiB
25Wrong answer0/2105ms3292 KiB
26Wrong answer0/2101ms3480 KiB
27Wrong answer0/2105ms3280 KiB
28Wrong answer0/2101ms3544 KiB
29Wrong answer0/2107ms3280 KiB
30Wrong answer0/2105ms3536 KiB
31Accepted2/2107ms3380 KiB
32Accepted2/2109ms3276 KiB