题目描述
四平方和定理,又称为拉格朗日定理:每个正整数都可以表示为至多4个正整数的平方和。
如果把0包括进去,就正好可以表示为4个数的平方和。
比如:
5 = 0^2 + 0^2 + 1^2 + 2^2
7 = 1^2 + 1^2 + 1^2 + 2^2(^符号表示乘方的意思)
对于一个给定的正整数N,可能存在多种平方和的表示法。
要求你对4个数排序:0 <= a <= b <= c <= d
并对所有的可能表示法按 a,b,c,d 为联合主键升序排列,最后输出第一个表示法
输入格式
输入存在多组测试数据,每组测试数据输入一行为一个正整数N (N<5000000)
输出格式
对于每组测试数据,要求输出4个非负整数,按从小到大排序,中间用空格分开
代码
#include<iostream>
#include<cmath>
using namespace std;
int main() {
int N;
while (cin >> N) {
bool flag = false;
int limit = sqrt(N); //根据N值设置循环的上界
for (int a = 0; a <= limit; a++) {
for (int b = a; b <= limit; b++) {
for (int c = b; c <= limit; c++) {
int d = sqrt(N - a * a - b * b - c * c);
if (a * a + b * b + c * c + d * d == N && d >= c) {
cout << a << " " << b << " " << c << " " << d << endl;
flag = true;
break;
}
}
if (flag) break; //根据题意只需输出第一种情况
}
if (flag) break;
}
}
return 0;
}