93462024-02-20 17:30:19FarkasZénóTúra (75 pont)pypy3Futási hiba 71/75328ms169012 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ÖsszpontTesztVerdiktIdőMemória
base71/75
1Elfogadva0/052ms81840 KiB
2Elfogadva0/054ms85552 KiB
3Elfogadva6/646ms83504 KiB
4Elfogadva3/393ms97244 KiB
5Elfogadva3/387ms97388 KiB
6Elfogadva4/483ms97808 KiB
7Elfogadva4/483ms97840 KiB
8Elfogadva4/4108ms98320 KiB
9Elfogadva2/2238ms154348 KiB
10Elfogadva2/2233ms156816 KiB
11Elfogadva2/2254ms156132 KiB
12Elfogadva2/2246ms155792 KiB
13Elfogadva2/2250ms157608 KiB
14Elfogadva3/3246ms156728 KiB
15Elfogadva2/2252ms156524 KiB
16Elfogadva2/2263ms154948 KiB
17Elfogadva2/2254ms158448 KiB
18Elfogadva2/2234ms158744 KiB
19Elfogadva3/3231ms159904 KiB
20Elfogadva2/2229ms160924 KiB
21Elfogadva2/2237ms162652 KiB
22Elfogadva3/3261ms159996 KiB
23Elfogadva2/2225ms165780 KiB
24Elfogadva2/2222ms162884 KiB
25Elfogadva2/2266ms163944 KiB
26Futási hiba0/2244ms169012 KiB
27Elfogadva2/2231ms163488 KiB
28Elfogadva2/2224ms168132 KiB
29Elfogadva2/2250ms166256 KiB
30Futási hiba0/2230ms168160 KiB
31Elfogadva2/2328ms167948 KiB
32Elfogadva2/2301ms164796 KiB