洛谷P1102 A-B 数对(双指针解法,ac代码带注释)

题目描述

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

输入格式

输入共两行。

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

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

输出格式

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

题解

思路

这道题首先我们可以先将这一串数字排序,然后我们把A-B=C转换成A-C=B来看,其实也就是对于每一个B来说它对应的A都有一段连续的区间

现在要考虑的是如何找到这一区间,我们要找到这一个区间的左端点和右端点,在我们将这一个数组排序了之后我们发现了左端点和右端点是随着B的增大而增大的,它们是单调不降的,所以我们可以用两个指针来表示左端点和右端点.

对于每一个B我们找到它所对应的A的区间,也就是说我们维护两个端点r1,r2.每次将r1右移到满足

a[r1]-B<=C的下一位,将r2右移到满足a[r2]-B<c的最后一位.

此时如果a[r1]-B==c&&a[r2]-B==c,那么从r1到r2都是满足条件的

我们只要让cnt+=r1-r2就可以的出答案

代码

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
#define maxsize 10;
int main(){
     int n,c;
     long long cnt=0;
     scanf("%d %d",&n,&c);
     int a[n+1];
     for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
     sort(a+1,a+n+1); //将数组排序
       int r1=1,r2=1;//初始化指针
     for(int i=1;i<=n;i++){
     
        while(r1<=n&&a[r1]-a[i]<=c) r1++; //移动r1和r2到满足条件
        while(r2<=n&&a[r2]-a[i]<c) r2++;
        cnt+=r1-r2;
     } 
    /* sort之后使用双指针,r1,r2只做一次遍历,
     时间复杂度仅有sort函数的nlogn
     a[r1]----a[r2]是A,a[i]是B
     */  

     printf("%lld",cnt);
}