【蓝桥杯】43699-四平方和

四平方和

题目描述

四平方和定理，又称为拉格朗日定理：
每个正整数都可以表示为至多 4 个正整数的平方和。如果把 0 包括进去，就正好可以表示为 4 个数的平方和。
比如：
5=0²+0²+1²+2²
7=1²+1²+1²+2²;
对于一个给定的正整数，可能存在多种平方和的表示法。

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

输入描述

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

输出描述

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

输入输出样例

示例

输入

12

输出

0 2 2 2

解题思路

穷举法
对于给定的正整数 N，我们可以使用穷举法来找到所有可能的表示法。穷举法的思路是，我们逐个检查所有可能的 a、b、c 和 d 值，其中 a、b、c、d 都是非负整数，并且满足 a≤b≤c≤d。
优化穷举范围
为了提高效率，我们可以对 a、b、c 的取值范围进行优化。由于 a、b、c、d 都是非负整数，并且 a≤b≤c≤d，所以 a 的最大值可以取到 N 的平方根，因为 a 的平方不可能大于 N。同理，b 的取值范围可以从 a 开始，最大值可以取到 (N - a²) 的平方根。c 的取值范围可以从 b 开始，最大值可以取到 (N - a² - b²) 的平方根。
计算 d 的值
在确定了 a、b、c 的值之后，我们可以计算 d 的值。d 的值是 (N - a² - b²- c²) 的平方根，并且 d 必须是一个整数。
检查是否满足条件
如果 a² + b² + c² + d² 等于 N，那么我们就找到了一个满足条件的表示法。由于我们按照从小到大的顺序进行穷举，所以找到的第一个表示法就是最小的表示法。
输出结果
最后，我们将找到的四个数按照从小到大的顺序输出，中间用空格分隔。
复杂度分析
这个算法的时间复杂度是 O (N^3/2)，因为我们使用了三层嵌套循环，每层循环的次数最多是 N 的平方根。这个算法在 N 的值不是很大时是可行的，但是对于非常大的 N，这个算法可能会非常慢。

代码实现

def find_four_squares(n):
    # 遍历所有可能的 a 值，从 0 到 sqrt(n)
    for a in range(int(n**0.5) + 1):
        # 遍历所有可能的 b 值，从 a 到 sqrt(n - a^2)
        for b in range(a, int((n - a*a)**0.5) + 1):
            # 遍历所有可能的 c 值，从 b 到 sqrt(n - a^2 - b^2)
            for c in range(b, int((n - a*a - b*b)**0.5) + 1):
                # 计算 d 的平方值
                d_squared = n - a*a - b*b - c*c
                # 检查 d_squared 是否为非负数
                if d_squared >= 0:
                    # 计算 d 的值
                    d = int(d_squared**0.5)
                    # 检查 d 是否为整数
                    if d * d == d_squared:
                        # 返回结果，确保 a <= b <= c <= d
                        return f"{a} {b} {c} {d}"
    # 如果没有找到，则返回报错信息
    return "No solution found"

# 输入一个正整数n
number = int(input())

# 获取结果并输出
result = find_four_squares(number)
print(result)

运行结果

>>> 12
0 2 2 2