結果

問題 No.2479 Sum of Squares
ユーザー ntuda
提出日時 2025-02-11 16:41:25
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 837 bytes
コンパイル時間 217 ms
コンパイル使用メモリ 82,224 KB
実行使用メモリ 102,108 KB
最終ジャッジ日時 2025-02-11 16:41:36
合計ジャッジ時間 10,126 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2 WA * 1
other RE * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
sys.setrecursionlimit(10050)

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
    xdy = x // (y * y)
    tmp += [y * y] * xdy
    for i in reversed(range(1, xdy + 1)):
        dfs(x - y * y * i, 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