算法刷题总结—双指针算法

双指针算法是解题过程中常用的优化算法，可将两重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;
}