from math import isqrt def main(): n=int(input()) n2=n*n x=5000*5000 t=[0 for i in range(n2)] s=[0 for i in range(n2)] z=0 for a in range(1,n): a2=a*a for b in range(1,a+1): if a2+b*b<n2: t[a2+b*b]+=1 for h in range(1,n): h2=h*h for p in range(h+1,n+1): s[p*p-h2]+=1 #print(*t) #print(*s) for i in range(1,n2): z+=s[i]*t[i] print(z) main()
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | from math import isqrt def main(): n=int(input()) n2=n*n x=5000*5000 t=[0 for i in range(n2)] s=[0 for i in range(n2)] z=0 for a in range(1,n): a2=a*a for b in range(1,a+1): if a2+b*b<n2: t[a2+b*b]+=1 for h in range(1,n): h2=h*h for p in range(h+1,n+1): s[p*p-h2]+=1 #print(*t) #print(*s) for i in range(1,n2): z+=s[i]*t[i] print(z) main() |