P3966 [TJOI2013]单词

P3966 [TJOI2013]单词

题目描述

小张最近在忙毕设,所以一直在读论文。一篇论文是由许多单词组成但小张发现一个单词会在论文中出现很多次,他想知道每个单词分别在论文中出现了多少次。

输入输出格式

输入格式：

第一行一个整数N,表示有N个单词。接下来N行每行一个单词,每个单词都由小写字母(a-z)组成。(N≤200)

输出格式：

输出N个整数,第i行的数表示第i个单词在文章中出现了多少次。

输入输出样例

输入样例#1： 

3
a
aa
aaa

输出样例#1： 

6
3
1

说明

数据范围

30%的数据， 单词总长度不超过10^3

100%的数据，单词总长度不超过10^6

看上去题目貌似很简单，不就是把trie树一建然后对每个单词跑一遍AC自动机吗？然而这样会被卡死，考虑如何来优化，由于题目中说每一个单词会多次出现，那么我们就可以统计一下每个单词出现的次数，把这个出现次数看做这个单词的权值，然后在跑AC自动机时直接把这个权值加入到答案中，那么我们如何处理重复出现的单词并统计次数呢？其实我们在trie树上就可以搞定，如果两个单词相同，那么他们在trie树上的结束节点一定相同，这样就可以进行统计了。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<cstring>
#include<map>
#include<queue>
#include<stack>
#include<algorithm>
#include<vector>
#define maxn 1050005
using namespace std;

inline int read()
{
    char c=getchar();
    int res=0,x=1;
    while(c<'0'||c>'9')
    {
        if(c=='-')
        x=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9')
    {
        res=res*10+(c-'0');
        c=getchar();
    }
    return res*x;
}

int n,tot=1;
int tree[maxn][30],nt[maxn],bo[maxn],ed[maxn],f[maxn],t[maxn];
char a[205][maxn];
queue<int>q;

void trie(char *s,int num)
{
    int len=strlen(s),u=1;
    for(register int i=0;i<len;i++)
    {
        int c=s[i]-'a';
        if(!tree[u][c])
        {
            tree[u][c]=++tot;
        }
        u=tree[u][c];
    }
    if(bo[u])
    t[num]=1;//t数组表示这个单词之前是否出现过 
    bo[u]++;
    ed[num]=u;
}

void bfs()
{
    for(register int i=0;i<26;i++)
    {
        tree[0][i]=1;
    }
    nt[1]=0;q.push(1);
    while(q.size())
    {
        int u=q.front();q.pop();
        for(register int i=0;i<26;i++)
        {
            if(!tree[u][i])
            tree[u][i]=tree[nt[u]][i];
            else
            {
                int v=tree[u][i];
                q.push(v);
                nt[v]=tree[nt[u]][i];
            }
        }
    }
}

void find(char *s,int num)
{
    int len=strlen(s),u=1,k,ans=bo[ed[num]];//ans表示这个单词出现的次数 
    for(register int i=0;i<len;i++)
    {
        int c=s[i]-'a';
        k=tree[u][c];
        while(k>1)
        {
            if(bo[k])
            f[k]+=ans;
            k=nt[k];
        }
        u=tree[u][c];
    }
}

int main()
{
    n=read();
    for(register int i=1;i<=n;i++)
    {
        scanf("%s",a[i]);
        trie(a[i],i);
    }
    bfs();
    for(register int i=1;i<=n;i++)
    {
        if(!t[i])//这里我们只需要将每个不重复出现的单词跑一边AC自动机 
        find(a[i],i);
    }
    for(register int i=1;i<=n;i++)
    {
        printf("%d
",f[ed[i]]);
    }
    return 0;
}