三种语言实现双指针判断子序列(C++/Python/Java)

题目

给定一个长度为 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≤1e5,
−1e9≤ai,bi≤1e9

输入样例:

3 5
1 3 5
1 2 3 4 5

输出样例:

Yes

C++

#include <bits/stdc++.h>

using namespace std;

const int N = 1e5;
int a[N], b[N];

int main()
{
    int n,m;
    cin >> n >> m;
    for(int i = 1; i <= n; i++)
        cin >> a[i];
    for(int i = 1; i <= m; i++)
        cin >> b[i];
    int i = 1, j = 1;
    for(; i <= n && j <= m; j++)
    {
        while(i <= n && j <= m && a[i] != b[j])
            j++;
        if(i <= n && j <= m && a[i] == b[j])
            i++;
    }
    if(i > n)
        cout << "Yes";
    else
        cout << "No";
}

Python

n, m = map(int, input().split())
 
a = list(map(int, input().split()))
b = [int(x) for x in input().split()]
 
i = 0
j = 0
 
while i < n and j < m:
    while i < n and j < m and a[i] != b[j]:
        j += 1
    if i < n and j < m and a[i] == b[j]:
        i += 1
    j += 1

if i >= n:
    print("Yes")
else:
    print("No")

Java

import java.util.*;
 
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int[] a = new int[n+1];
        int[] b = new int[m+1];
        for(int i = 1; i <= n; i++)
            a[i] = sc.nextInt();
        for(int i = 1; i <= m; i++)
            b[i] = sc.nextInt();
        int i = 1, j = 1;
        for(; i <= n && j <= m; j++)
        {
            while(i <= n && j <= m && a[i] != b[j])
                j++;
            if(i <= n && j <= m && a[i] == b[j])
                i++;
        }
        if(i > n)
            System.out.print("Yes");
        else
            System.out.print("No");
    }
}
posted @ 2024-08-01 11:27  火车驶向云外0218  阅读(21)  评论(0)    收藏  举报