洛谷 P1102 A-B 数对 题解

目录

题目

题目传送门

题解

尾声

---

题目

A-B 数对

题目背景

出题是一件痛苦的事情！

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

题目描述

给出一串正整数数列以及一个正整数 C，要求计算出所有满足 A - B = C 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

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

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

输出格式

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

样例 #1

样例输入 #1

4 1
1 1 2 3

样例输出 #1

3

提示

对于 75\% 的数据，1 ≤ N ≤ 2000。

对于 100\% 的数据，1 ≤ N ≤ 2×10^5，0 ≤ a[i] < 2^30，1 ≤ C < 2^30。

2017/4/29 新添数据两组

题目传送门

A-B 数对 - 洛谷

题解

本题有一个容易想到的算法（枚举任意两个数，判断是否满足 a - b == c），但是TLE：

#include <iostream>
#include <algorithm>
#define N 200005
using namespace std;

int n, c, a[N], ans;

int main() {
    cin >> n >> c;
    for(int i = 1; i <= n; ++i)
        cin >> a[i];
    sort(a + 1, a + n + 1);
    for(int i = 1; i <= n; ++i)
        for(int j = i + 1; j <= n; ++j)
            if(a[j] - a[i] == c)
                ++ans;
    cout << ans << endl;

    return 0;
}

 提交结果：

我们有一个新的思路，记录每个数字出现的次数，最后累加所有 a[i] - c（也就是 b） 的次数（a - b = c → a - c  = b）就可以了

话不多说，直接上代码：

#include <iostream>
#include <map>
#define N 200005
using namespace std;

typedef long long ll;

ll n, c, a[N], ans;
map <ll, ll> m;									// 记录数字出现次数 

int main() {
	cin >> n >> c;
	for(ll i = 1; i <= n; ++i) {				// 循环输入并记录次数 
		cin >> a[i];
		++m[a[i]];								// a[i] 的出现次数加 1 
		a[i] -= c;								// a - b = c -> a - c = b，最后将 a[i] - c（b） 的个数累加起来就可以了 
	}
	for(ll i = 1; i <= n; ++i)
		ans += m[a[i]];							// 累加有多少个 b 
	cout << ans << endl;						// 输出最终答案 
	
	return 0;
}

提交结果：

尾声

如果本博客对您有帮助的话，please点个赞，加个关注！

最后，祝愿您在 OI 的路上一路顺风！