蓝桥题库 四平方和

四平方和定理，又称为拉格朗日定理：

每个正整数都可以表示为至多 4 个正整数的平方和。

如果把 0 包括进去，就正好可以表示为 4 个数的平方和。

比如：

对于一个给定的正整数，可能存在多种平方和的表示法。

要求你对 4 个数排序：

并对所有的可能表示法按 a,b,c,d 为联合主键升序排列，最后输出第一个表示法。

输入描述

程序输入为一个正整数 N(N<5×10^6)。

输出描述

要求输出 4 个非负整数，按从小到大排序，中间用空格分开

输入输出样例

示例

输入

12

输出

0 2 2 2

运行限制

• 最大运行时间：3s

• 最大运行内存: 256M

解题思路

暴力解法，遍历0到根号n之间的每一个数，嵌套三层循环，寻找一个数，满足和i, j, k的平方和为n。

第一层循环：寻找i

第二层循环：寻找j，j范围大于等于i

第三层循环：寻找k，k范围大于等于j

通过n和前三个数i, j, k计算第四个数，并验证第四个数是否满足平方和等于n。

import math

n = int(input())
m = int(math.pow(n, 1/2))    # 指数计算math.pow(a, b) -> a的b次方

def f(n):
    for i in range(m):    # 第一个数
        for j in range(i, m):    # 第二个数
            for k in range(j, m):    # 第三个数
                tmp = (n - i*i - j*j - k*k)    # 假设满足条件的第四个数存在
                if tmp < 0: continue          # 若第四个数小于0,则不存在
                elif math.pow(int(math.sqrt(tmp)),2) + i*i + j*j + k*k == n:    #验证平方和是否满足要求
                    print(i, end = " ")    # 依次打印，end = " ",表示以空格结尾
                    print(j, end = " ")
                    print(k, end = " ")
                    print(int(math.sqrt(n - i*i - j*j - k*k)))
                    return

f(n)