Submission details
Task:Download Speed
Sender:aalto26gh_031
Submission time:2026-10-22 16:12:03 +0300
Language:Python3 (PyPy3)
Status:READY
Result:ACCEPTED
Test results
testverdicttime
#1ACCEPTED0.04 sdetails
#2ACCEPTED0.04 sdetails
#3ACCEPTED0.04 sdetails
#4ACCEPTED0.04 sdetails
#5ACCEPTED0.05 sdetails
#6ACCEPTED0.06 sdetails
#7ACCEPTED0.06 sdetails
#8ACCEPTED0.04 sdetails
#9ACCEPTED0.04 sdetails
#10ACCEPTED0.05 sdetails
#11ACCEPTED0.05 sdetails
#12ACCEPTED0.04 sdetails

Code

import sys 

from collections import deque

inp = sys.stdin.read().split()
n = int(inp[0])
m = int(inp[1])
adj = [[] for _ in range(n + 1)]
capacities  = [[0] * (n + 1) for _ in range(n + 1)]
ind = 2
for i in range(m):
    u = int(inp[ind])
    v = int(inp[ind+1])
    c = int(inp[ind+2]) 
    ind+=3

    if capacities[u][v] == 0 and capacities[v][u] == 0:
            adj[u].append(v)
            adj[v].append(u)
    capacities[u][v]+=c

def bfs(s,t,parent):
    for i in range(1,n+1):
        parent[i] = -1
    parent[s] = -2 #source is considered visited

    queue = deque([(s, float('inf'))])
    while queue:
            u, flow = queue.popleft()
            
            for v in adj[u]:
                # If v is unvisited and there is available capacity
                if parent[v] == -1 and capacities[u][v] > 0:
                    parent[v] = u
                    new_flow = min(flow, capacities[u][v])
                    
                    if v == t:
                        return new_flow
                        
                    queue.append((v, new_flow))
    return 0

max_flow = 0
parent = [-1] * (n + 1)
    
while True:
    flow = bfs(1, n, parent)
    if flow == 0:
        break
    
    max_flow += flow
    
    curr = n
    while curr != 1:
        prev = parent[curr]
        capacities[prev][curr] -= flow
        capacities[curr][prev] += flow 
        curr = prev
        
print(max_flow)

Test details

Test 1

Verdict: ACCEPTED

input
4 3
1 2 5
2 3 3
3 4 6

correct output
3

user output
3

Test 2

Verdict: ACCEPTED

input
4 5
1 2 1
1 3 1
2 3 1
2 4 1
...

correct output
2

user output
2

Test 3

Verdict: ACCEPTED

input
4 5
1 2 1000000000
1 3 1000000000
2 3 1
2 4 1000000000
...

correct output
2000000000

user output
2000000000

Test 4

Verdict: ACCEPTED

input
2 1
2 1 100

correct output
0

user output
0

Test 5

Verdict: ACCEPTED

input
2 1000
1 2 1000000000
1 2 1000000000
1 2 1000000000
1 2 1000000000
...

correct output
1000000000000

user output
1000000000000

Test 6

Verdict: ACCEPTED

input
500 998
1 2 54
1 3 59
1 4 83
2 5 79
...

correct output
60

user output
60

Test 7

Verdict: ACCEPTED

input
500 998
1 2 530873053
1 3 156306296
1 4 478040476
3 5 303609600
...

correct output
1093765123

user output
1093765123

Test 8

Verdict: ACCEPTED

input
2 1
1 2 1

correct output
1

user output
1

Test 9

Verdict: ACCEPTED

input
4 5
1 2 3
2 4 2
1 3 4
3 4 5
...

correct output
6

user output
6

Test 10

Verdict: ACCEPTED

input
4 5
1 2 1
1 3 2
3 2 1
2 4 2
...

correct output
3

user output
3

Test 11

Verdict: ACCEPTED

input
10 999
1 2 1000000000
1 2 1000000000
1 2 1000000000
1 2 1000000000
...

correct output
111000000000

user output
111000000000

Test 12

Verdict: ACCEPTED

input
7 9
1 2 1
1 3 1
1 4 1
2 5 1
...

correct output
2

user output
2