結果

問題 No.2479 Sum of Squares
ユーザー ntuda
提出日時 2025-02-11 16:36:33
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 793 bytes
コンパイル時間 209 ms
コンパイル使用メモリ 82,496 KB
実行使用メモリ 948,464 KB
最終ジャッジ日時 2025-02-11 16:37:58
合計ジャッジ時間 70,399 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2 MLE * 1
other TLE * 18 MLE * 4
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
sys.setrecursionlimit(200050)

def search(x):  # a * a <= y を満たす最大のaを求める
    lb = 1
    ub = x + 1
    while ub - lb > 1:
        mid = (ub + lb) // 2
        if mid * mid <= x:
            lb = mid
        else:
            ub = mid
    return lb

def dfs(x,y,n):
    global ans,tmp,ans2
    if n > 15:
        return
    if x == 0:
        print(n)
        print(*tmp)
        exit()
        return
    if y == 1:
        if n + x < 16:
            print(n + x)
            print(*(tmp + [1] * x))
            exit()
        return
    ret = 0
    if x - y * y >= 0:
        tmp.append(y * y)
        dfs(x - y * y, y, n + 1)
        tmp.pop()
    y -= 1
    if x // (y * y) + n < 16:
        dfs(x, y, n)

tmp = []
S = int(input())
y = search(S)
dfs(S,y,0)
0