94962024-02-22 12:01:22Leventusz09Túra (75 pont)python3Futási hiba 0/75465ms66148 KiB
def dfs(node, parent):
    global ans, beauty, visited
    visited[node] = True
    for child in graph[node]:
        if child != parent:
            if visited[child]:
                ans = 1
                return
            beauty += values[child]
            dfs(child, node)
            if ans:
                return
            beauty -= values[child]


t = int(input())
for _ in range(t):
    n = int(input())
    values = list(map(int, input().split()))
    graph = {i+1: [] for i in range(n)}
    for _ in range(n-1):
        a, b = map(int, input().split())
        graph[a].append(b)
        graph[b].append(a)

    visited = [False] * (n+1)
    ans = 0
    for i in range(1, n+1):
        if not visited[i]:
            beauty = values[i-1]
            dfs(i, 0)
            if ans:
                break

    if ans:
        print("NEM")
        bad_tour = []
        visited = [False] * (n+1)
        for i in range(1, n+1):
            if visited[i]:
                break
            visited[i] = True
            if values[i-1] == -1:
                bad_tour.append(i)
            for child in graph[i]:
                visited[child] = True
                if values[child-1] == -1:
                    bad_tour.append(child)
            if sum(values[i-1] for i in bad_tour) < 0:
                break
        print(len(bad_tour))
        print(*bad_tour)
    else:
        print("IGEN")
RészfeladatÖsszpontTesztVerdiktIdőMemória
base0/75
1Futási hiba0/019ms11640 KiB
2Futási hiba0/018ms11844 KiB
3Futási hiba0/618ms12076 KiB
4Futási hiba0/321ms12828 KiB
5Futási hiba0/321ms12988 KiB
6Futási hiba0/421ms13016 KiB
7Futási hiba0/421ms13364 KiB
8Futási hiba0/420ms13124 KiB
9Futási hiba0/254ms17504 KiB
10Futási hiba0/256ms18048 KiB
11Futási hiba0/254ms17832 KiB
12Futási hiba0/252ms18260 KiB
13Futási hiba0/256ms18320 KiB
14Futási hiba0/356ms18276 KiB
15Futási hiba0/254ms18500 KiB
16Futási hiba0/252ms18580 KiB
17Futási hiba0/252ms18612 KiB
18Futási hiba0/254ms18592 KiB
19Futási hiba0/356ms18596 KiB
20Futási hiba0/252ms18916 KiB
21Futási hiba0/250ms18768 KiB
22Futási hiba0/352ms18940 KiB
23Időlimit túllépés0/2465ms33068 KiB
24Időlimit túllépés0/2456ms66148 KiB
25Futási hiba0/2230ms41076 KiB
26Futási hiba0/2206ms40172 KiB
27Futási hiba0/2209ms40676 KiB
28Futási hiba0/2215ms40488 KiB
29Futási hiba0/2194ms40412 KiB
30Futási hiba0/2219ms40176 KiB
31Futási hiba0/2231ms40808 KiB
32Futási hiba0/2228ms40156 KiB