标签：出现   queue   printf   字符串   math   scanf   建立   push   turn   

LuoguP3966

题目大意：给定若干个字符串，求它们组成的文章中，每个单词出现的次数。输入有重复。

\(Solution\)#:

多模式匹配，直接上\(AC\)自动机咯。但是注意到，因为有重复，所以造成建立自动机的时候，\(Trie\)上的字符串编号会被覆盖。对答案造成错误影响。

那么我们可以人工去重，记录一下每个重复的串的最早编号，用最早编号替代即可。

那么考虑\(TLE\)问题。

对于一些\(Trie\)上的点，显然有些点我们重复走过而且多次跳过失配指针。这显然是浪费时间的。

那我们用两个数组\(apr,vis\)分别记录一个节点出现的次数以及当前节点是否被统计过。

那么，走到一个节点\(u\)时，看看它是否走过，没走过就一次性把它产生的所有答案统计掉。如果走过显然就不用再走了。\(vis\)数组标记判断。

这题就这么做完了。

\(Code:\)

#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<queue>
using namespace std;
const int MAXN=1e6+10;
const int N=201;
char S[MAXN];
namespace AC{
    int cnt[N],tot,tr[MAXN][26],apr[MAXN];
    int fail[MAXN],idx[MAXN],vis[MAXN];
    int sm[N];
    void Ins(char *s,int pos){
        int u=0;
        for(int i=1;s[i];++i){
            int x=s[i]-'a';
            if(!tr[u][x])tr[u][x]=++tot;
            u=tr[u][x];apr[u]++;
        }
        if(idx[u])sm[pos]=idx[u];
        else idx[u]=pos;
    }
    queue<int>q;
    void build(){
        for(int i=0;i<26;++i)
            if(tr[0][i])q.push(tr[0][i]);
        while(!q.empty()){
            int u=q.front();
            q.pop();
            for(int i=0;i<26;++i){
                if(tr[u][i])
                    fail[tr[u][i]]=tr[fail[u]][i],q.push(tr[u][i]);
                else tr[u][i]=tr[fail[u]][i];
            }
        }
    }
    void query(char *t){
        int u=0;
        for(int i=1;t[i];++i){
            int x=t[i]-'a';
            u=tr[u][x];
            for(int j=u;j;j=fail[j])
                if(idx[j]&&!vis[u])cnt[idx[j]]+=apr[u];
            vis[u]=1;
        }
    }
}
int n;
char s[N][MAXN];
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;++i)scanf("%s",s[i]+1),AC::Ins(s[i],i); 
    AC::build();
    for(int i=1;i<=n;++i)if(!AC::sm[i])AC::query(s[i]); 
    for(int i=1;i<=n;++i)printf("%d\n",AC::sm[i]?AC::cnt[AC::sm[i]]:AC::cnt[i]);
    return 0;
}

【题解】单词

标签：出现   queue   printf   字符串   math   scanf   建立   push   turn   

原文地址：https://www.cnblogs.com/h-lka/p/12189422.html