9211 2024. 02. 18 16:37:57 FarkasZénó Túra (75 pont) python3 Időlimit túllépés 67/75 460ms 88140 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 67/75
1 Elfogadva 0/0 20ms 12116 KiB
2 Elfogadva 0/0 20ms 12764 KiB
3 Elfogadva 6/6 19ms 12524 KiB
4 Elfogadva 3/3 45ms 13864 KiB
5 Elfogadva 3/3 45ms 13880 KiB
6 Elfogadva 4/4 43ms 14164 KiB
7 Elfogadva 4/4 45ms 14276 KiB
8 Elfogadva 4/4 43ms 14508 KiB
9 Elfogadva 2/2 266ms 21216 KiB
10 Elfogadva 2/2 270ms 21488 KiB
11 Elfogadva 2/2 270ms 21596 KiB
12 Elfogadva 2/2 272ms 22012 KiB
13 Elfogadva 2/2 272ms 22232 KiB
14 Elfogadva 3/3 263ms 22008 KiB
15 Elfogadva 2/2 279ms 22380 KiB
16 Elfogadva 2/2 270ms 22676 KiB
17 Elfogadva 2/2 270ms 22476 KiB
18 Elfogadva 2/2 272ms 22276 KiB
19 Elfogadva 3/3 270ms 22212 KiB
20 Elfogadva 2/2 264ms 22384 KiB
21 Elfogadva 2/2 273ms 22860 KiB
22 Elfogadva 3/3 280ms 23084 KiB
23 Időlimit túllépés 0/2 460ms 47204 KiB
24 Időlimit túllépés 0/2 423ms 88140 KiB
25 Elfogadva 2/2 335ms 55180 KiB
26 Elfogadva 2/2 331ms 56940 KiB
27 Elfogadva 2/2 365ms 54672 KiB
28 Elfogadva 2/2 324ms 56912 KiB
29 Elfogadva 2/2 338ms 57000 KiB
30 Elfogadva 2/2 358ms 57008 KiB
31 Időlimit túllépés 0/2 409ms 58836 KiB
32 Időlimit túllépés 0/2 435ms 57300 KiB