9212 2024. 02. 18 16:38:39 FarkasZénó Túra (75 pont) python3 Időlimit túllépés 71/75 490ms 58600 KiB
from sys import stdin
from collections import defaultdict

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 = defaultdict(set)
	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 21ms 12184 KiB
2 Elfogadva 0/0 21ms 12476 KiB
3 Elfogadva 6/6 21ms 12464 KiB
4 Elfogadva 3/3 45ms 13496 KiB
5 Elfogadva 3/3 43ms 13924 KiB
6 Elfogadva 4/4 43ms 14096 KiB
7 Elfogadva 4/4 45ms 13952 KiB
8 Elfogadva 4/4 43ms 14388 KiB
9 Elfogadva 2/2 270ms 21540 KiB
10 Elfogadva 2/2 270ms 21752 KiB
11 Elfogadva 2/2 277ms 21644 KiB
12 Elfogadva 2/2 270ms 21640 KiB
13 Elfogadva 2/2 275ms 21752 KiB
14 Elfogadva 3/3 266ms 22080 KiB
15 Elfogadva 2/2 270ms 21776 KiB
16 Elfogadva 2/2 268ms 22072 KiB
17 Elfogadva 2/2 268ms 22268 KiB
18 Elfogadva 2/2 268ms 22312 KiB
19 Elfogadva 3/3 268ms 22352 KiB
20 Elfogadva 2/2 272ms 22268 KiB
21 Elfogadva 2/2 280ms 23328 KiB
22 Elfogadva 3/3 282ms 22636 KiB
23 Időlimit túllépés 0/2 460ms 47160 KiB
24 Időlimit túllépés 0/2 490ms 43460 KiB
25 Elfogadva 2/2 368ms 55072 KiB
26 Elfogadva 2/2 351ms 57232 KiB
27 Elfogadva 2/2 333ms 55332 KiB
28 Elfogadva 2/2 354ms 56636 KiB
29 Elfogadva 2/2 375ms 56868 KiB
30 Elfogadva 2/2 361ms 57008 KiB
31 Elfogadva 2/2 377ms 58600 KiB
32 Elfogadva 2/2 398ms 57576 KiB