四平方和【枚举 + 哈希】

因为数据范围是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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值