分治算法——两元素和(归并排序+二分查找)

一、问题描述

两元素和
Time Limit: 1000 MSMemory Limit: 5000 KB

Description

给定一个N(N<=50000)个int型整数的集合以及一个int型整数X, 问集合中是否存在两个元素的和等于X.

Input

第一行输入M表示有M组测试. 每组测试首先输入N和X,接下来输入N个int型整数.

Output

若否存在两个元素的和等于X则输出yes, 否则输出no.

Sample Input

2
8 7 
1 5 11 5 4 3 9 6
8 7 
1 5 11 5 5 3 9 5

Sample Output

yes
no

二、问题分析

一、大部分人第一反应想到的可能都是暴力求解,即挨个查询,但此思路时间复杂度为n平方;当数组中数据很多事,查询时间指数增长,该方法不再适用;

二、如果数组是有序的,则查找效率会大幅提高,比如二分查找,时间复杂度为O(logn),最坏情况也仅为O(n);而排序算法时间复杂度为O(nlogn),所以整体复杂度仅为O(nlogn),所以我们可以先对数组排序,后进行查找

三、代码示例

#include <iostream>
#include <algorithm>

using namespace std;

const int MAXN = 50005;

int arr[MAXN];

bool has_pair_with_sum(int arr[], int n, int X) {
    // 先对整个数组进行快速排序
    sort(arr, arr + n);
    // 对于每个元素
    for (int i = 0; i < n; i++) {
        // 使用二分查找来查找 X 减去该元素的值是否在数组中出现过
        int j = lower_bound(arr + i + 1, arr + n, X - arr[i]) - arr;
        // 如果出现过,则说明存在两个元素的和等于 X
        if (j < n && arr[j] == X - arr[i]) {
            return true;
        }
    }
    // 如果不存在两个元素的和等于 X,则返回 false
    return false;
}

int main() {
    int t;
    cin >> t;
    while (t--) {
        int n, X;
        cin >> n >> X;
        for (int i = 0; i < n; i++) {
            cin >> arr[i];
        }
        if (has_pair_with_sum(arr, n, X)) {
            cout << "yes" << endl;
        } else {
            cout << "no" << endl;
        }
    }
    return 0;
}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Mu Haitian

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值