9209 2024. 02. 18 16:24:33 FarkasZénó Túra (75 pont) python3 Időlimit túllépés 67/75 465ms 101552 KiB
from sys import stdin
from collections import defaultdict

def csunyakereso(graf, csunyak, eddigi_ut):
	mehet_tovabb = len(eddigi_ut) < 2
	for kovetkezo in graf[eddigi_ut[-1]]:
		if kovetkezo in eddigi_ut:
			continue
		if kovetkezo in csunyak:
			print('NEM')
			print(len(eddigi_ut)+1)
			print(' '.join(str(x) for x in eddigi_ut + (kovetkezo,)))
			return True
		if mehet_tovabb:
			if csunyakereso(graf, csunyak, eddigi_ut + (kovetkezo,)):
				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)
	for _ in range(N-1):
		honnan, hova = [int(x) for x in stdin.readline().split()]
		utak[honnan].add(hova)
		utak[hova].add(honnan)
	for cs in csunyak:
		if csunyakereso(utak, csunyak, (cs,)):
			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 11856 KiB
2 Elfogadva 0/0 21ms 12456 KiB
3 Elfogadva 6/6 19ms 12300 KiB
4 Elfogadva 3/3 45ms 13812 KiB
5 Elfogadva 3/3 45ms 13892 KiB
6 Elfogadva 4/4 46ms 14396 KiB
7 Elfogadva 4/4 43ms 14520 KiB
8 Elfogadva 4/4 45ms 15016 KiB
9 Elfogadva 2/2 264ms 22220 KiB
10 Elfogadva 2/2 277ms 23124 KiB
11 Elfogadva 2/2 270ms 23740 KiB
12 Elfogadva 2/2 275ms 24332 KiB
13 Elfogadva 2/2 279ms 25300 KiB
14 Elfogadva 3/3 266ms 26952 KiB
15 Elfogadva 2/2 275ms 27656 KiB
16 Elfogadva 2/2 284ms 28396 KiB
17 Elfogadva 2/2 268ms 29844 KiB
18 Elfogadva 2/2 268ms 31332 KiB
19 Elfogadva 3/3 264ms 31728 KiB
20 Elfogadva 2/2 266ms 32568 KiB
21 Elfogadva 2/2 289ms 33872 KiB
22 Elfogadva 3/3 284ms 35064 KiB
23 Időlimit túllépés 0/2 465ms 59460 KiB
24 Időlimit túllépés 0/2 423ms 101552 KiB
25 Elfogadva 2/2 333ms 70196 KiB
26 Elfogadva 2/2 323ms 73152 KiB
27 Elfogadva 2/2 328ms 72048 KiB
28 Elfogadva 2/2 326ms 75208 KiB
29 Elfogadva 2/2 365ms 76584 KiB
30 Elfogadva 2/2 363ms 77660 KiB
31 Időlimit túllépés 0/2 432ms 80872 KiB
32 Időlimit túllépés 0/2 425ms 80320 KiB