因为数据范围是5 * 1e6,每个数的范围都在(0,根号n)的范围,所以最多只能枚举两重循环(5 * 1e6开根号在2200左右,时间复杂度不能超过1e7,(2200)^2 约等于10的6次方),根据等式
a ^ 2 + b ^ 2 + c ^2 + d ^ 2 == x,变换一下 等价于a ^ 2 + b ^ 2 == x - c ^ 2 - d ^ 2
所以可以先循环枚举c和d,将x - (c ^ 2 + d ^2)的值作为数组的值,将c的取值作为数组的映射,再从小到大枚举较小的两个数,若平方和的值在数组中存在,即为第一个合理的解。
建立哈希表,存储小于n的数x,假设e = c^2 + d^2,则h[e] = c,从小到大遍历c, d只存储先出现的(因为字典序小)
从小到大遍历a、b,查找1中建立的哈希表,若 h[n - a^2 - b^2]存在,则算出d,
时间复杂度 O(n)
空间复杂度 O(n)
#include<iostream>
#include<cmath>
using namespace std;
const int N = 1e8 + 10;
int h[N];
int main() {
int n;
cin >> n;
记录所有完全平方数两两之和,只记第一次出现(题目要求找出字典序小的)
for (int i = 0; i * i * 2<= n; i++)
{
for (int j = i; j * j + i * i <= n; j++)
{
if (!h[i * i + j * j])//记录每个值第一次出现的c和d
h[i * i + j * j] = i + 1;//防止i = 0时在后面判断查找跳过 i = 0的情况
}
}
//0<= a <= b <= c <= d,可以得出a^2 <= n / 4, a^2 + b^ 2 <= n / 2;
for (int i = 0; i * i * 4 <= n; i++)
{
for (int j = i; j * j + i * i <= n / 2; j++)
{
int t = n - i * i - j * j;
if (h[t]) {
int c = h[t] - 1;
int d = (sqrt(t - c * c));
cout<<i<<" "<<j<<" "<<c<<" " <<d;
return 0;
}
}
}
return 0;
}