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 \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 新添数据两组
思路
考虑到A-B=C情况可以转化为A-C=B,此时先将输入的情况排序,枚举每一个A的情况,如果存在第一个符合这一条件的数,定为左边;再找最后一个符合这一条件的数定为右边,ans加上r-l即可。
完整代码
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<vector>
#include<queue>
#include<set>
#include<unordered_map>
using namespace std;
int n,c;
int a[200005];
int main(){
long long ans=0;
cin>>n>>c;
for(int i=1;i<=n;i++){
cin>>a[i];
}
sort(a+1,a+n+1);
int r1=1;
int r2=1;
for(int i=1;i<=n;i++){
while(r1<=n&&a[r1]-a[i]<c)
++r1;
while(r2<=n&&a[r2]-a[i]<=c)
++r2;
if(a[r1]-a[i]==c&&a[r2-1]-a[i]==c)
ans+=r2-r1;
}
cout<<ans;
return 0;
}
标签:正整数,p1102,int,数对,样例,leq,include
From: https://www.cnblogs.com/simdow/p/16942618.html