【双指针】

最新推荐文章于 2024-03-08 21:56:34 发布

玛卡巴卡要ac

最新推荐文章于 2024-03-08 21:56:34 发布

阅读量66

点赞数

文章标签： c++ 算法

本文链接：https://blog.csdn.net/weixin_62651239/article/details/123189879

版权

这篇博客介绍了一种解决简单数对问题的算法，即在给定的一串数字中找出所有满足A-B=C的数对，并且计算其数量。文章通过排序和双指针技术优化了搜索过程，避免了暴力求解可能导致的整数溢出问题。示例代码展示了如何在O(n log n)的时间复杂度内完成这一任务。

摘要由CSDN通过智能技术生成

简单A-B
描述
这是一道简单题,给出一串数以及一个数字C，要求计算出所有A - B = C的数对的个数（不同位置的数字一样的数对算不同的数对）。
输入
输入共两行。
第一行，两个整数N, C。1=<N<=2e5 , C>=1
第二行，N个整数，作为要求处理的那串数。

输出
一行，表示该串数中包含的满足A - B =C的数对的个数。

输入样例 1
4 1
1 1 2 3
输出样例 1
3

输入样例 2
6 1
1 1 1 2 2 2
输出样例 2
9

本题值得注意的是,有可能暴int

#include <bits/stdc++.h>
using namespace std;
int const N=200010;
typedef long long ll;
int a[N],n,c;
ll res;
int main()
{
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    //节约cin,cout时间;
    cin>>n>>c;
    for(int i=1; i<=n; i++)
        cin>>a[i];
    sort(a+1,a+1+n);
    int kk=2,k=2;
    for(int i=1; i<=n; i++)
    {
        while(kk<=n&&a[kk]-a[i]<=c)
            kk++; //因为是有序的,下次寻找从kk开始;减少不必要的枚举
        while(k<=n&&a[k]-a[i]<c)
            k++;
        if(a[kk-1]-a[i]==c&&a[k]-a[i]==c)
        {
            
            res+=kk-k;
        }
    }
    cout<<res<<endl;
    return 0;
}