AC自动机 zoj3228 Searching the String

传送门:点击打开链接

题意:告诉你原串,告诉你很多个子串,有两种匹配模式,一种是可以重叠的匹配,一种是不能重叠的,问每个子串出现的次数。子串可能有重复的出现。

思路:这题揭示了AC自动机的两种处理方法。

首先是当AC自动机中出现有串重复的时候,我们应该怎么处理,我这里用到了一个数组pre,如果这个节点结尾的子串是第一次出现,pre标记为自己,如果这个节点结尾的子串不是第一次出现,那么就把pre标记成第一次出现的串。最后输出答案的时候,输出的是ans[pre[i]]],这里其实就是有许多个相同子串,一个子串我们只要算一次就行了,然后把相同子串的答案和我们计算的那个相等就行。

其次是遇到不能重叠的子串匹配,我们应当如何去处理。我们维护了一个数组last来表示某个子串上次匹配的位置,看这个位置加上子串的长度,是否小于等于当前匹配的位置,如果小于等于,则说明不会和上次重叠,那么这次找到了就可以更新答案。

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define fuck(x) cout<<"["<Q;
        Fail[root] = root;
        for(int i = 0; i < P; i++) {
            if(Next[root][i] == -1) {
                Next[root][i] = root;
            } else {
                Fail[Next[root][i]] = root;
                Q.push(Next[root][i]);
            }
        }

        while(!Q.empty()) {
            int u = Q.front();
            Q.pop();

            for(int i = 0; i < P; i++) {
                if(Next[u][i] == -1) {
                    Next[u][i] = Next[Fail[u]][i];
                } else {
                    Fail[Next[u][i]] = Next[Fail[u]][i];
                    Q.push(Next[u][i]);
                }
            }
        }
    }

    void Query_1(char *S) {
        int n = strlen(S), now = root, ret = 0;

        for(int i = 0; i < n; i++) {
            now = Next[now][S[i] - 'a'];
            int temp = now;

            while(temp != root) {
                if(End[temp]) ans[End[temp]]++;
                temp = Fail[temp];
            }
        }
    }

    void Query_2(char *S) {
        int n = strlen(S), now = root, ret = 0;

        for(int i = 0; i < n; i++) {
            now = Next[now][S[i] - 'a'];
            int temp = now;

            while(temp != root) {
                if(End[temp] && (last[End[temp]] == -1 || last[End[temp]] + len[temp] <= i)){
                    ans[End[temp]]++;
                    last[End[temp]] = i;
                }
                temp = Fail[temp];
            }
        }
    }
} AC;

char word[10], line[MN];
vector >A;

int main() {
    int m, ansk = 0; //FIN;
    while(~scanf("%s", line)) {
        memset(ans, 0, sizeof(ans));
        A.clear();
        scanf("%d", &m);
        AC.Init(m);

        int r1 = 0, r2 = 0;
        for(int i = 1; i <= m; i++) {
            int type;
            scanf("%d%s", &type, word);
            if(type == 0) {
                AC.Add(word, i);
            } else A.push_back(make_pair(string(word), i));
        }
        AC.Build();
        AC.Query_1(line);

        AC.Init(A.size());
        for(int i = 0; i < A.size(); i++) {
            AC.Add(A[i].first.c_str(), A[i].second);
        }

        AC.Build();
        AC.Query_2(line);

        printf("Case %d\n", ++ansk);
        for(int i = 1; i <= m; i++) printf("%d\n", ans[pre[i]]);
        printf("\n");
    }
    return 0;
}


你可能感兴趣的:(ACM_DP,ACM_自动机)