sort和binary_search的妙用

最新推荐文章于 2024-01-20 16:20:14 发布

南雨

最新推荐文章于 2024-01-20 16:20:14 发布

阅读量226

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/qq_44509100/article/details/93663633

版权

算法专栏收录该内容

11 篇文章 0 订阅

订阅专栏

sort和binary_search的妙用

问题描述
数据规模
样例输入
样例输出
问题求解

问题描述

给定 $n$ 个数 $a_1$ 、 $a_2$ … $a_n$ ，要求判断这组数里是否存在某两个数(可以重复选取同一个数)的和为 $s u m$ 。

数据规模

$\le n \le 100000$
$\le a_i \le 10000$
$\le sum \le 1000000000$

样例输入

2 3 4 5

样例输出

YES

问题求解

很简单的一个思路是枚举出所有的情况，一一和 sum 比较，但是这样算法的复杂度是 $O(n^2)$ ，无法完成需求，所以必须简化算法。
想一下： $a + b = s u m$ 可以转化为 $a = s u m - b$ ，所以我们可以利用二分查找来简化算法，代码如下：

#include <iostream>
#include <algorithm>
using namespace std;

const int NUM = 1e5 + 50;
int a[NUM];
int n;
int sum;

void solve() {
	sort(a, a + n);
	for (int i = 1; i <= n; ++i) {
		int diff = sum - a[i];
		if (binary_search(a, a + n, diff)) {
			cout << "YES" << endl;
			return;
		}
	}
	cout << "NO" << endl;
}

int main(int argc, char** argv) {
	cin >> n;
	for (int i = 1; i <= n; ++i) {
		cin >> a[i];
	}
	cin >> sum;
	solve();
	
	return 0;
}