51Nod 1001 数组中和等于K的数对(二分)

Problem:

给出一个整数K和一个无序数组A,A的元素为N个互不相同的整数,找出数组A中所有和等于K的数对。例如K = 8,数组A:{-1,6,5,3,4,2,9,0,8},所有和等于8的数对包括(-1,9),(0,8),(2,6),(3,5)。

Input

第1行:用空格隔开的2个数,K N,N为A数组的长度。(2 <= N <= 50000,-10^9 <= K <= 10^9)
第2 - N + 1行:A数组的N个元素。(-10^9 <= A[i] <= 10^9) 

Output

第1 - M行:每行2个数,要求较小的数在前面,并且这M个数对按照较小的数升序排列。
如果不存在任何一组解则输出:No Solution。

Input示例

8 9
-1
6
5
3
4
2
9
0
8

Output示例

-1 9
0 8
2 6
3 5

分析:题目给的数据规模大小为1e5,暴力的话肯定会超时,这里我用一层for循环加二分查找,时间复杂度为O(n*log2n);

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
const int MAXN = 50010;
typedef long long ll;
ll a[MAXN];
int main() {
	ll k, n;
	scanf("%lld %lld", &k, &n);
	for(int i = 0; i < n; i++) {
		scanf("%lld", &a[i]);
	}
	sort(a, a+n);
	int flag = 0;
	for(int i = 0; i < n; i++) {
		int l = i + 1, r = n - 1, mid;
		while(l <= r) {
			mid = (l + r) >> 1;
			if(a[i] + a[mid] > k) {
				r = mid - 1;
			} else if(a[i] + a[mid] < k) {
				l = mid + 1;
			} else {
				flag = 1;
				printf("%lld %lld\n", a[i], a[mid]);
				break;
			}
		}
	}
	if(!flag) {
		printf("No Solution\n");
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/Adusts/article/details/81607742
今日推荐