結果

問題 No.2263 Perms
ユーザー aaaaaaaaaa2230
提出日時 2023-04-07 22:19:03
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 996 bytes
コンパイル時間 214 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 79,756 KB
最終ジャッジ日時 2024-10-02 19:47:49
合計ジャッジ時間 9,631 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 8 RE * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

import random

n,m = map(int,input().split())
A = [list(map(int,input().split())) for i in range(n)]

for i in range(n):
    r = 0
    c = 0
    for j in range(n):
        r += A[i][j]
        c += A[j][i]
    if r != m or c != m:
        print(-1)
        exit()


ans = []
for _ in range(m):
    l = [0]*n

    H = [0]*n
    W = [0]*n
    for i in range(n):
        ma = m
        for h in range(n):
            for w in range(n):
                if H[h] or W[w] or A[h][w] == 0:
                    continue
                ma = min(ma,A[h][w])
        
        cands = []
        
        for h in range(n):
            for w in range(n):
                if H[h] or W[w]:
                    continue
                if A[h][w] == ma:
                    cands.append([h,w])

        # assert cands
        rind = random.randint(0,len(cands)-1)
        h,w = cands[rind]
        A[h][w] -= 1
        H[h] = 1
        W[w] = 1
        l[h] = w+1
    ans.append(l)


for i in ans:
    print(*i)
0