四平方和定理,又称为拉格朗日定理:
每个正整数都可以表示为至多 44 个正整数的平方和。
如果把 00 包括进去,就正好可以表示为 44 个数的平方和。
比如:
5=02+02+12+225=02+02+12+22
7=12+12+12+227=12+12+12+22
对于一个给定的正整数,可能存在多种平方和的表示法。
要求你对 44 个数排序:
0≤a≤b≤c≤d0≤a≤b≤c≤d
并对所有的可能表示法按 a,b,c,da,b,c,d 为联合主键升序排列,最后输出第一个表示法。
输入格式
输入一个正整数 NN。
输出格式
输出4个非负整数,按从小到大排序,中间用空格分开。
数据范围
0<N<5∗1060<N<5∗106
输入样例:
5
输出样例:
0 0 1 2
#include<iostream>
#include<cstring>
using namespace std;
const int N = 5e6 + 10;
int C[N], D[N];
int main()
{
int n;
cin >> n;
memset(C, -1, sizeof C);
for (int c = 0; c * c <= n; c++)
{
for (int d = c; d * d + c * c <= n; d++)
{
int s = c * c + d * d;
if (C[s] == -1)
{
C[s] = c, D[s] = d;
}
}
}
for (int a = 0; a * a <= n; a++)
{
for (int b = a; b * b + a * a <= n; b++)
{
int s = n - a * a - b * b;
if (C[s] != -1)
{
cout << a << ' ' << b << ' ' << C[s] << ' ' << D[s] << endl;
return 0;
}
}
}
return 0;
}
我的至尊代码!!!(有些数据就是不行。。。)
#include<iostream>
#include<queue>
#include<cstring>
#include<math.h>
#include<algorithm>
using namespace std;
const int N = 5e6 + 10;
int m, n;
int arr[4];
int d[3010][4];
int main()
{
cin >> m;
n = m;
int a = sqrt(n);
int sum = 0;
int b;
int ans = 0;
int ans1 = 0;
for (int i = 0; i < 4; i++)
{
//cout << a << ' ' << n << endl;
arr[i] = a;
sum += a * a;
b = a;
a = sqrt(n - a * a);
n = n - b * b;
if (i == 3 && sum == m)
{
sort(arr, arr + 4);
d[ans1][0] = arr[0], d[ans1][1] = arr[1], d[ans1][2] = arr[2], d[ans1][3] = arr[3];
ans1++;
memset(arr, 0, sizeof arr);
i = 0;
ans++;
a = sqrt(m) - ans;
n = m;
sum = 0;
if (a == 0)break;
}
if (i == 3 && sum != m )
{
memset(arr, 0, sizeof arr);
i = 0;
ans++;
a = sqrt(m) - ans;
n = m;
sum = 0;
if (a == 0)break;
}
}
int qwe[4] = { d[0][0],d[0][1],d[0][2],d[0][3] };
for (int i = 1; i < ans1; i++)
{
if (d[i][0] < qwe[0])
{
qwe[0] = d[i][0], qwe[1] = d[i][1], qwe[2] = d[i][2], qwe[3] = d[i][3];
}
else if (d[i][0] == qwe[0])
{
if (d[i][1] < qwe[1])
{
qwe[0] = d[i][0], qwe[1] = d[i][1], qwe[2] = d[i][2], qwe[3] = d[i][3];
}
else if (d[i][1] == qwe[1])
{
if (d[i][2] < qwe[2])
{
qwe[0] = d[i][0], qwe[1] = d[i][1], qwe[2] = d[i][2], qwe[3] = d[i][3];
}
}
}
}
for (int i = 0; i < 4; i++)
{
cout << qwe[i] << ' ';
}
/*for (int i = 0; i < ans1; i++)
{
cout << d[i][0]<<' ' << d[i][1] << ' ' << d[i][2] << ' ' << d[i][3] << endl;
}*/
cout << endl;
return 0;
}