結果
問題 | No.217 魔方陣を作ろう |
ユーザー | kei |
提出日時 | 2018-09-24 15:17:56 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 529 ms / 5,000 ms |
コード長 | 884 bytes |
コンパイル時間 | 88 ms |
コンパイル使用メモリ | 12,084 KB |
実行使用メモリ | 42,464 KB |
最終ジャッジ日時 | 2023-10-22 02:15:23 |
合計ジャッジ時間 | 13,148 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge10 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 502 ms
42,460 KB |
testcase_01 | AC | 509 ms
42,444 KB |
testcase_02 | AC | 510 ms
42,444 KB |
testcase_03 | AC | 503 ms
42,460 KB |
testcase_04 | AC | 503 ms
42,448 KB |
testcase_05 | AC | 529 ms
42,464 KB |
testcase_06 | AC | 508 ms
42,448 KB |
testcase_07 | AC | 508 ms
42,464 KB |
testcase_08 | AC | 505 ms
42,464 KB |
testcase_09 | AC | 505 ms
42,448 KB |
testcase_10 | AC | 514 ms
42,464 KB |
testcase_11 | AC | 511 ms
42,448 KB |
testcase_12 | AC | 504 ms
42,464 KB |
testcase_13 | AC | 511 ms
42,464 KB |
testcase_14 | AC | 513 ms
42,448 KB |
testcase_15 | AC | 509 ms
42,448 KB |
testcase_16 | AC | 520 ms
42,448 KB |
testcase_17 | AC | 503 ms
42,448 KB |
ソースコード
import numpy as np def magic(n): n = int(n) if n % 2 == 1: p = np.arange(1, n+1) return n*np.mod(p[:, None] + p - (n+3)//2, n) + np.mod(p[:, None] + 2*p-2, n) + 1 elif n % 4 == 0: J = np.mod(np.arange(1, n+1), 4) // 2 K = J[:, None] == J M = np.arange(1, n*n+1, n)[:, None] + np.arange(n) M[K] = n*n + 1 - M[K] else: p = n//2 M = magic(p) M = np.block([[M, M+2*p*p], [M+3*p*p, M+p*p]]) i = np.arange(p) k = (n-2)//4 j = np.concatenate((np.arange(k), np.arange(n-k+1, n))) M[np.ix_(np.concatenate((i, i+p)), j)] = M[np.ix_(np.concatenate((i+p, i)), j)] M[np.ix_([k, k+p], [0, k])] = M[np.ix_([k+p, k], [0, k])] return M n = int(input()) ret = magic(n); for i in range(n): for j in range(n): print(ret[i][j],end=" ") print("")