提出 #72208509

---

ソースコード 拡げる

from math import isqrt

def main():
    s = input().strip()
    if not s:
        return
    N = int(s)
    cnt = bytearray(N + 1)

    x_max = isqrt(N // 2)
    for x in range(1, x_max + 1):
        xx = x * x
        y_max = isqrt(N - xx)
        start = x + 1
        if start > y_max:
            continue
        for y in range(start, y_max + 1):
            ssum = xx + y * y
            if cnt[ssum] == 0:
                cnt[ssum] = 1
            elif cnt[ssum] == 1:
                cnt[ssum] = 2

    goods = [str(n) for n in range(1, N + 1) if cnt[n] == 1]
    print(len(goods))
    if goods:
        print(" ".join(goods))
    else:
        print()

if __name__ == "__main__":
    main()

提出情報

ジャッジ結果