蓝桥杯：四平方和

题目

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

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

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

比如：

5 = 0^2 + 0^2 + 1^2 + 2^25=02+02+12+22；

7 = 1^2 + 1^2 + 1^2 + 2^27=12+12+12+22；

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

要求你对 4 个数排序：

0 \leq a \leq b \leq c \leq d0≤a≤b≤c≤d

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

输入
12

输出
0 2 2 2

思路

暴力枚举，最后一个d，用n减其他三个的平方，再去根号，求d和去根号的平方是不是相等，相等则是结果（因为是枚举的，和拉格朗日定理推出）。

代码

#include <iostream>
#include<cmath>
using namespace std;
typedef long long ll;
ll N = 5000;
int main()
{
  ll n;
  cin>>n;
  for(ll a = 0;a < N;a++)
  {
    for(ll b = 0;b < N;b++)
    {
      if(a*a+b*b > n) break;
      for(ll c = 0;c < N;c++)
      {
      	if(a*a+b*b+c*c > n)break;
		int t = n - (a*a+b*b+c*c);
		int d = sqrt(t);
		if(d*d == t)
		{
			printf("%ld %ld %ld %ld",a,b,c,d);
			return 0;
		} 
      }
    }
  }
  return 0;
}

• 总结：做题先想暴力方法，优化。