• bzoj 2434 ac自动机


    ac自动机中,如果以trie中的节点为节点,(fail[i],i)为边,可以建立一颗树,该树有如下特点:“节点u是节点v的祖先 当且仅当 u代表的字符串是v代表的字符串的一个后缀”。(u代表的字符串是由根节点到u路径上所有的边代表的字符顺次组合成的,我们记作str(u))。

    本题中的每一个P都对应trie中的一个节点,所以本题就是求str(b)中有多少个str(a)子串:

    如果len(str(b))<len(str(a)),则为0

    如果len(str(b))==len(str(a)),则判断a和b是否是同一个字符串。

    如果len(str(b))<len(str(a)),则str(a)一定是str(b)一个前缀的后缀,str(b)的前缀就是根到b路径上所有点代表的字符串,而如果str(a)是str(b)的一个后缀,则在fail树中,b一定在a的子树中。

      1 /**************************************************************
      2     Problem: 2434
      3     User: idy002
      4     Language: C++
      5     Result: Accepted
      6     Time:580 ms
      7     Memory:18912 kb
      8 ****************************************************************/
      9  
     10 #include <cstdio>
     11 #include <cstring>
     12 #include <queue>
     13 #define maxn 100010
     14 #define fprintf(...)
     15 using namespace std;
     16  
     17 struct Query {
     18     int a, id;
     19     Query( int a, int id ):a(a),id(id){}
     20 };
     21  
     22 int n, m, ans[maxn];
     23 char str[maxn];
     24 int son[maxn][26], pre[maxn], fail[maxn], ppos[maxn], ntot;
     25 int ww[maxn], dl[maxn], dr[maxn], dfs_clock;
     26 vector<int> g[maxn];
     27 vector<Query> qry[maxn];
     28  
     29  
     30 void modify( int pos, int delta ) {
     31     for( int i=pos; i<=n; i+=i&-i )
     32         ww[i] += delta;
     33 }
     34 int query( int pos ) {
     35     int rt=0;
     36     for( int i=pos; i; i-=i&-i )
     37         rt += ww[i];
     38     return rt;
     39 }
     40 void build_trie( int n, const char *P ) {
     41     int u = 0;
     42     int pid_clock=0;
     43     for( int i=0; i<n; i++ ) {
     44         if( P[i]=='P' ) {
     45             pid_clock++;
     46             ppos[pid_clock] = u;
     47             fprintf( stderr, "the No.%d P is at node %d
    ", pid_clock, u );
     48         } else if( P[i]=='B' ) {
     49             u = pre[u];
     50             fprintf( stderr, "up to %d
    ", u );
     51         } else {
     52             int c=P[i]-'a';
     53             if( !son[u][c] ) {
     54                 ntot++;
     55                 son[u][c] = ntot;
     56                 pre[ntot] = u;
     57             }
     58             fprintf( stderr, "%d->%d with %c
    ", u, son[u][c], c+'a' );
     59             u = son[u][c];
     60         }
     61     }
     62 }
     63 void build_fail() {
     64     queue<int> qu;
     65     for( int c=0; c<26; c++ ) {
     66         int v=son[0][c];
     67         if( !v ) continue;
     68         fail[v] = 0;
     69         fprintf( stderr, "fail[%d] = %d
    ", v, fail[v] );
     70         g[0].push_back(v);
     71         qu.push( v );
     72     }
     73     while( !qu.empty() ) {
     74         int u=qu.front();
     75         qu.pop();
     76         for( int c=0; c<26; c++ ) {
     77             int v=son[u][c];
     78             int w=fail[u];
     79             if( !v ) continue;
     80             while( w && !son[w][c] ) w=fail[w];
     81             fail[v] = son[w][c];
     82             fprintf( stderr, "fail[%d] = %d
    ", v, fail[v] );
     83             g[son[w][c]].push_back( v );
     84             qu.push( v );
     85         }
     86     }
     87 }
     88 void dfs( int u ) {
     89     dl[u] = ++dfs_clock;
     90     fprintf( stderr, "(%d ", u );
     91     for( int t=0; t<g[u].size(); t++ )
     92         dfs( g[u][t] );
     93     dr[u] = dfs_clock;
     94     fprintf( stderr, ")" );
     95 }
     96 void solve( int n, const char *P ) {
     97     int u=0;
     98     for( int i=0; i<n; i++ ) {
     99         if( P[i]=='P' ) {
    100             for( int t=0; t<qry[u].size(); t++ ) {
    101                 Query &q = qry[u][t];
    102                 if( q.a==u ) ans[q.id]=1;
    103                 else ans[q.id] = query(dr[q.a])-query(dl[q.a]-1);
    104             }
    105         } else if( P[i]=='B' ) {
    106             if( u==0 ) continue;
    107             modify( dl[u], -1 );
    108             u = pre[u];
    109         } else {
    110             int c=P[i]-'a';
    111             u = son[u][c];
    112             modify( dl[u], +1 );
    113         }
    114     }
    115 }
    116 int main() {
    117     scanf( "%s", str );
    118     n = strlen(str);
    119     build_trie(n,str);
    120     build_fail();
    121     dfs(0);
    122     fprintf( stderr, "
    " );
    123     scanf( "%d", &m );
    124     for( int i=1,a,b; i<=m; i++ ) {
    125         scanf( "%d%d", &a, &b );
    126         a = ppos[a];
    127         b = ppos[b];
    128         qry[b].push_back( Query(a,i) );
    129     }
    130     solve(n,str);
    131     for( int i=1; i<=m; i++ )
    132         printf( "%d
    ", ans[i] );
    133 }
    134     
    View Code

    收获:

    1、“a是b的子串 当且仅当 a是b的后缀的前缀或前缀的后缀“ 所以统计时就可以根据这个分类判断。

    2、 fail指针对应的树的含义:同一条链上,深度小的字符串是深度大的字符串的后缀。

     

  • 相关阅读:
    TypesScript+Webpack
    TypeScript 类型
    git操作
    kafka
    java: cannot find symbol symbol: variable log
    Angular结构型指令,模块和样式
    Angular 自定义拖拽指令
    Angular changeDetction
    Angular 依赖注入
    RXJS Observable的冷,热和Subject
  • 原文地址:https://www.cnblogs.com/idy002/p/4337709.html
Copyright © 2020-2023  润新知