c++【入门】四平方和

时间限制 : 1 秒

内存限制 : 128 MB

四平方和定理，又称为拉格朗日定理：每个正整数都可以表示为至多 44 个正整数的平方和。如果把 00 包括进去，就正好可以表示为 44 个数的平方和。

比如：

5=02+02+12+22

7=12+12+12+22

则对于一个给定的正整数 n，可以表示为：n=a2+b2+c2+d2。

你需要求出字典序最小的一组解a,b,c,d。

字典序大小：从左到右依次比较，如果相同则比较下一项，直到有一项不同，较小的一方字典序更小，反之字典序更大，所有项均相同则二者字典序相同。

输入

程序输入为一个正整数 N(1≤N≤5000000)。

输出

输出 4 个非负整数 a,b,c,d，中间用空格分开。

样例

输入

5

输出

0 0 1 2

输入

12

输出

0 2 2 2

提示

其实我们只需要枚举a,b,c 就可以了，因为d 是可以通过a,b,c 的值计算出来的。这样我们就可以减少一层循环，从而降低程序运行时间，以避免运行超时。

代码

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n;
	cin>>n;
	for(int a=0;a<=sqrt(n/4);a++)
	{
		for(int b=a;b<=sqrt((n-a*a)/3);b++)
		{
			for(int c=b;c<=sqrt((n-a*a-b*b)/2);c++)
			{
				int d=sqrt(n-a*a-b*b-c*c);
				if(a*a+b*b+c*c+d*d==n)
				{
					cout<<a<<" "<<b<<" "<<c<<" "<<d;
					return 0;
				}
			}
		}
	}
}