2816. 判断子序列 - AcWing题库

Question

Content

给定一个长度为 n 的整数序列 a₁,a₂,…,a_n 以及一个长度为 m 的整数序列 b₁,b₂,…,b_m 。

请你判断 a 序列是否为 b 序列的子序列。

子序列指序列的一部分项按原有次序排列而得的序列，例如序列 {a₁,a₃,a₅} 是序列 {a₁,a₂,a₃,a₄,a₅} 的一个子序列。

输入格式

第一行包含两个整数 n,m 。

第二行包含 n 个整数，表示 a₁,a₂,…,a_n 。

第三行包含 m 个整数，表示 b₁,b₂,…,b_m 。

输出格式

如果 a 序列是 b 序列的子序列，输出一行 Yes。

否则，输出 No。

数据范围

1 ≤ n ≤ m ≤ 10⁵ ,

−10⁹ ≤ a_i,b_i ≤ 10⁹

输入样例：

3 5
1 3 5
1 2 3 4 5

输出样例：

Yes

Solution

Java

import java.util.Scanner;
import java.util.stream.IntStream;

public class Main {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int m = scanner.nextInt();

        int[] a = IntStream.range(0, n).map(e -> scanner.nextInt())
                .toArray();
        int[] b = IntStream.range(0, m).map(e -> scanner.nextInt())
                .toArray();

        boolean subsequence = isSubsequence(a, b);

        System.out.println(subsequence ? "Yes" : "No");
    }

    private static boolean isSubsequence(int[] a, int[] b) {
        int n = a.length, m = b.length;

        int i = 0;
        for (int j = 0; j < m; j++) {
            if (i < n && a[i] == b[j]) {
                i++;
            }
        }

        return i == n;
    }

}

Summary

Java program checks if array 'a' is a subsequence of 'b', outputs "Yes" or "No."