蓝桥杯python组——四平方和

蓝桥杯python组——四平方和

import os
import sys
import math
n=int(input())
sqn=int(math.sqrt(n))

def f(n):
for a in range(sqn):
bm=int(math.sqrt(n-a*a))+1
for b in range(a,bm):
cm=int(math.sqrt(n-a*a-b*b))+1
for c in range(b,cm):
t=n-a*a-b*b-c*c
d=int(math.sqrt(t))
if(d*d==t and d>=c):
print(a,b,c,d,sep=' ')
return
f(n)

​谢谢大家的支持，您的一键三连是 罡罡同学前进的最大动力！​