結果

問題 No.2263 Perms
コンテスト
ユーザー aaaaaaaaaa2230
提出日時 2023-04-07 22:15:29
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 980 bytes
コンパイル時間 226 ms
コンパイル使用メモリ 81,788 KB
実行使用メモリ 79,616 KB
最終ジャッジ日時 2024-10-02 19:45:26
合計ジャッジ時間 10,529 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 11 WA * 23 RE * 5
権限があれば一括ダウンロードができます

ソースコード

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 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