题目大意
四平方和定理,又称为拉格朗日定理:每个正整数都可以表示为至多 4 个正整数的平方和。如果把 0包括进去,就正好可以表示为 4 个数的平方和。
比如:
5=0^2+0^2+1^2+2^2
7=1^2+1^2+1^2+2^2
则对于一个给定的正整数 n,可以表示为:n=a^2+b^2+c^2+d……。
你需要求出字典序最小的一组解 a,b,c,d。
字典序大小:从左到右依次比较,如果相同则比较下一项,直到有一项不同,较小的一方字典序更小,反之字典序更大,所有项均相同则二者字典序相同。
输入格式
程序输入为一个正整数 N(1≤N≤5000000)。
输出格式
输出 4个非负整数 a,b,c,d中间用空格分开。
样例输入1
5
样例输出1
0 0 1 2
样例输入2
12
样例输出2
0 2 2 2
其实我们只需要枚举 a,b,ca,b,c 就可以了,因为 dd 是可以通过 a,b,c
a,b,c 的值计算出来的。这样我们就可以减少一层循环,从而降低程序运行时间,以避免运行超时。
#include<iostream>
#include<cmath>
using namespace std;
int main(){
int n;
cin>>n;
for(int i1=0;i1<2300;i1++){
for(int i2=0;i2<2300;i2++){
for(int i3=0;i3<2300;i3++){
int i4=sqrt(n-i1*i1-i2*i2-i3*i3);
if(i1*i1+i2*i2+i3*i3+i4*i4==n){
cout<<i1<<" "<<i2<<" "<<i3<<" "<<i4<<endl;
return 0;
}
}
}
}
}