NOI--和为给定数

07:和为给定数

描述

给出若干个整数，询问其中是否有一对数的和等于给定的数。

输入

共三行：
第一行是整数n(0 < n <= 100,000)，表示有n个整数。
第二行是n个整数。整数的范围是在0到10^8之间。
第三行是一个整数m（0 <= m <= 2^30)，表示需要得到的和。

输出

若存在和为m的数对，输出两个整数，小的在前，大的在后，中间用单个空格隔开。若有多个数对满足条件，选择数对中较小的数更小的。若找不到符合要求的数对，输出一行No。

样例输入

4
2 5 1 4
6

样例输出

1 5

解题：

WA了一次，发现如果测试数据为

4
1 2 3 4
2

输出 1 1，实际上输出为“No”，说明不能重复找，所以把每次的查找起始位置换为该元素的下一个元素。

AC代码:

#include<bits/stdc++.h>
using namespace std;

int n,m;
int a[100005];
	
bool check(int x, int y) {
	int l = y;
	int r = n;
	while(l < r) {
		int mid = (l + r) / 2;
		if(x > a[mid]) l = mid + 1;
		else if(x < a[mid])r = mid;
		else if(x == a[mid]) return true;
	}
	return false;
}

int main() {
	cin>>n;
	for(int i = 0; i < n; i++) {
		cin>>a[i];
	}
	cin>>m;
	sort(a, a + n);
	for(int i = 0; i < n; i++) {
		int p = a[i];
		int q = m - a[i];
        //这里，每次查找从该元素的下一个元素开始找就不会重复啦
		if(check(q, i + 1)) {
			cout<<p<<" "<<q<<endl;
			return 0;
		}
	}
	cout<<"No"<<endl;
	return 0;
}