結果
問題 |
No.2510 Six Cube Sum Counting
|
ユーザー |
|
提出日時 | 2023-10-20 22:48:57 |
言語 | PyPy3 (7.3.15) |
結果 |
MLE
|
実行時間 | - |
コード長 | 506 bytes |
コンパイル時間 | 310 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 859,328 KB |
最終ジャッジ日時 | 2024-09-20 20:55:10 |
合計ジャッジ時間 | 6,704 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 MLE * 1 -- * 1 |
other | -- * 26 |
ソースコード
from collections import defaultdict from itertools import combinations_with_replacement X = int(input()) C = defaultdict(list) D = defaultdict(list) for a, b, c in combinations_with_replacement(range(301), 3): x = a**3 + b**3 + c**3 if 2 * x <= X: C[x].append(c) elif x <= X: D[x].append(a) cnt = 0 for x in C: if 2 * x > X: break y = X - x C[x].sort() j = 0 for i in range(len(C[x])): while j < len(D[y]) and D[y][j] < C[x][i]: j += 1 cnt += len(D[y]) - j print(cnt)