HDU5880 Family View（AC自动机）

链接：http://acm.hdu.edu.cn/showproblem.php?pid=5880

题意：给定N个串，一个长文本，把那N个串在长文本里出现的部分用×代替。

思路：把N个串放到AC自动机上然后在结点标记上串长，接着把长文本放到自动机上去匹配，匹配到了就替换掉。

至于替换的方法，不用暴力的去修改，只要同类似括号匹配的方式打上标记就好了。要注意2种情况的处理

1：   N个串种有 abcd,bcde这种交叉的情况

2：  N个串有 abcd bc这种包含的情况

然后，千万不要memset整个数组，因为测评鸡算内存只会算用过的，一memset就算全用过了，会爆内存，要用一个就初始化一个。

代码：

#include <bits/stdc++.h>

using namespace std;

#define fur(i,a,b) for(int i=(a);i<=(b);i++)
#define furr(i,a,b) for(int i=(a);i>=(b);i--)
#define cl(a) memset((a),0,sizeof(a))

typedef long long ll;

const int maxsz = 1000000;

const int maxn = 1010000;
int cg[maxn];

char str[maxsz+10];	

struct AC
{
    struct Node
    {
        int fail,cnt,next[26];
    }node[maxn+10];

    int sz;
    queue<int>q;

    void init(){
        sz = 0;
        newnode();
        while(!q.empty()) q.pop();
    }

    int newnode(){
        cl(node[sz].next);
        node[sz].fail = 0;
        node[sz].cnt = 0;
        return sz++;
    }
    
    void insert(char *s){
        int len = strlen(s);
        int now = 0;
        for(int i = 0; i < len; ++i){
            char c = s[i];
            if(!node[now].next[c-'a'])
                node[now].next[c-'a'] = newnode();
            now = node[now].next[c-'a'];
        }
        node[now].cnt = len;
    }

    void build(){
        node[0].fail = -1;
        q.push(0);
        while(!q.empty()){
            int u = q.front();
            q.pop();
            fur(i,0,25){
                if(node[u].next[i]){
                    if(u == 0)
                        node[node[u].next[i]].fail = 0;
                    else{
                        int v = node[u].fail;
                        while(v!=-1){
                            if(node[v].next[i]){
                                node[node[u].next[i]].fail = node[v].next[i];
                                break;
                            }
                            v = node[v].fail;
                        }
                        if(v == -1) node[node[u].next[i]].fail = 0;
                    }
                    q.push(node[u].next[i]);
                }
            }
        }
    }

    void getret(int u, int i){
        while(u){
            if(node[u].cnt){
                cg[i+1]++;
                cg[i-node[u].cnt+1]--;
            }
            u = node[u].fail;
        }
    }

    void match(char *s){
        cl(cg);
        int len = strlen(s);
        int now = 0;
        for(int i = 0; i < len; ++i){
            int idx;
            if(s[i]>='A'&&s[i]<='Z') idx=s[i]-'A';
            else if(s[i]>='a'&&s[i]<='z') idx=s[i]-'a';
            else{
                now = 0;
                continue;
            }
            if(node[now].next[idx])
                now = node[now].next[idx];
            else{
                int u = node[now].fail;
                while(u!=-1 && !node[u].next[idx]) u = node[u].fail;
                if(u == -1) now = 0;
                else now = node[u].next[idx];
            }
            getret(now,i);
        }

        int tmp = 0;

        for(int i = 0; i < len; ++i){
            tmp+=cg[i];
            if(tmp<0) putchar('*');
            else putchar(s[i]); 
        }
        puts("");
    }

}ac;


int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        ac.init();
        int n;
        scanf("%d",&n);
        while(n--){
            scanf("%s",str);
            ac.insert(str);
        }
        ac.build();
        getchar();
        gets(str);
        ac.match(str);
    }
    return 0;
}