【洛谷】P5826 【模板】子序列自动机

题目地址：

https://www.luogu.com.cn/problem/P5826

题目描述：
给定一个长度为$n$的正整数序列$a$，有$q$次询问，第$i$次询问给定一个长度为$L_i$的序列$b_i$，请你判断$b_i$是不是$a$的子序列。序列$a$和所有$b_i$中的元素都不大于一个给定的正整数$m$。本题中，若$x$是$y$的子序列，则等价于存在一个单调递增序列$z$，满足$\mid z\mid =\mid x\mid$，$z_{\mid x\mid }\le \mid y\mid$，且$\forall i\in [1,\mid x\mid ],y_{z_i}=x_i$。其中$\mid x\mid ,\mid y\mid ,\mid z\mid$分别代表序列$x,y,z$的长度，$x_i,y_i,z_i$分别代表序列$x,y,z$的第$i$项。

输入格式：
每个测试点有且仅有一组数据。输入的第一行是四个用空格隔开的整数，分别代表type，$n$，$q$，$m$。其中type代表测试点所在的子任务编号，其余变量的含义见题目描述。输入的第二行是$n$个用空格隔开的整数，第$i$个数字代表序列$a$的第$i$个元素$a_i$。第$3$行至第$(q+2)$行，每行代表一次询问。第$(i+2)$行的输入格式为：第$(i+2)$行的行首有一个整数$l_i$，代表第$i$次询问的序列长度。一个空格后有$l_i$个用空格隔开的整数。该行的第$(j+1)$个整数代表序列$b_i$的第$j$个元素$b_{i,j}$。

输出格式：
对于每次询问，输出一行一个字符串，若给定的序列是$a$的子序列，则输出Yes，否则输出No。

序列自动机是一种有限状态自动机，给定一个字符串$s$，该自动机根据$s$构建，可以用来判断另一个字符串$t$是否是$s$的子序列，并且保证在 O ( min ⁡ { l s , l t } ) O(\min\{l_s,l_t\}) O(min{ls​,lt​})时间内完成（当然一般情况下$t$的长度远小于$s$的长度，所以可以认为是$O(l_t)$）。该自动机可以由一个二维数组$A$表示，$A[k][c]$表示当匹配到$s$的第$k$个字符处的时候（这里下标从$1$开始，是为了方便用数组实现），接下来如果读入了字符$c$，应该跳转到哪一个位置。初始位置在$0$，接下来每读入一个字符，就会跳转到该字符在当前位置之后第一次出现的位置；如果该字符之后不存在，则规定跳转到了$0$（所以位置$0$可以标记不是子序列的情况）。例如对于字符串"abcb"，我们要判断"abb"是否是其子序列，初始位置为$0$，先读入$a$，那么$A[0][a]=1$，因为$a$第一次出现在下标$1$的位置，就跳到了$1$，接下来$A[1][b]=2$，因为之后第一次出现$b$是在下标$2$的位置，就跳到了$2$，再接下来$A[2][b]=4$，因为在$2$之后第一次出现$b$的位置是$4$，再接下来"abb"遍历完了，说明其是个子序列。再比如判断"aba"是否是其子序列，则跳转过程是$0\to A[0][a]=1\to A[1][b]=2\to A[2][a]=0$，即遍历到最后一个$a$的时候跳转到了$0$，则说明不是子序列。子序列自动机的建立过程可以从后向前用递推来实现，初始条件$A[l_s][\alpha ]=0$，代表当前已经在$s$的最后一个字符的位置了，后面再也没有字符了；接着向前递推：$$A[k][\alpha ]=\{\begin{array}{l}A[k+1][\alpha ],\alpha \ne s[k+1]\\ k+1,\alpha =s[k+1]\end{array}$$这里的$s$的下标是从$1$开始的。这个递推关系式很好理解，如果某个位置的后一个位置就是当前输入字符，那显然接下来就要跳转到后一个位置；否则按照定义，跳转到从后一个位置之后的第一次出现的位置。但是本题如果直接开数组或者unordered_map来做爆空间。可以用vector动态开空间，然后用二分来解决。每个vector存某个数出现的所有位置，跳转的时候找大于当前位置的最前的匹配的位置。代码如下：

#include <iostream>
#include <vector>
using namespace std;

const int N = 1e5 + 10;
int n, q, m;
int a[N];
vector<int> pos[N];

// 找在i之后第一次出现x的位置，若不存在则返回0
int get_larger(int x, int i) {
    if (!pos[x].size()) return 0;

    int l = 0, r = pos[x].size() - 1;
    while (l < r) {
        int mid = l + (r - l >> 1);
        if (pos[x][mid] > i) r = mid;
        else l = mid + 1;
    }

    return pos[x][l] > i ? pos[x][l] : 0;
}

int main() {
    int type;
    scanf("%d%d%d%d", &type, &n, &q, &m);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        pos[a[i]].push_back(i);
    }

    while (q--) {
        int len;
        scanf("%d", &len);
        bool success = true;
        int cur = 0;
        while (len--) {
            int x;
            scanf("%d", &x);
            cur = get_larger(x, cur);
            if (!cur || !success) success = false;
        }

        if (success) printf("Yes\n");
        else printf("No\n");
    }

    return 0;
}

预处理时间复杂度$O(n)$，每次询问时间复杂度$O(m\log n)$，$m$是询问的序列长度，空间$O(n)$。