9346 2024. 02. 20 17:30:19 FarkasZénó Túra (75 pont) pypy3 Futási hiba 71/75 328ms 169012 KiB
from sys import stdin

def csunyakereso(graf, csunyak, indulas):
	for szomszed in graf[indulas]:
		ut = (indulas, szomszed)
		if szomszed in csunyak:
			print('NEM')
			print(2)
			print(indulas, szomszed)
			return True
		for masodik_szomszed in graf[szomszed]:
			if masodik_szomszed == indulas:
				continue
			if masodik_szomszed in csunyak:
				print('NEM')
				print(3)
				print(indulas, szomszed, masodik_szomszed)
				return True
	return False

T = int(stdin.readline())
for _ in range(T):
	N = int(stdin.readline())
	utak = [set() for _ in range(N+1)]
	sz = [int(x) for x in stdin.readline().split()]
	csunyak = set(i+1 for i, x in enumerate(sz) if x == -1)
	if len(csunyak) < 2:
		for _ in range(N-1):
			stdin.readline()
		print('IGEN')
		continue
	for _ in range(N-1):
		honnan, hova = [int(x) for x in stdin.readline().split()]
		utak[honnan].add(hova)
		utak[hova].add(honnan)
	for indulas in csunyak:
		if csunyakereso(utak, csunyak, indulas):
			break
	else:
		print('IGEN')
	# print(sz)
	# print(csunyak)
	# print(utak)
Részfeladat Összpont Teszt Verdikt Idő Memória
base 71/75
1 Elfogadva 0/0 52ms 81840 KiB
2 Elfogadva 0/0 54ms 85552 KiB
3 Elfogadva 6/6 46ms 83504 KiB
4 Elfogadva 3/3 93ms 97244 KiB
5 Elfogadva 3/3 87ms 97388 KiB
6 Elfogadva 4/4 83ms 97808 KiB
7 Elfogadva 4/4 83ms 97840 KiB
8 Elfogadva 4/4 108ms 98320 KiB
9 Elfogadva 2/2 238ms 154348 KiB
10 Elfogadva 2/2 233ms 156816 KiB
11 Elfogadva 2/2 254ms 156132 KiB
12 Elfogadva 2/2 246ms 155792 KiB
13 Elfogadva 2/2 250ms 157608 KiB
14 Elfogadva 3/3 246ms 156728 KiB
15 Elfogadva 2/2 252ms 156524 KiB
16 Elfogadva 2/2 263ms 154948 KiB
17 Elfogadva 2/2 254ms 158448 KiB
18 Elfogadva 2/2 234ms 158744 KiB
19 Elfogadva 3/3 231ms 159904 KiB
20 Elfogadva 2/2 229ms 160924 KiB
21 Elfogadva 2/2 237ms 162652 KiB
22 Elfogadva 3/3 261ms 159996 KiB
23 Elfogadva 2/2 225ms 165780 KiB
24 Elfogadva 2/2 222ms 162884 KiB
25 Elfogadva 2/2 266ms 163944 KiB
26 Futási hiba 0/2 244ms 169012 KiB
27 Elfogadva 2/2 231ms 163488 KiB
28 Elfogadva 2/2 224ms 168132 KiB
29 Elfogadva 2/2 250ms 166256 KiB
30 Futási hiba 0/2 230ms 168160 KiB
31 Elfogadva 2/2 328ms 167948 KiB
32 Elfogadva 2/2 301ms 164796 KiB