結果
問題 | No.800 四平方定理 |
ユーザー |
|
提出日時 | 2020-04-24 10:44:42 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 368 bytes |
コンパイル時間 | 95 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 167,664 KB |
最終ジャッジ日時 | 2024-10-15 01:46:10 |
合計ジャッジ時間 | 40,321 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 TLE * 3 |
ソースコード
import numpy as np def main(): n,d = map(int,input().split()) square = np.arange(1,n + 1) ** 2 Add = np.add.outer(square,square).flatten() Sub = (square[:,None] - square[None,:]).flatten() + d Sub = np.sort(Sub) ans = np.searchsorted(Sub,Add,'right') - np.searchsorted(Sub,Add,'left') print(ans.sum()) if __name__ =='__main__': main()