四平方和定理,又称为拉格朗日定理:
每个正整数都可以表示为至多 4 个正整数的平方和。
如果把 0 包括进去,就正好可以表示为4 个数的平方和。
比如:
5=0^2+0^2+1^2+2^2
7=1^2+1^2+1^2+2^2
对于一个给定的正整数,可能存在多种平方和的表示法。
要求你对 4 个数排序:
0≤a≤b≤c≤d
并对所有的可能表示法按 a,b,c,d为联合主键升序排列,最后输出第一个表示法。
输入格式
输入一个正整数 N。
输出格式
输出4个非负整数,按从小到大排序,中间用空格分开。
数据范围
0<N<5e6
输入样例:
5
输出样例:
0 0 1 2
分析:
1.直接枚举, 四层循环,肯定就超时了。
2.四层循环,每层枚举1~sqrt(n),时间复杂度约为1e12级别,也就超时了。。
想象一下,减少时间复杂度的方法无非是两种,一种是减少循环层数,另一种是减少每次循环的次数。
每层循环次数好像只能减少到sqrt(n)了,那么再从减少循环层数上下手。
设依次选出的数为a,b,c,d
1.d可以使用sqrt(n-a*a-b*b-c*c)表示,如果d是一个正整数,那么就存在a*a+b*b+c*c+d*d=n,这样就可以去掉最后一层循环。
2.在哈希数组vis[]中标记所有小于n且可以使用两个数的平方和表示的数。
3.如果vis[n-a*a-b*b]存在标记,说明n-a*a-b*b能用c和d表示出来,那么就枚举c,然后看int(sqrt(n-a*a-b*b-c*c))是否为正整数n-a*a-b*b-c*c的平方根。
#include<bits/stdc++.h>
using namespace std;
int vis[5000000+10];
int main()
{
int n;
cin>>n;
for(int i = 0; i*i<=n; i++)
{
for(int j = 0; j*j<=n; j++)
{
if(i*i+j*j<=n)
vis[i*i+j*j]=1;
}
}
for(int i = 0; i*i<=n; i++)
{
for(int j = 0; j*j<=n; j++)
{
if(i*i+j*j<=n&&vis[n-i*i-j*j]==0)
{
continue;
}
for(int k=0;k*k<=n;k++)
{
int temp = n-i*i-j*j-k*k;
double w = sqrt(temp);
if(w==(int)w)
{
printf("%d %d %d %d\n",i,j,k,(int)w);
return 0;
}
}
}
}
return 0;
}
还有一个类似的题