洛谷 P1102 A-B数对 (双指针)

题目大意:

在An中,我们找出Ai - Aj == c的所有数对的个数。

n<=1e6

解题思路:

方法很多这里用一个巧妙的双指针完成,这种方法应该是代码量最短的。首先,我们可以对A从小到大进行排序。接着我们枚举指针j,找出Aj - Ai2>=c中的最靠右的位置,同时找出Aj - Ai1<=c中的最靠左的位置,当Aj - Ai1 == c时,两个的交集的个数就是当前枚举j时候的答案贡献数。

废话:

哎,在做指针操作的时候,假如用while操作,注意指针是否有多加一次的情况啊。

#include <bits/stdc++.h>
#define int long long 
using namespace std;
int32_t main(){
	int n,c;cin>>n>>c;
	vector<int> A(n);
	for(int i=0;i<n;i++)cin>>A[i];
	sort(A.begin(),A.end());
	int i=0,j=0;
	int i1=0,i2=0;
	int ans=0;
	int suc;
	for(;j<n;j++){
		while(i1<j && A[j]-A[i1]>c)suc=1,i1++;
		suc=0;
		while(i2<j && A[j]-A[i2]>=c)suc=1,i2++;
		if(suc&&A[j]-A[i2]<c)i2--;
		if(A[j]-A[i1]==c){
			ans+=i2-i1+1>0?i2-i1+1:0;
		}
	}
	cout<<ans<<endl;
	return 0;
}
发布了171 篇原创文章 · 获赞 4 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/FrostMonarch/article/details/103554568
今日推荐