結果

問題 No.401 数字の渦巻き
コンテスト
ユーザー はむ吉🐹
提出日時 2016-07-24 13:20:22
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 33 ms / 2,000 ms
コード長 1,092 bytes
コンパイル時間 474 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-11-06 15:49:48
合計ジャッジ時間 2,559 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3

import array


UNDEF = 0


def generate_vortex(size):
    def out_of_stage(r, c):
        return r < 0 or c < 0 or r >= size or c >= size or stage[r][c] != UNDEF

    def advance(r0, c0, d):
        m = d % 4
        if m == 0:
            return (r0, c0 + 1)
        elif m == 1:
            return (r0 + 1, c0)
        elif m == 2:
            return (r0, c0 - 1)
        elif m == 3:
            return (r0 - 1, c0)
        else:
            assert False
    stage = [array.array("I", (UNDEF for _ in range(size)))
             for _ in range(size)]
    dire = 0
    stage[0][0] = 1
    row0, col0 = 0, 0
    for i in range(2, size * size + 1):
        row, col = advance(row0, col0, dire)
        while out_of_stage(row, col):
            dire += 1
            row, col = advance(row0, col0, dire)
        stage[row][col] = i
        row0, col0 = row, col
    return stage


def main():
    size = int(input())
    stage = generate_vortex(size)
    print(*(" ".join("{:03d}".format(x) for x in r) for r in stage), sep="\n")


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