結果
問題 | No.800 四平方定理 |
ユーザー |
![]() |
提出日時 | 2020-01-02 11:44:16 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 709 ms / 2,000 ms |
コード長 | 414 bytes |
コンパイル時間 | 84 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 196,408 KB |
最終ジャッジ日時 | 2024-11-22 17:55:19 |
合計ジャッジ時間 | 21,284 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines import numpy as np N,D = map(int,read().split()) U = 10 ** 7 sq = np.arange(1,N+1,dtype=np.int64) ** 2 xy = sq[:,None] + sq[None,:] xy_cnt = np.bincount(xy.ravel(), minlength=U) rhs = sq[:,None] - sq[None,:] + D rhs = rhs.ravel(); rhs = rhs[rhs >= 0] answer = xy_cnt[rhs].sum() print(answer)