2018 ICPC北京 H ac自动机

n=40的01串,求有多少m=40的01串中包含它,包含的定义是存在子串有至多一个字符不相同

600组n=15的数据 15组n=40的数据,所以我们只能支持n^5的算法。

陷入两个比较有意思的坑:

1 如果手动构建fail树,建立两个并行的串,左串代表当前未使用那一个可以不相同的字符的名额,右串代表已经使用了这个名额,那么按照bfs m步的想法,左串可以通过“使用名额”到达右串,也可以通过“不使用名额”走一个fail,而右串一旦失配,只能在右串的对应位置进行fail。这样跑一遍矩阵乘法,复杂度是(2*n)^3

但是不对,因为假设右串失配了,可能距离一个点从左串跳过来已经过了很久了,那个名额已经可以再次使用了,所以它可以再跳回左串。这样就会导致我们失去一些答案。

2 如果插入n+1个串,代表原串,和原串第i个位置不同的串,这样点是n^3,那么矩阵乘法是n^6,还有一个logm的快速幂时间,复杂度会超

陷入了误区:快速幂一定比直接转移快。快速幂的快体现在将m转移到logm,但是二维矩阵的自乘是(点数^3)的,加一个快速幂的优化是从m(n*n)到logm(n*n*n)。

所以暴力插入n+1个串建树,暴力(n*n)^2的枚举边,跑m次转移。

L read() {L x;scanf("%lld" , &x) ; return x ; } ;

char s[50] ;

L c[40 * 40 + 5] ;
L a[40 * 40 + 5][40 * 40 + 5] ;
L b[40 * 40 + 5] ;
L n , m ;

struct AC{
    L next[40 * 40 + 5][2] , fail[40 * 40 + 5] , en[40 * 40 + 5] ;
    L ro , to ;
    L newnode() {
        to++;
        for(L i = 0 ; i < 2 ; i ++ ) next[to][i] = -1 ;
        en[to] = 0 ;
        return to ;
    }
    void init() {
        to = -1 ;
        ro = newnode() ;
    }
    void inse() {
        L now = ro ;
        for(L i = 1 ; i <= n ; i ++ ) {
            if(next[now][s[i]-'0'] == -1) next[now][s[i]-'0'] = newnode();
            now = next[now][s[i]-'0'];
        }
        en[now]++;
    }
    void build() {
        queue<int>q;
        fail[ro]=ro;
        for(L i=0;i<2;i++){
            if(next[ro][i] == -1) next[ro][i]=ro;
            else {
                fail[next[ro][i]]=ro;
                q.push(next[ro][i]);
            }
        }
        while(!q.empty()){
            L now=q.front();q.pop();
            for(L i=0;i<2;i++){
                if(next[now][i]==-1)next[now][i] = next[fail[now]][i] ;
                else {
                    fail[next[now][i]] = next[fail[now]][i];
                    q.push(next[now][i]);
                }
            }
        }
    }
    void gao() {
        inse() ;
        for(int i = 1 ; i <= n ; i ++ ) {
            s[i] = '0' + ((s[i] - '0') ^ 1) ;
            inse();
            s[i] = '0' + ((s[i] - '0') ^ 1) ;
        }
        build() ;
        for(int i = 0 ; i <= to ; i ++ ) {
            for(int j = 0 ; j <= 1 ; j ++ ) {
                int v = next[i][j];
                if(en[i] == 0)
                    a[i][v] ++ ;
            }
        }
        for(int i = 0 ; i <= to ; i ++ ) {
            if(en[i] != 0) a[i][i] += 2 ;
        }
        b[0] = 1 ;
        for(int i = 1 ; i <= m ; i ++ ) {
            memset(c, 0 , sizeof(c)) ;
            for(int u = 0 ; u <= to ; u ++ ) {
                for(int v = 0 ; v <=  to ; v ++ ) {
                    c[v] += b[u] * a[u][v] ;
                }
            }
            for(int j = 0 ; j <= to ; j ++ )
                b[j] = c[j] ;
        }
        L ans = 0 ;
        for(int i = 0 ; i <= to ; i ++ ) {
            if(en[i] != 0) ans += b[i] ;
        }
        cout << ans << endl ;
    }

}ac;

int main () {

    L t = read() ;

    while(t -- ) {
        n = read() ;  m = read() ;
        scanf("%s" , s+1) ;
        memset(a,0,sizeof(a)) ;
        memset(b,0,sizeof(b)) ;
        ac.init() ;
        ac.gao() ;
    }
}

  

posted @ 2018-11-20 14:26  天翎月  阅读(321)  评论(0编辑  收藏  举报