結果

問題 No.2510 Six Cube Sum Counting
ユーザー navel_tos
提出日時 2023-10-20 23:19:25
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,859 bytes
コンパイル時間 327 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 920,156 KB
最終ジャッジ日時 2024-09-20 22:54:35
合計ジャッジ時間 10,981 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample MLE * 1 -- * 3
other -- * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#yukicoder409C

#愚直だと余裕を持ってTLE
def simple():
    X=int(input()); L=[pow(i,3) for i in range(301)]; ans=0
    for a in range(301):
        if L[a]*6>=X: break
        cnt=L[a]
        for b in range(a,301):
            if cnt+L[b]*5>=X: break
            cnt=L[a]+L[b]
            for c in range(b,301):
                if cnt+L[c]*4>=X: break
                cnt=L[a]+L[b]+L[c]
                for d in range(c,301):
                    if cnt+L[d]*3>=X: break
                    cnt=L[a]+L[b]+L[c]+L[d]
                    for e in range(d,301):
                        if cnt+L[e]*2>=X: break
                        cnt=L[a]+L[b]+L[c]+L[d]+L[e]
                        for f in range(e,301):
                            if   cnt+L[f]==X: ans+=1
                            elif cnt+L[f] >X: break
    return ans


#方針が立たない。仕方ないので半分全列挙。
from collections import defaultdict as dd

X=int(input()); P=[pow(i,3) for i in range(301)]
g=lambda x,y,z: P[x]+P[y]+P[z]

#cを固定して全列挙を行う
#S: g(-,-,c以下)の値をListに格納  L: g(c以上,-,-)の数をDictに格納
S,L=dd(int),dd(int)

#先にg(d,e,f)の全パターンをLに格納する  引数のfと関数のfが重複するのを避けるため関数をgに
for a in range(301):
    for b in range(a,301):
        for c in range(b,301): L[g(a,b,c)]+=1

#cを固定して全探索
from time import perf_counter as pf
Z=pf()
ans=0
for c in range(301):
    #あり得るパターンを加算
    for a in range(c+1):
        for b in range(a,c+1): S[P[a]+P[b]+P[c]]+=1
    #Sを全探索
    for now in S: ans+=S[now]*L[X-now]
    #終了処理 最小値(d)がcであるパターンをLから除去
    for e in range(c,301):
        for f in range(c,301): L[P[c]+P[e]+P[f]]-=1
    if c%100==0: print(c,pf()-Z)
0