A-B数对（洛谷p1102）

最新推荐文章于 2024-10-11 08:00:00 发布

spiritualfood

最新推荐文章于 2024-10-11 08:00:00 发布

阅读量291

点赞数 5

文章标签：算法

本文链接：https://blog.csdn.net/spiritualfood/article/details/142052074

版权

P1102 A-B 数对 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

这道题和之前那道题类似，相当于是拓展，都是对lower_bound和upper_bound的应用

查找（洛谷p2249）-CSDN博客

大家可以看完那道来看这道，他要找到所有A-B=C的数对，那么我们的一个基本思路是什么，给出B和C，那么我们就可以算出A，然后在对应的数组里找到A，算一个。那这里有个细节，有几个A呢？不好确定，那我们此时就可以用到lower_bound和upper_bound

我们用lower_bound找一遍，再用upper_bound找一遍，之间的不就是A的数量，然后就是这样，贴代码。

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
int a[200000];
int main() {
	int n, m;
	long long score = 0;
	scanf("%d%d", &n, &m);
	for (int i = 0; i < n; i++)
		scanf("%d", &a[i]);
	sort(a, a + n);
	for (int i = 0; i < n; i++) {
		int poi1=lower_bound(a, a + n, a[i] + m) - a;
		int poi2 = upper_bound(a, a + n, a[i] + m) - a;
			if (a[poi1] == a[i] + m) {
				score += poi2 - poi1;
			}
			if (poi1 == n - 1 && a[poi1] !=a[i]+m) {
				break;
		}
	}
	cout << score;
	return 0;
}

注意，题目上并没有说给出的序列是有序的，所以我们需要用到sort函数先排下序

然后这段代码的作用就是减少运行次数，如果这个A没找到，后面的数越来越大，那更找不到，直接退出循环就好