洛谷 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 
#include 
#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;
}

 提交结果:

洛谷 P1102 A-B 数对 题解_第1张图片

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

话不多说,直接上代码:

#include 
#include 
#define N 200005
using namespace std;

typedef long long ll;

ll n, c, a[N], ans;
map  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;
}

提交结果:

洛谷 P1102 A-B 数对 题解_第2张图片

尾声

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

最后,祝愿您在 OI 的路上一路顺风!

你可能感兴趣的:(c++)