結果
問題 |
No.634 硬貨の枚数1
|
ユーザー |
|
提出日時 | 2020-05-16 12:19:57 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 617 ms / 2,000 ms |
コード長 | 408 bytes |
コンパイル時間 | 152 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 166,516 KB |
最終ジャッジ日時 | 2024-09-22 05:04:38 |
合計ジャッジ時間 | 50,302 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 75 |
ソースコード
import sys sys.setrecursionlimit(100000000) MOD = 10 ** 9 + 7 INF = 10 ** 15 import numpy as np def main(): N = int(input()) A = np.arange(5000,dtype = np.int32) tri = A * (A + 1)//2 if np.any(tri == N): print(1) return tri2 = (tri[:,None] + tri[None,:]).ravel() if np.any(tri2 == N): print(2) else: print(3) if __name__ == '__main__': main()