結果

問題 No.2263 Perms
ユーザー KazunKazun
提出日時 2023-04-07 22:28:30
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 113 ms / 2,000 ms
コード長 2,498 bytes
コンパイル時間 304 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 77,460 KB
最終ジャッジ日時 2024-10-02 19:55:34
合計ジャッジ時間 4,943 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
52,736 KB
testcase_01 AC 38 ms
52,352 KB
testcase_02 AC 38 ms
52,352 KB
testcase_03 AC 39 ms
52,480 KB
testcase_04 AC 39 ms
52,224 KB
testcase_05 AC 52 ms
61,440 KB
testcase_06 AC 112 ms
77,032 KB
testcase_07 AC 106 ms
77,196 KB
testcase_08 AC 96 ms
76,916 KB
testcase_09 AC 113 ms
76,924 KB
testcase_10 AC 39 ms
52,736 KB
testcase_11 AC 110 ms
77,176 KB
testcase_12 AC 113 ms
76,944 KB
testcase_13 AC 113 ms
77,184 KB
testcase_14 AC 113 ms
76,820 KB
testcase_15 AC 38 ms
52,864 KB
testcase_16 AC 66 ms
67,968 KB
testcase_17 AC 72 ms
70,656 KB
testcase_18 AC 66 ms
67,712 KB
testcase_19 AC 79 ms
73,728 KB
testcase_20 AC 110 ms
77,460 KB
testcase_21 AC 108 ms
76,804 KB
testcase_22 AC 107 ms
76,904 KB
testcase_23 AC 93 ms
76,288 KB
testcase_24 AC 72 ms
70,528 KB
testcase_25 AC 105 ms
76,820 KB
testcase_26 AC 111 ms
77,080 KB
testcase_27 AC 91 ms
76,544 KB
testcase_28 AC 87 ms
77,056 KB
testcase_29 AC 82 ms
74,496 KB
testcase_30 AC 57 ms
63,360 KB
testcase_31 AC 48 ms
60,160 KB
testcase_32 AC 72 ms
70,400 KB
testcase_33 AC 70 ms
69,632 KB
testcase_34 AC 40 ms
52,992 KB
testcase_35 AC 39 ms
52,352 KB
testcase_36 AC 59 ms
65,024 KB
testcase_37 AC 41 ms
53,376 KB
testcase_38 AC 104 ms
77,200 KB
testcase_39 AC 39 ms
52,352 KB
testcase_40 AC 39 ms
52,608 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

def Hungarian(A, minimize=False, mode=False):
    """ 行列 A に対してハンガリアン法を適用し, 最大値を求める.

    A: 行列
    minimize: True にすると, 最小値を求める
    mode: True にすると, 最大値 (最小値) を達成する例も出力する.

    Reference: https://judge.yosupo.jp/submission/34963
    """
    if minimize:
        if mode:
            score, x=Hungarian([[-a_ij for a_ij in Ai] for Ai in A], False, True)
            return -score,x
        else:
            return -Hungarian([[-a_ij for a_ij in Ai] for Ai in A], False, False)

    inf=1<<31
    N=len(A)
    fx=[inf]*N
    fy=[0]*N
    x=[-1]*N
    y=[-1]*N
    i=0
    while i < N:
        t=[-1]*N
        s=[i]*(N + 1)
        p=q=0
        while p <= q and x[i] < 0:
            k, j = s[p], 0
            while j < N and x[i] < 0:
                if fx[k] + fy[j] == A[k][j] and t[j] < 0:
                    q += 1
                    s[q] = y[j]
                    t[j] = k
                    if s[q] < 0:
                        p = j
                        while p >= 0:
                            y[j] = k = t[j]
                            p = x[k]
                            x[k] = j
                            j = p
                j += 1
            p += 1
        if x[i] < 0:
            d = inf
            for k in range(q + 1):
                for j in range(N):
                    if t[j] < 0:
                        d = min(d, fx[s[k]] + fy[j] - A[s[k]][j])
            for j in range(N):
                if t[j] >= 0:
                    fy[j] += d
            for k in range(q + 1):
                fx[s[k]] -= d
        else:
            i += 1

    score=0
    for i,j in enumerate(x):
        score+=A[i][j]

    return (score,x) if mode else score

#==================================================
def solve():
    N,M=map(int,input().split())

    A=[None]*N
    S=[0]*N
    for i in range(N):
        A[i]=list(map(int,input().split()))

        if sum(A[i])!=M:
            print(-1)
            return

        for j in range(N):
            S[j]+=A[i][j]

    if not all(S[j]==M for j in range(N)):
        print(-1)
        return

    for _ in range(M):
        B=[[1 if A[i][j]>0 else 0 for j in range(N)] for i in range(N)]

        K,P=Hungarian(B,False,True)
        for i in range(N):
            A[i][P[i]]-=1

        print(*[P[i]+1 for i in range(N)])

    return



#==================================================
solve()
0