Internet connection

"""https://cses.fi/345/task/B"""
from __future__ import print_function
import collections
import sys

def eprint(*args, **kwargs):
    print(*args, file=sys.stderr, **kwargs)

def get_path_with_bfs(adj_matrix, V):
    """DFS uses a stack/list. BFS uses a queue. DFS will time out."""
    root = 1  # Teemu
    target = V  # Taina
    seen = []  # Stuff you've seen.
    to_visit = collections.deque( [root])
    path = []
    while to_visit:
        v = to_visit.popleft()  # Pop last inserted element.
        seen.append(v)
        found = False
        for neighbor in range(1, V+1):
            weight = adj_matrix[v][neighbor]
            if (weight < 1) or (neighbor in seen) or (neighbor in path):
                continue
            # eprint(f"weight from {v} to {neighbor} is {weight}")
            found = True
            if neighbor == target:
                path.append(v)
                path.append(target)
                return path
            to_visit.append(neighbor)
            # eprint('to visit', to_visit)
        if found:
            path.append(v)
            # eprint("FOUND: path is ", path)
    return None  # No path found


def max_rate(adj_matrix, V):
    total_max_rate = 0
    path = get_path_with_bfs(adj_matrix, V)
    while path:
        # eprint("path", path)
        path_weights = []
        for i in range(len(path)-1):
            a, b = path[i], path[i+1]
            weight = adj_matrix[a][b]
            path_weights.append(weight)
        # eprint('weights', path_weights)

        max_rate = min(path_weights)
        total_max_rate += max_rate
        # Deduct the max_rate from all weights
        for i in range(len(path)-1):
            a, b = path[i], path[i+1]
            adj_matrix[a][b] -= max_rate
            adj_matrix[b][a] += max_rate
        path = get_path_with_bfs(adj_matrix, V)
    print(int(total_max_rate))
    return


# Read number oof vertices (computers) and edges (connections).
V, E = [int(x) for x in input().split()]
adj_matrix = [[0 for i in range(V + 1)] for i in range(V + 1)]
for i in range(E):
    a, b, weight = [int(x) for x in input().split()]
    adj_matrix[a][b] = weight
max_rate(adj_matrix, V)

Test 1

Verdict: TIME LIMIT EXCEEDED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: TIME LIMIT EXCEEDED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: TIME LIMIT EXCEEDED

Test 7

Verdict: TIME LIMIT EXCEEDED

Test 8

Verdict: TIME LIMIT EXCEEDED

Test 9

Verdict: TIME LIMIT EXCEEDED

Test 10

Verdict: TIME LIMIT EXCEEDED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED

Test 20

Verdict: ACCEPTED

Test 21

Verdict: ACCEPTED

Test 22

Verdict: ACCEPTED