結果

問題 No.2263 Perms
ユーザー aaaaaaaaaa2230
提出日時 2023-04-07 22:10:11
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 979 bytes
コンパイル時間 207 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 78,736 KB
最終ジャッジ日時 2024-10-02 19:40:52
合計ジャッジ時間 9,057 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 11 RE * 28
権限があれば一括ダウンロードができます

ソースコード

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 = 0
        for h in range(n):
            for w in range(n):
                if H[h] or W[w]:
                    continue
                ma = max(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 ma > 0
        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