双指针
双指针主要用于遍历一组数据(如一个数组 / 链表)或分别遍历两组数据(如两个数组 / 链表),将一些需要双层循环的算法变为单层循环减少时间复杂度,通常有滑动窗口、快慢指针。
接下来,我们结合一些例题来理解。
遍历单组数据
滑动窗口
常用来解决一些连续区间的问题。
例如:一题洛谷P1147 连续自然数和
最简单的思路就是用两层for循环,第一层 i 从 1 开始遍历,第二层则为 j = i,j 开始向后遍历相加,直到和大于或等于 M(等于则输出 i 和 j) 则第二层结束,然后第一层循环 i += 1 后继续进行第二层的循环 j = i ,j 向后遍历相加,直到和大于或等于 M,这样重复上述操作,时间复杂度为 O(n2)。
这个算法可以优化一下,使用双指针,定义两个指针 i 和 j ,刚开始都指向数组中的第一个元素,通过将 i 或 j 向右移动来限定范围,这个限定的范围也可称为窗口,通过移动 i 和 j 来移动这个窗口,将这个窗口中的数全部相加看是否满足条件,大于 M 则指针 i 右移缩小范围,小于 M 则指针 j 右移扩大范围,等于 M 则输出 i 和 j 。时间复杂度为O(n)。
#include <iostream>
using namespace std;
int M;
int a[2000001];
int main()
{
cin>>M;
int sum = 0;
int i = 1,j = 1; // 定义双指针来限定窗口
while(j <= M/2 + 1){
a[j] = j;
sum += a[j];
if(sum > M){
while(sum > M){
sum -= a[i];
i++;
}
if(sum == M){
cout<<i<<" "<<j<<endl;
}
}
else if(sum == M){
cout<<i<<" "<<j<<endl;
}
j++;
}
return 0;
}
快慢指针
常用来解决一些带环的问题
例如:一题力扣141.环形链表
通过定义一个两个指针,同时遍历这个链表,一个指针每次向后遍历1个位置(即慢指针),另一个每次向后遍历2个位置(即快指针),若两个指针还能相遇则说明有环,有一个指针遍历到NULL则无环。
bool hasCycle(ListNode *head) {
if(!head){
return false;
}
ListNode *slow = head,*fast = head; // 定义快慢指针
while(fast->next && fast->next->next){
slow = slow->next; // 慢指针
fast = fast->next->next; // 快指针
if(fast == slow){ // 双指针相遇,则有环
return true;
}
}
return false;
}
遍历两组数据
定义两个指针 i、j 分别用来遍历 num1 和 num2 两个数组,比较指针 i 和 j指向的元素来决定元素的位置,并将对应指针继续右移遍历比较,直到两个数组全部遍历完。