Cod sursa(job #3190685)

Utilizator luca224Ion Luca-Mihail luca224 Data 7 ianuarie 2024 20:19:27
Problema Cc Scor 0
Compilator py Status done
Runda Arhiva de probleme Marime 0.56 kb
import numpy as np
from scipy.optimize import linear_sum_assignment

def solve(input_filename, output_filename):
    with open(input_filename, 'r') as file:
        N = int(file.readline().strip())
        distances = [list(map(int, file.readline().split())) for _ in range(N)]

    cost_matrix = np.array(distances)
    row_ind, col_ind = linear_sum_assignment(cost_matrix)

    total_distance = cost_matrix[row_ind, col_ind].sum()

    with open(output_filename, 'w') as output_file:
        output_file.write(str(total_distance))

solve('cc.in', 'cc.out')