蓝桥杯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)

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