bzoj4892

后缀数组

先开始nc了，觉得自动机做法是指数级的，就写了个后缀数组

具体方法是暴力，枚举起点，然后用lcp向后暴力匹配，如果失配就减少一次，我们一共有3次机会，这样每次匹配复杂度是O(1)的，所以总复杂度是O(nlogn+n)，然后t掉了，交了发别人代码，bzoj怎么那么慢，洛谷跑的飞快。调了很长时间发现sa板子写错了，明明是粘过来的。。。

后缀自动机就是在自动机上匹配，如果不匹配可以随便走，每次匹配完统计就行了

#include<bits/stdc++.h>
using namespace std;
const int N = 2e5 + ;
int n, m, k, len, pos, ans;
char s[N], t[N];
int p[N], a[N], b[N], rank[N], lcp[N], sa[N], mn[N][], mp[], Log[N], tmp[N];
void radix(int *s, int *a, int *b, int n, int m)
{
    int count[N]; memset(count, , sizeof(count));
    for(int i = ; i <= n; ++i) ++count[s[a[i]]];
    for(int i = ; i <= m; ++i) count[i] += count[i - ];
    for(int i = n; i; --i) b[count[s[a[i]]]--] = a[i];
}
void Sa(int *s, int n)
{
    for(int i = ; i <= n; ++i) rank[i] = i;
    radix(s, rank, sa, n, );
    rank[sa[]] = ;
    for(int i = ; i <= n; ++i) rank[sa[i]] = rank[sa[i - ]] + (s[sa[i]] != s[sa[i - ]]);
    for(int k = ; k <= n; k <<= )
    {
        for(int i = ; i <= n; ++i)
        {
            a[i] = rank[i];
            b[i] = i + k <= n ? rank[i + k] : ;
            sa[i] = i;
        }
        radix(b, sa, rank, n, n);
        radix(a, rank, sa, n, n);
        rank[sa[]] = ;
        for(int i = ; i <= n; ++i) rank[sa[i]] = rank[sa[i - ]] + (a[sa[i]] != a[sa[i - ]] || b[sa[i]] != b[sa[i - ]]);
    }
}
void Lcp(int *s, int n)
{
    int h = ;
    for(int i = ; i <= n; ++i) rank[sa[i]] = i;
    for(int i = ; i <= n; ++i)
    {
        int j = sa[rank[i] - ];
        if(rank[i] <= ) continue;
        if(h > ) --h;
        for(; i + h <= n && j + h <= n; ++h) if(s[i + h] != s[j + h]) break;
        mn[rank[i] - ][] = h;
    }
    for(int j = ; j <= ; ++j)
        for(int i = ; i + ( << j) -  <= n; ++i)
            mn[i][j] = min(mn[i][j - ], mn[i + ( << (j - ))][j - ]);
}
int query(int l, int r)
{
    l = rank[l];
    r = rank[r];
    if(l > r) swap(l, r);
    --r;
    int x = Log[r - l + ];
    return min(mn[l][x], mn[r - ( << x) + ][x]);
}
int main()
{
    int T;
    scanf("%d", &T);
    mp['A'] = ;
    mp['G'] = ;
    mp['C'] = ;
    mp['T'] = ;
    for(int i = ; i < N; ++i) Log[i] = Log[i >> ] + ;
    while(T--)
    {
        ans = ;
        scanf("%s%s", s + , t + );
        len = ;
        n = strlen(s + );
        m = strlen(t + );
        for(int i = ; i <= n; ++i) p[++len] = mp[s[i]];
        p[++len] = ;
        pos = len + ;
        for(int i = ; i <= m; ++i) p[++len] = mp[t[i]];
        Sa(p, len);
        Lcp(p, len);
        for(int i = ; i <= n - m + ; ++i)
        {
            int tmp = m, cnt = , p1 = i, p2 = pos;
            while(tmp > )
            {
                int x = query(p1, p2);
                tmp -= x;
                p1 += x;
                p2 += x;
                if(tmp <= ) break;
                while(cnt >=  && p[p1] != p[p2] && p1 <= n && p2 <= len)
                {
                    ++p1;
                    ++p2;
                    --tmp;
                    --cnt;
                }
                if(cnt <  || p2 > len || p1 > n) break;
            }
            if(cnt >=  && tmp <= ) ++ans;
        }
        printf("%d\n", ans);
    }
    return ;
}