A-B 数对
思路:可以先从小到大遍历出数组中的每一个数,相当于固定B,然后二分去找是否有满足条件的A,,因为会有重复的数,所以满足条件的A可能不止一个,所以可以像基础课例题中的“数的范围”中那样,找到一个区间,最后在求和就可以得出结果。注意:题目说是一个正整数数列,但是测试数据里不是有序的,所以要先排序。
题目背景
出题是一件痛苦的事情!
相同的题目看多了也会有审美疲劳,于是我舍弃了大家所熟悉的 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 \leq N \leq 2000$。
对于 $100\%$ 的数据,$1 \leq N \leq 2 \times 10^5$,$0 \leq a_i <2^{30}$,$1 \leq C < 2^{30}$。
2017/4/29 新添数据两组
#include<iostream>
#include<algorithm>
#include<stdio.h>
using namespace std;
typedef long long LL;
const int N = 200010;
int n;
LL c;
LL a[N];
int main()
{
scanf("%d%lld", &n, &c);
for (int i = 0; i < n; i++) scanf("%lld", &a[i]);
sort(a, a + n);
LL cnt = 0;
for (int i = 0; i < n; i++)
{
int l = i, r = n - 1;
while (l < r)
{
LL mid = l + r >> 1;
if (a[mid] - a[i] >= c) r = mid;
else l = mid + 1;
}
if (a[l] - a[i] != c) continue;
else
{
int ll = i, rr = n - 1;
while (ll < rr)
{
LL mid = ll + rr + 1 >> 1;
if (a[mid] - a[i] <= c) ll = mid;
else rr = mid - 1;
}
cnt += ll - l + 1;
}
}
printf("%lld", cnt);
return 0;
}