AcWing——四平方和

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

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

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

比如：

5=02+02+12+22

7=12+12+12+22

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

要求你对 4个数排序：

0≤a≤b≤c≤d

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

输入格式

输入一个正整数 N。

输出格式

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

数据范围

0<N<5∗106

输入样例：

5

输出样例：

0 0 1 2

题意：

对给出的一个数x，找出满足a^2+b^2+c^2+d^2==x的最小的字典序的a，b，c，d

分析：

在不考虑任何算法的情况下，我们可以用四层循环找出a，b，c，d的值

但四层喜欢必然超时，因此我们可以考虑对其进行优化。

第一次优化：如果我们知道a，b，c，那么d其实可以根据n-a^2-b^2-c^2算出，因此可以优化为3层循环，但显然还是不够，因此我们要继续对其进行优化

第二次优化：一般来讲，三层循环要优化成两层循环都是通过拆分成两个二重循环来实现的，我们可以通过先hash保存c^2+d^2以及c的值， 然后再从小到大枚举a,b的值，这样就可以保证第一个出现的答案是所求答案

tips：

很多人在看到上面之后会有两个疑问：第一：为什么不能先保存a，b的值，在枚举c，d的值

第二：为什么所求结果就是最小字典序呢？

对于第一个问题：如果我们先保存a^2+b^2的话，那么枚举c，d是根据c，d从大到小枚举的，可能虽然c，d小，但匹配到的a，b会更大点，这样答案就错了

第二个问题：我们在保存了c^2+d^2的结果后，开始枚举a，b，那么a，b一定是最小的，既然a，b是最小的，那么又正好有c跟d跟他们配对，那么a，b，c，d就一定是所求的最小字典序了（不知道这样能不能解释清楚）

#include <iostream>
#include <math.h>
#include <cstdio>
#include <cstring>
#include<algorithm>
using namespace std;
const int N=5e6+5;
int r[N];
int main()
{
    int n;
    cin>>n;
    memset(r,-1,sizeof(r));
    for(int c=0;c*c<=n;++c){
        for(int d=c;d*d+c*c<=n;++d){
            int t=c*c+d*d;
//t在之前没出现过才保存，有出现的话就不要这一组，因为前一组的字典序更小
            if(r[t]==-1){
                r[t]=c;
            }
        }
    }
    for(int a=0;a*a<=n;++a){
        for(int b=a;b*b+a*a<=n;++b){
            int t=n-a*a-b*b;
            if(r[t]==-1){
                continue;
            }
            int c=r[t];
            int d=sqrt(t-c*c);
            cout<<a<<' '<<b<<' '<<c<<' '<<d;
            return 0;
        }
    }
    return 0;
}