P1102 A-B 数对(双指针解法)

最新推荐文章于 2024-08-12 18:10:44 发布

CodeWizard～

最新推荐文章于 2024-08-12 18:10:44 发布

阅读量1.2k

点赞数 27

分类专栏：双指针和前缀和在算法竞赛中的高效应用文章标签：算法数据结构 c++ 开发语言 c语言

本文链接：https://blog.csdn.net/2301_80662593/article/details/140389457

版权

双指针和前缀和在算法竞赛中的高效应用专栏收录该内容

5 篇文章 1 订阅

订阅专栏

P1102 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

提示

对于 $75\%$ 的数据， $\leq N \leq 2000$ 。

对于 $100\%$ 的数据， $\leq N \leq 2 \times 10^5$ ， $\leq a_i <2^{30}$ ， $\leq C < 2^{30}$ 。

2017/4/29 新添数据两组

容易错误的code：

比如：两个相同的元素会直接出现漏加的情况。

#include<iostream>
#include<algorithm>
using namespace std;
const int N=2e5+10;
int n,c;
int q[N];
int main(){
    cin>>n>>c;
    for(int i=1;i<=n;i++){
        cin>>q[i];
    }
    sort(q+1,q+1+n);
    long long cnt=0;
    for(int i=1,j=1;i<=n;i++){
        while(j<=i&&q[i]-q[j]>=c){
            if(q[i]-q[j]==c) cnt++;
            j++;
        }
    }
    cout<<cnt;
}

全解code：

A-B=C开始时，A是i，B是j，B开始就是最大，当A-B小于C时，A-B中可以增大A或者减少B，当增大A时，结果只可能小于等于0，当减小B时，结果还是只可能小于等于0，所以对撞指针不可以，本题应该使用快慢指针。

#include<iostream>
#include<algorithm>
using namespace std;
const int N=2e5+10;
int n,c;
int q[N];
int main(){
    cin>>n>>c;
    for(int i=1;i<=n;i++){
        cin>>q[i];
    }
    sort(q+1,q+1+n);
    long long cnt=0;
    for(int i=1,j=1,k=1;i<=n;i++){
        while(j<=i&&q[i]-q[j]>=c){
            j++;
        }
        while(k<=i&&q[i]-q[k]>c){
            k++;
        }
        cnt+=j-k;
    }
    cout<<cnt;
}