9345 2024. 02. 20 17:29:03 FarkasZénó Túra (75 pont) python3 Időlimit túllépés 69/75 465ms 77772 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 69/75
1 Elfogadva 0/0 17ms 11296 KiB
2 Elfogadva 0/0 18ms 11928 KiB
3 Elfogadva 6/6 17ms 12256 KiB
4 Elfogadva 3/3 41ms 13720 KiB
5 Elfogadva 3/3 41ms 13980 KiB
6 Elfogadva 4/4 41ms 14224 KiB
7 Elfogadva 4/4 41ms 14308 KiB
8 Elfogadva 4/4 39ms 14708 KiB
9 Elfogadva 2/2 239ms 25472 KiB
10 Elfogadva 2/2 250ms 26140 KiB
11 Elfogadva 2/2 250ms 26232 KiB
12 Elfogadva 2/2 246ms 26480 KiB
13 Elfogadva 2/2 246ms 26252 KiB
14 Elfogadva 3/3 250ms 26328 KiB
15 Elfogadva 2/2 250ms 26652 KiB
16 Elfogadva 2/2 250ms 26476 KiB
17 Elfogadva 2/2 256ms 26764 KiB
18 Elfogadva 2/2 247ms 27116 KiB
19 Elfogadva 3/3 248ms 26828 KiB
20 Elfogadva 2/2 246ms 27184 KiB
21 Elfogadva 2/2 266ms 27360 KiB
22 Elfogadva 3/3 261ms 27384 KiB
23 Időlimit túllépés 0/2 465ms 43524 KiB
24 Elfogadva 2/2 389ms 77772 KiB
25 Elfogadva 2/2 386ms 69284 KiB
26 Elfogadva 2/2 382ms 73960 KiB
27 Elfogadva 2/2 326ms 68960 KiB
28 Elfogadva 2/2 375ms 73648 KiB
29 Elfogadva 2/2 389ms 70696 KiB
30 Elfogadva 2/2 389ms 74116 KiB
31 Időlimit túllépés 0/2 437ms 75536 KiB
32 Időlimit túllépés 0/2 446ms 70748 KiB