183492025-10-20 20:35:40birozsTúra (75 pont)pypy3Accepted 75/75344ms47704 KiB
E = int(input())
mo = []
for i in range(E):
    rossz = []
    N = int(input())
    T = list(map(int,input().split()))
    D = {}
    csunya = False
    for _ in range(N-1):
        A,B = map(int,input().split())
        if not csunya:
            if A not in D:
                D[A] = []
            if B not in D:
                D[B] = []
            if T[B-1] == -1:
                D[A].append(B)
            if T[A-1] == -1:
                D[B].append(A)
            if T[A-1] == -1 and len(D[A]) > 0:
                rossz = [A,D[A][0]]
                csunya = True
            elif T[A-1] == 1 and len(D[A]) == 2:
                rossz = [D[A][0],A,D[A][1]]
                csunya = True   
            elif T[B-1] == -1 and len(D[B]) > 0:
                rossz = [B,D[B][0]]
                csunya = True
            elif T[B-1] == 1 and len(D[B]) == 2:
                rossz = [D[B][0],B,D[B][1]]
                csunya = True
    mo.append(rossz)
for eset in mo:
    if len(eset) == 0:
        print("IGEN")
    else:
        print("NEM")
        print(len(eset))
        for k in eset:
            print(k,end=" ")
        print()

        
SubtaskSumTestVerdictTimeMemory
base75/75
1Accepted0/043ms19684 KiB
2Accepted0/054ms21100 KiB
3Accepted6/645ms21196 KiB
4Accepted3/3131ms23764 KiB
5Accepted3/3103ms23268 KiB
6Accepted4/494ms23012 KiB
7Accepted4/4114ms23528 KiB
8Accepted4/4100ms23272 KiB
9Accepted2/2215ms31648 KiB
10Accepted2/2287ms35280 KiB
11Accepted2/2314ms34972 KiB
12Accepted2/2298ms34792 KiB
13Accepted2/2263ms34016 KiB
14Accepted3/3307ms34044 KiB
15Accepted2/2286ms33768 KiB
16Accepted2/2266ms34792 KiB
17Accepted2/2312ms34024 KiB
18Accepted2/2259ms34280 KiB
19Accepted3/3248ms34532 KiB
20Accepted2/2252ms32744 KiB
21Accepted2/2256ms32484 KiB
22Accepted3/3308ms34792 KiB
23Accepted2/2141ms32404 KiB
24Accepted2/2272ms42224 KiB
25Accepted2/2194ms31464 KiB
26Accepted2/2182ms30520 KiB
27Accepted2/2237ms37316 KiB
28Accepted2/2231ms32744 KiB
29Accepted2/2231ms34788 KiB
30Accepted2/2201ms32220 KiB
31Accepted2/2272ms42868 KiB
32Accepted2/2344ms47704 KiB