結果
問題 |
No.800 四平方定理
|
ユーザー |
![]() |
提出日時 | 2020-08-19 16:09:53 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,086 bytes |
コンパイル時間 | 1,002 ms |
コンパイル使用メモリ | 82,328 KB |
実行使用メモリ | 83,184 KB |
最終ジャッジ日時 | 2024-10-12 04:54:51 |
合計ジャッジ時間 | 6,109 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | -- * 3 |
other | AC * 1 TLE * 1 -- * 28 |
ソースコード
import sys def input(): return sys.stdin.readline().strip() def list2d(a, b, c): return [[c] * b for i in range(a)] def list3d(a, b, c, d): return [[[d] * c for j in range(b)] for i in range(a)] def list4d(a, b, c, d, e): return [[[[e] * d for j in range(c)] for j in range(b)] for i in range(a)] def ceil(x, y=1): return int(-(-x // y)) def INT(): return int(input()) def MAP(): return map(int, input().split()) def LIST(N=None): return list(MAP()) if N is None else [INT() for i in range(N)] def Yes(): print('Yes') def No(): print('No') def YES(): print('YES') def NO(): print('NO') sys.setrecursionlimit(10 ** 9) INF = 10 ** 19 MOD = 10 ** 9 + 7 EPS = 10 ** -10 N, D = MAP() ans = 0 for w in range(N, 0, -1): for x in range(1, N+1): if x**2 - w**2 > D: break for y in range(1, N+1): if x**2 + y**2 - w**2 > D: break for z in range(1, N+1): if x**2 + y**2 + z**2 - w**2 > D: break if x**2 + y**2 + z**2 - w**2 == D: ans += 1 print(ans)