2016蓝桥杯C++A:四平方和（详细讲解）

八、题目：四平方和
四平方和定理，又称为拉格朗日定理：

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

比如：
5 = 0^ 2 + 0^ 2 + 1^ 2 + 2^2
7 = 1^ 2 + 1^ 2 + 1^ 2 + 2^2
（^符号表示乘方的意思）

对于一个给定的正整数，可能存在多种平方和的表示法。
要求你对4个数排序：
0 <= a <= b <= c <= d
并对所有的可能表示法按 a,b,c,d 为联合主键升序排列，最后输出第一个表示法

程序输入为一个正整数N (N<5000000)
要求输出4个非负整数，按从小到大排序，中间用空格分开

例如，输入：
5
则程序应该输出：
0 0 1 2

再例如，输入：
12
则程序应该输出：
0 2 2 2

再例如，输入：
773535
则程序应该输出：
1 1 267 838

资源约定：
峰值内存消耗（含虚拟机） < 256M
CPU消耗 < 3000ms

---

分析思路：
联合主键升序排列：
5 = 0^2 + 0^2 + 1^2 + 2^2    第一位相同的情况下，比较第二位，第二位相同的情况下，比较第三位，以此类推，最后输出字典序最小的一组
枚举法：输出a,b,c,d
暴力解法：

这个思路可以拿一部分分值，规模很大，N为5*10的6次方
四层算法就是(10 ^ 3) ^ 4 =10^12

• 枚举法需要进行优化，减少枚举范围但N无法减少
• 减少枚举的变量，就可以减少枚举的层数
  上面我们把a,b,c,d已经凑出来了
  N-a^ 2-b^ 2-c^ 2=d^2
  试着把d开平方（2000开平方），要是能够开出来算通过，三层规模变为10^9
• 进一步可以算c ^ 2+d ^2=x，把它缓存起来，再算N-(a ^2 + b ^2)=x’ ;只需要差x’是否为两个数的平方和c ^ 2+d ^2
• 用哈希表进行缓存
• 还要注意数字排序0 <= a <= b <= c <= d条件

#include<iostream>
#include<cstdio>
#include<map>
#include<cmath>
using namespace std;

int N;
map<int,int>cache;  //int和int映射，哈希表用于缓存数值
int main(){
	scanf("%d",&N);
	for(int c=0;c*c<=N/2;c++){ 
		for(int d=c;c*c+d*d<=N;d++){
			if(cache.find(c*c+ d*d)==cache.end()) //如果未找到，再去存，已经存在就不需要去存了 
			cache[c*c+d*d]=c;  //查平方数，先要看cache里面有没有，没有说明它不是一个平方数存起来 
		//有的话通过较小的数映射找到它 
		}
	}
	//c*c+d*d要比a*a+d*d要大（至少相同） 
	for(int a=0;a*a<=N/4;a++){  //单独看a*a小于等于N/4 
		for(int b=a;a*a+b*b<=N/2;b++){ //a*a+b*b小于等于N/2  
			if(cache.find(N-a*a-b*b)!=cache.end()){ //查表，能找到 
				int c=cache[N-a*a-b*b];//c就可以知道 
				int d=int(sqrt(N-a*a-b*b-c*c)); //d开方 
				printf("%d %d %d %d\n",a,b,c,d);
					return 0;  //跳出整个主函数 
			}
		}
	}
	return 0; 
} 
//整个过程不需要再排序，因为一开始在写代码时，就已经限定了顺序大小，c*c+d*d要比a*a+d*d要大 

总结：枚举加优化，减少其层数