P4465 [国家集训队] JZPSTR 题解

bitset 乱搞字符串匹配经典题。设原串为 $s$ 串。

维护 $p_i$ 表示 $i$ 字符的出现位置集合，这个在修改时容易用位运算做到。

考虑怎么匹配 $z$ 串。扫描 $z$ 串，设当前扫到第 $i$ 位，维护 $u$ 表示能匹配前 $i$ 位的结束位置集合。

考虑从 $i-1$ 位扫到 $i$ 位时，怎么更新 $u\gets u'$。

首先如果 $x$ 能匹配前 $i$ 位，那么 $x-1$ 一定能匹配前 $i-1$ 位，即 $x\in u'\Rightarrow x-1\in u$。

其次如果 $x$ 能匹配前 $i$ 位，那么 $s_x=z_i$，即 $x\in u'\Rightarrow x\in p_{z_i}$。

所以 $u'=\{i+1|i\in u\}\cap p_{z_i}$，更新 $u\gets u'$ 即可。容易位运算做到。

最终 $u$ 就是能匹配 $z$ 串的位置集合。

诶我怎么最优解了

#include <bitset>
#include <cstdio>
#include <cstring>
using namespace std;
int m, o, x, y, z;
char s[1000050];
bitset<1000001> u, v, p[10];
int main()
{
    scanf("%d", &m);
    while (m--)
    {
        scanf("%d%d", &o, &x);
        switch (o)
        {
        case 0:
            scanf("%s", s + x);
            y = strlen(s + x);
            u.set();
            u <<= x;
            v = ~u;
            for (int i = 0; i < 10; ++i)
                p[i] = p[i] & v | (p[i] & u) << y;
            for (int i = x; i < x + y; ++i)
                p[s[i] - '0'].set(i);
            break;
        case 1:
            scanf("%d", &y);
            u.set();
            v = ~(u << x);
            u <<= y;
            for (int i = 0; i < 10; ++i)
                p[i] = p[i] & v | (p[i] & u) >> y - x;
            break;
        case 2:
            scanf("%d%s", &y, s);
            z = strlen(s);
            u.set();
            u = u << x & ~(u << y);
            v = p[s[0] - '0'] & u;
            for (int i = 1; i < z; ++i)
                v = v << 1 & p[s[i] - '0'];
            printf("%d\n", (v & u).count());
            break;
        }
    }
    return 0;
}