Submission #72196503


Source Code Expand

from sys import stdin,stdout
# from os import _exit
# from bisect import bisect_left,bisect
# from heapq import heapify,heappop,heappush
# from sys import setrecursionlimit
# from collections import defaultdict,Counter,deque
# from itertools import permutations
from math import gcd,ceil,sqrt,factorial
# setrecursionlimit(int(1e5))
input,print = stdin.readline,stdout.write

n = int(input())
count = [0 for i in range(n+1)]
lim = int(sqrt(n))+1
for a in range(1,lim):
    a2 = a*a
    if a2>n:
        break
    for b in range(a+1,lim):
        s = a2+b*b
        if s>n:
            break
        count[s]+=1
ans = [i for i,c in enumerate(count) if c==1]
print(f"{len(ans)}\n")
print(" ".join(map(str,ans)))

Submission Info

Submission Time
Task C - 2026
User aviral
Language Python (PyPy 3.11-v7.3.20)
Score 300
Code Size 736 Byte
Status AC
Exec Time 230 ms
Memory 250524 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 300 / 300
Status
AC × 3
AC × 18
Set Name Test Cases
Sample 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt
All 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt, 01_random_00.txt, 01_random_01.txt, 01_random_02.txt, 01_random_03.txt, 01_random_04.txt, 01_random_05.txt, 01_random_06.txt, 01_random_07.txt, 01_random_08.txt, 01_random_09.txt, 02_corner_00.txt, 02_corner_01.txt, 02_corner_02.txt, 02_corner_03.txt, 02_corner_04.txt
Case Name Status Exec Time Memory
00_sample_00.txt AC 51 ms 80280 KiB
00_sample_01.txt AC 50 ms 79860 KiB
00_sample_02.txt AC 51 ms 79708 KiB
01_random_00.txt AC 129 ms 172948 KiB
01_random_01.txt AC 212 ms 240704 KiB
01_random_02.txt AC 127 ms 172448 KiB
01_random_03.txt AC 218 ms 244260 KiB
01_random_04.txt AC 100 ms 141124 KiB
01_random_05.txt AC 140 ms 186248 KiB
01_random_06.txt AC 114 ms 156308 KiB
01_random_07.txt AC 107 ms 149196 KiB
01_random_08.txt AC 138 ms 183560 KiB
01_random_09.txt AC 225 ms 247776 KiB
02_corner_00.txt AC 54 ms 86328 KiB
02_corner_01.txt AC 227 ms 250204 KiB
02_corner_02.txt AC 229 ms 250524 KiB
02_corner_03.txt AC 230 ms 250204 KiB
02_corner_04.txt AC 229 ms 250364 KiB