双指针
问题引入
例1
对一个给定的正整数 M,求出所有的连续的正整数段(每一段至少有两个数),这些连续的自然数段中的全部数之和为 M。(M <= 2e6)
定义
双指针是一种灵活的技巧,简单来说你需要同时维护两个指针,指向序列,链表(甚至更复杂的数据结构)上的位置或结点, 通过同向或相向移动来维护信息。
应用
维护区间信息
如果不与其他算法结合,双指针最简单的应用就是维护具有单调性的区间信息,如正整数的和,正整数的积等。
对于上文中的问题是典型的双指针维护区间信息的应用
我们可以维护两个指针l, r,来指向我们维护区间的左右端点,记sum为我们维护的区间的和。若sum < M, 我们需要新增元素到我们所维护的区间, 我们将r向后移动,sum 加上新增元素的值。若sum > M, 我们将l 向后移动, sum减去出去元素的值。
时间复杂为O(n), 因为指针不会回退,每个元素最多被扫描到两次
void solve(){
cin >> M;
int sum = 3;
for(int i = 1, j = 2; i <= M/2;){
while(sum < M) {
j++;
sum += j;
}
while(sum > M){
sum -= i;
i++;
}
if(sum == M){
cout << i <<" " << j << endl;
sum -= i;
i++;
}
}
}
利用序列有序性
问题 : 例2给定一个非严格单调递增的正整数序列,以及正整数x,求序列中有多少对数字ai与aj(i<j,i或j有一个不同即为不同的数字对),其差aj-ai 的值为x。
过程
我们可以先试试二分的做法,对于每个元素我们需要找到key = x - ai,我们再在序列中找到key的左右端点即可,ans += r - l + 1。
双指针做法
双指针做法相比二分做法少了O(log)的复杂度。我们维护两个指针,对于每个元素,我们使用类似于二分的做法,r的指针不断后移直到指向key的右端,l指针不断后移直到指向key的左端。因为元素都是递增的,指针不会回退,所以我们可以在O(n)的时间复杂完成。
int n, x, a[N];
void solve(){
cin >> n >> x;
for(int i = 1; i <= n; i++) cin >> a[i];
sort(a + 1, a + 1 + n);
ll sum = 0;
int i = 1, j = 1;
for(int k = 1; k <= n; k++){
while(i <= n && a[i] - a[k] < x) i++;
while(j <= n && a[j] - a[k] <= x) j++;
sum += j - i;
}
cout << sum << endl;
}