結果

問題 No.401 数字の渦巻き
ユーザー piconic_X
提出日時 2016-08-11 18:31:36
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 522 ms / 2,000 ms
コード長 1,237 bytes
コンパイル時間 153 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 43,904 KB
最終ジャッジ日時 2024-11-07 10:59:57
合計ジャッジ時間 16,999 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np

def make_arr(N):
    arr = np.array([[1 for i in range(N)] for j in range(N)])
    return arr


def next_pos(start, dist, arr, N): # dist = 1(right), -1(left)
    k = start[0]
    l = start[1]
    pos = k,l
    while arr[k,l+dist] == 1:
        if (k,l+dist) == (0,0):
            break
        num = arr[k,l]
        l += dist
        arr[k,l] = num + 1
        if l == N-1 or l == 0:
            break
    pos = l,k
    return pos


def splitter(col):
    l = len(col)
    s = ''
    for i in range(l):
        s += '{0:0>3} '.format(col[i])
    return s


def main():
    N = int(input())
    if N == 1:
        print('001')
    else:
        arr = make_arr(N)
        i = 0
        pos = 0,0
        while i >= 0:
            phase = i % 4
            pre_pos = pos
            if phase < 2:
                pos = next_pos(pre_pos, 1, arr, N)
            else:
                pos = next_pos(pre_pos, -1, arr, N)
            arr = arr.T
            if (pos[1], pos[0]) == pre_pos:
                col = arr.shape[0]
                for i in range(col):
                    print( splitter(arr[i]) )
                break
            else:
                i += 1
    return 0

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