結果

問題 No.1169 Row and Column and Diagonal
コンテスト
ユーザー ntk-ta01
提出日時 2020-08-14 23:18:18
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,186 bytes
コンパイル時間 220 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 105,472 KB
最終ジャッジ日時 2024-10-10 16:42:20
合計ジャッジ時間 4,526 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 4 TLE * 1 -- * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    import sys
    sys.setrecursionlimit(10**6)
    from random import shuffle
    N = int(input())
    ans = [[-1]*N for _ in range(N)]
    for i in range(N):
        ans[i][i] = i+1
    B = [i for i in range(1, N+1)]

    def dfs(i, j, ans):
        if i == j:
            if i == N-1:
                written = True
                return written, ans
            else:
                return dfs(i, j+1, ans)
        canuse = [True]*(N+1)
        for k in range(N):
            if ans[i][k] != -1:
                canuse[ans[i][k]] = False
            if ans[k][j] != -1:
                canuse[ans[k][j]] = False

        written = False
        shuffle(B)
        if any(c for c in canuse):
            for v in B:
                if not written and canuse[v]:
                    ans[i][j] = v
                    if j+1 < N:
                        written, ans = dfs(i, j+1, ans)
                    else:
                        written, ans = dfs(i+1, 0, ans)
            if not written:
                ans[i][j] = -1
        return written, ans

    _, ans = dfs(0, 0, ans)

    for i in range(N):
        print(*ans[i])


if __name__ == '__main__':
    main()
0