题目;
前言:
看见题目有些人可能想到用双重循环历遍数据,如果满足a-b=c就sum++.想法是对的但是我们看一下题目的数据范围就会明白在o(n^2)的时间复杂度下肯定会TLE.那么是否有时间复杂度更低的算法来解决呢?
二.双指针
在这里我不过多展开,只简单介绍一下与本题相关的部分.
题目要求A-B=C.满足这要求的A与B,我们可以这样看,将原式改为A-C=B.易知每一个A都有一个B对应,我们只需要从头开始历遍给出的数据当作A,找此时是否在数据中存在对应的B(B=A-C)如果存在就sum++,同时我们用两个变量l,r(双指针),框定B在数据中存在的范围,此时是不是感觉什么乱七八糟的听不懂,没关系听不懂很正常,接下来我通过一个例子,几个示意图演示一下你就懂了,上图!
假设给出数据为:1,2,2,3,4,4,5.C=1;将数据放入数组a[]中如图:
我们取A=5时作例子讲解,此时B=A-C=4,那么我们怎么样找到1223445,中的4并记录下来呢?l,r初始化为1,当a[l]大于等于B时停下来,当a[r]刚好大于B时停下来,这样子会达到如下图效果:
此时l=5,r=7,那么我们可知当A=5时存在两个符合题意的数对sum=r-l=2,显而易见可以在数据中找到两个4,所以思路直观且正确,以上只是A=5时的情况.同理,历遍A的取值,对每一个A用相同的方法求出当下的数对的数量存储在sum中,最后输出sum即可,当然以上说的是理想情况,假如当下A算出的B在数组中找不到那么通过一些手段可以处理,不是重点留在下面代码中说明.
三.AC代码
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+5;
int a[N];
int n,c,B;
long long sum;
int main()
{
int l=1,r=1;
cin>>n>>c;
for(int i=1;i<=n;i++)
{
cin>>a[i]; 循环输入,将数据输入数组,a[0]不要从1开始方便算
}
sort(a+1,a+n+1); 排序,保证数据单调性
for(int i=1;i<=n;i++) 循坏历遍A的取值
{
B=a[i]-c; 算出对应的B,然后找数据中是否有这个B
while(l<=n&&a[l]<=B)l++; 左指针,上文中以A=5为例讲解中 是指向第一个出现的4的指针
while(r<l&&a[r]<B)r++; 右指针,上文中是指向5的指针
sum+=l-r; 算出有多少个相同B,同上文例子
}
cout<<sum;
return 0;
}
四.最后
如果有问题可以留言,有错误欢迎指正.
最后大家学习顺利,永远不要放弃,分享一句我很喜欢的话:
计算机的所有东西都是人做出来的,别人能写出来的,我也一定能想出来,在计算机里头没有任何黑魔法,所有的东西只不过是我现在不知道而已,总有一天,我会把所有的细节,所有的内部东西全都搞明白的. -------翁恺
加油!!