双指针算法是解题过程中常用的优化算法,可将两重for循环通过双指针算法进行优化
大致模板如下:
for(int i = 0, j = 0;i < n; i ++){
while(j < i && check(i, j) j++;
//每道题的具体逻辑
}
时间复杂度O(n),i最多遍历n次,j最多遍历n次
关键点:分析i和j之间满足的单调性
下面看几道具体的双指针算法的例子
问题1 最长连续不重复子序列
问题描述
给定一个长度为 n 的整数序列,请找出最长的不包含重复的数的连续区间,输出它的长度。
输入格式
第一行包含整数 n。
第二行包含 n 个整数(均在 0∼1050∼105 范围内),表示整数序列。
输出格式
共一行,包含一个整数,表示最长的不包含重复的数的连续区间的长度。
数据范围
1≤n≤10^5
输入样例
5
1 2 2 3 5
输出样例
3
代码
#include<iostream>
#include<unordered_map>
using namespace std;
const int N = 100010;
int a[N];
int n;
int main(){
scanf("%d", &n);
for(int i = 0; i < n; i ++) scanf("%d", &a[i]);
int res = 0;
unordered_map<int, int> mydict; //记录某个区间内各个数字出现的次数
//双指针
for(int i = 0, j = 0; i < n; i ++){
mydict[a[i]] ++;
while(mydict[a[i]] > 1){ //新加入的数导致区间出现重复数字, 则重复数字一定是该数
mydict[a[j]] --;
j ++;
}
res = max(res, i - j + 1);
}
cout << res;
return 0;
}
分析
①对每一个i,j表示其左端最远能到达的满足不重复序列的位置
②单调性:随着i的右移,j只可能向右移动
③若新加入的i产生了重复元素,则重复的元素一定为a[i],此时需要向右移动j,直至a[i]在序列中出现的次数为1
那么如何计算某个序列是否出现重复数字了呢?
采用哈希表存储,当a[i]加入序列后,mydict[a[i]] ++,若此时a[i]出现次数大于1,则代表出现了重复元素,右移j即可
问题2 数组元素的目标和
问题描述
给定两个升序排序的有序数组 A 和 B,以及一个目标值 x。
数组下标从 0 开始。
请你求出满足 A[i]+B[j]=x 的数对 (i,j)。
数据保证有唯一解。
输入格式
第一行包含三个整数 n,m,x,分别表示 A 的长度,B 的长度以及目标值 x。
第二行包含 n 个整数,表示数组 A。
第三行包含 m 个整数,表示数组 B。
输出格式
共一行,包含两个整数 i 和 j。
数据范围
数组长度不超过 10^5。
同一数组内元素各不相同。
1≤数组元素≤10^9
输入样例
4 5 6
1 2 4 7
3 4 6 8 9
输出样例
1 1
代码
#include<iostream>
using namespace std;
const int N = 100010;
int a[N], b[N];
int n, m, x;
int main(){
scanf("%d%d%d", &n, &m, &x);
for(int i = 0; i < n; i ++) scanf("%d", &a[i]);
for(int j = 0; j < m; j ++) scanf("%d", &b[j]);
for(int i = 0, j = m - 1; i < n; i ++){
while(j >= 0 && (a[i] + b[j]) > x) j --;
if(j >= 0 && (a[i] + b[j] == x)) printf("%d %d", i, j);
}
return 0;
}
分析
把握两个数组的单调性,i指针将a数组从小到大遍历,j指针将b数组从大到小遍历,寻找a[i] + b[j]出现<=x的位置作为划分点
问题3 判断子序列
问题描述
给定一个长度为 n 的整数序列 a1,a2,…,an 以及一个长度为 m 的整数序列 b1,b2,…,bm。
请你判断 a 序列是否为 b 序列的子序列。
子序列指序列的一部分项按原有次序排列而得的序列,例如序列 {a1,a3,a5} 是序列 {a1,a2,a3,a4,a5}的一个子序列。
输入格式
第一行包含两个整数 n,m。
第二行包含 n 个整数,表示 a1,a2,…,an。
第三行包含 m 个整数,表示 b1,b2,…,bm。
输出格式
如果 a 序列是 b 序列的子序列,输出一行 Yes
。
否则,输出 No
。
数据范围
1≤n≤m≤10^5,
−10^9≤ai,bi≤10^9
输入样例
3 5
1 3 5
1 2 3 4 5
输出样例
Yes
代码
法1 遍历子序列
#include<iostream>
using namespace std;
const int N = 100010;
int a[N], b[N];
int n, m;
int main()
{
scanf("%d%d", &n, &m);
for(int i = 0; i < n; i ++) scanf("%d", &a[i]);
for(int j = 0; j < m; j ++) scanf("%d", &b[j]);
bool check = true;
for(int i = 0, j = 0; i < n; i ++){
while(j < m && a[i] != b[j]) j ++;
// cout << i << " " << j << endl;
j ++;
if(j > m) check = false;
}
if(check) cout << "Yes" << endl;
else cout << "No" << endl;
return 0;
}
法2 遍历父序列
#include<iostream>
using namespace std;
const int N = 100010;
int a[N], b[N];
int n, m;
int main()
{
scanf("%d%d", &n, &m);
for(int i = 0; i < n; i ++) scanf("%d", &a[i]);
for(int j = 0; j < m; j ++) scanf("%d", &b[j]);
int i = 0;
for(int j = 0; j < m; j ++){
if(i < n && a[i] == b[j]) i ++;
}
if(i == n) cout << "Yes" << endl;
else cout << "No" << endl;
return 0;
}