9346 | 2024. 02. 20 17:30:19 | FarkasZénó | Túra (75 pont) | pypy3 | Futási hiba 71/75 | 328ms | 169012 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 | 71/75 | ||||||
1 | Elfogadva | 0/0 | 52ms | 81840 KiB | |||
2 | Elfogadva | 0/0 | 54ms | 85552 KiB | |||
3 | Elfogadva | 6/6 | 46ms | 83504 KiB | |||
4 | Elfogadva | 3/3 | 93ms | 97244 KiB | |||
5 | Elfogadva | 3/3 | 87ms | 97388 KiB | |||
6 | Elfogadva | 4/4 | 83ms | 97808 KiB | |||
7 | Elfogadva | 4/4 | 83ms | 97840 KiB | |||
8 | Elfogadva | 4/4 | 108ms | 98320 KiB | |||
9 | Elfogadva | 2/2 | 238ms | 154348 KiB | |||
10 | Elfogadva | 2/2 | 233ms | 156816 KiB | |||
11 | Elfogadva | 2/2 | 254ms | 156132 KiB | |||
12 | Elfogadva | 2/2 | 246ms | 155792 KiB | |||
13 | Elfogadva | 2/2 | 250ms | 157608 KiB | |||
14 | Elfogadva | 3/3 | 246ms | 156728 KiB | |||
15 | Elfogadva | 2/2 | 252ms | 156524 KiB | |||
16 | Elfogadva | 2/2 | 263ms | 154948 KiB | |||
17 | Elfogadva | 2/2 | 254ms | 158448 KiB | |||
18 | Elfogadva | 2/2 | 234ms | 158744 KiB | |||
19 | Elfogadva | 3/3 | 231ms | 159904 KiB | |||
20 | Elfogadva | 2/2 | 229ms | 160924 KiB | |||
21 | Elfogadva | 2/2 | 237ms | 162652 KiB | |||
22 | Elfogadva | 3/3 | 261ms | 159996 KiB | |||
23 | Elfogadva | 2/2 | 225ms | 165780 KiB | |||
24 | Elfogadva | 2/2 | 222ms | 162884 KiB | |||
25 | Elfogadva | 2/2 | 266ms | 163944 KiB | |||
26 | Futási hiba | 0/2 | 244ms | 169012 KiB | |||
27 | Elfogadva | 2/2 | 231ms | 163488 KiB | |||
28 | Elfogadva | 2/2 | 224ms | 168132 KiB | |||
29 | Elfogadva | 2/2 | 250ms | 166256 KiB | |||
30 | Futási hiba | 0/2 | 230ms | 168160 KiB | |||
31 | Elfogadva | 2/2 | 328ms | 167948 KiB | |||
32 | Elfogadva | 2/2 | 301ms | 164796 KiB |