CSES - Aalto Competitive Programming 2024 - wk7 Homework - Results
Submission details
Task:Download Speed
Sender:esya_rae
Submission time:2024-10-23 22:34:34 +0300
Language:Python3 (PyPy3)
Status:READY
Result:ACCEPTED
Test results
testverdicttime
#1ACCEPTED0.04 sdetails
#2ACCEPTED0.04 sdetails
#3ACCEPTED0.04 sdetails
#4ACCEPTED0.04 sdetails
#5ACCEPTED0.06 sdetails
#6ACCEPTED0.09 sdetails
#7ACCEPTED0.15 sdetails
#8ACCEPTED0.04 sdetails
#9ACCEPTED0.04 sdetails
#10ACCEPTED0.04 sdetails
#11ACCEPTED0.06 sdetails
#12ACCEPTED0.04 sdetails

Code

import math
import sys
sys.setrecursionlimit(10**8)

def find_path(s, f, visited=None, path=None):
    global g, n
    if visited is None:
        visited = [False] * n
    if path is None:
        path = []

    path.append(s)
    visited[s] = True

    if s == f:
        return path

    for i, w in enumerate(g[s]):
        if not visited[i] and w > 0:
            res_path = find_path(i, f, visited=visited, path=path)
            if res_path:
                return res_path

    path.pop()
    return None

def ff(s, f):
    global g, n, paths
    res = 0

    path = find_path(s, f)
    while path:
        # print(*path)
        # print(g)
        min_flow = math.inf
        for i in range(len(path) - 1):
            a, b = path[i], path[i + 1]
            min_flow = min(min_flow, g[a][b])

        for i in range(len(path) - 1):
            a, b = path[i], path[i + 1]
            g[a][b] -= min_flow
            g[b][a] += min_flow

        res += min_flow

        path = find_path(s, f)

    return res


n, m = map(int, input().split())
g = [[0 for _ in range(n)] for _ in range(n)]
paths = []
for i in range(m):
    a, b, c = map(int, input().split())
    g[a - 1][b - 1] += c
res = ff(0, n - 1)
print(res)



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