基础算法复习:双指针算法
题目描述
给定一个长度为 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≤105,
−109≤ai,bi≤109
样例
输入样例:
3 5
1 3 5
1 2 3 4 5
输出样例:
Yes
审题
本题要求很简单,要我们判断子序列,显然我们要用两个指针来维护这两个序列,使用双指针算法。
算法
开始时分别用i,j指向小序列a和大序列b的头部,对于a[i],如果当前的b[j]==a[i],那么指针都前进,准备检查下一个数,然后移动j在b中来寻找下一个能匹配的数,最后在每次大循环结束时判断一下是否判断到最后一个数了,如果最后一个数也匹配就break。这样一来如果扫描是通过每位匹配,最后一位匹配完直接break来结束的,j就一定不会==m,如果j==m说明没有某个位置匹配失败。因此最后根据j是否==m输出结果。
C++代码
#include <iostream>
#include <cstdio>
using namespace std;
const int N=100010;
int a[N],b[N];
int main(){
int n,m;
cin>>n>>m;
for(int i=0;i<n;i++) scanf("%d",&a[i]);
for(int i=0;i<m;i++) scanf("%d",&b[i]);
int i=0,j=0;
while(i<n&&j<m){
if(a[i]==b[j]) i++,j++;
while(j<m&&a[i]!=b[j]) j++;
if(a[i]==b[j]&&i==n-1) break;
}
if(j==m) cout<<"No"<<endl;
else cout<<"Yes"<<endl;
return 0;
}