結果

問題 No.519 アイドルユニット
コンテスト
ユーザー rin204
提出日時 2022-11-08 18:25:06
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,472 bytes
コンパイル時間 245 ms
コンパイル使用メモリ 82,196 KB
実行使用メモリ 65,256 KB
最終ジャッジ日時 2024-07-22 02:33:39
合計ジャッジ時間 2,962 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 27 WA * 7
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def hungarian(A):
    inf = 1 << 40
    n = len(A) + 1
    m = len(A[0]) + 1
    P = [0] * m
    way = [0] * m
    U = [0] * n
    V = [0] * m
    for i in range(1, n):
        P[0] = i
        minV = [inf] * m
        used = [False] * m
        j0 = 0
        while P[j0] != 0:
            i0 = P[j0]
            j1 = 0
            used[j0] = True
            delta = inf
            for j in range(1, m):
                if used[j]:
                    continue
                if i0 == 0 or j == 0:
                    cur =  -U[i0] - V[j]
                else:
                    cur = A[i0 - 1][j - 1] - U[i0] - V[j]
                if cur < minV[j]:
                    minV[j] = cur
                    way[j] = j0
                if minV[j] < delta:
                    delta = minV[j]
                    j1 = j
            for j in range(m):
                if used[j]:
                    U[P[j]] += delta
                    V[j] -= delta
                else:
                    minV[j] -= delta
            j0 = j1
        P[j0] = P[way[j0]]
        j0 = way[j0]
        while j0 != 0:
            P[j0] = P[way[j0]]
            j0 = way[j0]
    
    ret = [-1] * (n - 1)
    for i in range(1, m):
        if P[i] != 0:
            ret[P[i] - 1] = i - 1
    return -V[0], ret

n = int(input())
F = [list(map(int, input().split())) for _ in range(n)]
for i in range(n):
    for j in range(n):
        F[i][j] *= -1

ans = hungarian(F)[0]
print(-ans // 2)
0