Company Queries II

from collections import deque


def bfs(s, t, parent, capacity, adj):
    parent[:] = [-1] * len(parent)
    parent[s] = -2
    q = deque([(s, float('inf'))])

    while q:
        cur, flow = q.popleft()
        for nxt in adj[cur]:
            if parent[nxt] == -1 and capacity[cur][nxt]:
                parent[nxt] = cur
                new_flow = min(flow, capacity[cur][nxt])
                if nxt == t:
                    return new_flow
                q.append((nxt, new_flow))
    return 0


n, m = map(int, input().split())
capacity = [[0] * (n + 1) for _ in range(n + 1)]
adj = [[] for _ in range(n + 1)]

for _ in range(m):
    a, b, c = map(int, input().split())
    capacity[a][b] += c
    adj[a].append(b)
    adj[b].append(a)

flow = 0
parent = [-1] * (n + 1)

while True:
    new_flow = bfs(1, n, parent, capacity, adj)
    if new_flow == 0:
        break
    flow += new_flow
    cur = n
    while cur != 1:
        prev = parent[cur]
        capacity[prev][cur] -= new_flow
        capacity[cur][prev] += new_flow
        cur = prev

print(flow)

Test 1

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 26, in <module>
    a, b, c = map(int, input().split())
ValueError: too many values to unpack (expected 3)

Test 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 26, in <module>
    a, b, c = map(int, input().split())
ValueError: too many values to unpack (expected 3)

Test 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 26, in <module>
    a, b, c = map(int, input().split())
ValueError: too many values to unpack (expected 3)

Test 4

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 26, in <module>
    a, b, c = map(int, input().split())
ValueError: too many values to unpack (expected 3)

Test 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 26, in <module>
    a, b, c = map(int, input().split())
ValueError: too many values to unpack (expected 3)

Test 6

Verdict: RUNTIME ERROR

Test 7

Verdict: RUNTIME ERROR

Test 8

Verdict: RUNTIME ERROR

Test 9

Verdict: RUNTIME ERROR

Test 10

Verdict: RUNTIME ERROR

Test 11

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 26, in <module>
    a, b, c = map(int, input().split())
ValueError: not enough values to unpack (expected 3, got 1)

Test 12

Verdict: RUNTIME ERROR