洛谷 P1102 A-B 数对题解

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

OkayPJ

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

阅读量459

点赞数

分类专栏：题解文章标签： c++ 数据结构 NOIP

本文链接：https://blog.csdn.net/digitalWrite/article/details/107336070

版权

题解专栏收录该内容

3 篇文章 0 订阅

订阅专栏

P1102 A-B 数对

链接

题目描述

出题是一件痛苦的事情！

相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 A+B Problem，改用 A-B 了哈哈！

好吧，题目是这样的：给出一串数以及一个数字 $C$ ，要求计算出所有 $A - B = C$ 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

第一行，两个整数 $N, C$ 。

第二行， $N$ 个整数，作为要求处理的那串数。

输出格式

一行，表示该串数中包含的满足 $A - B = C$ 的数对的个数。

输入输出样例

输入 #1

4 1
1 1 2 3

输出 #1

说明/提示

对于 $75\%$ 的数据， $\leq N \leq 2000$ 。

对于 $100\%$ 的数据， $\leq N \leq 2 \times 10^5$ 。

保证所有输入数据都在 $32$ 位带符号整数范围内。

2017/4/29 新添数据两组

代码

方法一

使用map找出记录，相同的数，并在之后相乘

#include <algorithm>
#include <iostream>
#include <cstdio>
#include <map>

using namespace std;

typedef long long ll;

const int N = 2e5;

int n, c;
int a[N + 5];
ll ans;
map<int, int> mp;

int main() {
	scanf("%d%d", &n, &c);
	for (int i = 0; i < n; i++) {
		scanf("%d", a + i);
		mp[a[i]]++;
	}
	sort(a, a + n);
	for (int i = n - 1; i > 1; i--)
		if (mp[a[i]] && mp[a[i] - c]) {
			ans += (ll)mp[a[i]] * mp[a[i] - c];
			mp[a[i]] = 0;
		}
	printf("%lld\n", ans);
	return 0;
}

方法二

使用upper_bound和lower_bound找出相同数字的个数。

#include <algorithm>
#include <iostream>
#include <cstdio>

using namespace std;

typedef long long ll;

const int N = 2e5;

ll n, c;
ll a[N + 5];
ll ans;

int main() {
	scanf("%lld%lld", &n, &c);
	for (int i = 0; i < n; i++)
		scanf("%lld", a + i);
	sort(a, a + n);
	for (int i = 0; i < n; i++) {
		ll num = (upper_bound(a, a + n, a[i] + c) - a) - (lower_bound(a, a + n, a[i] + c) - a);
		ans += num;
	}
	printf("%lld\n", ans);
	return 0;
}