洛谷p4180 ac自动机

匹配字符串时，对于重复的单词我们只考虑一次，我们开一个数组记录，重复单词的第一个id将重复单词的出现次数全部变为第一次出现的个数相加。且在匹配时，对于每个now只扫描一次，不重复扫描。

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <queue>
#include <cstring>
#include <algorithm>
#include <string>
using namespace std;
//由于题目中子串美丽度计算时不可重复，因此记录每个字符串长度，

struct Tire{
    int nxt[1000100][26],fail[1000100],ed[1000100],sum[1000100],same[1000100];
    int cnt,rt;
    int ans[1000100],vis[1000100];
    int newnode()
    {
        for(int i=0;i<26;i++)
        {
            nxt[cnt][i]=-1;
        }
        ed[cnt++]=0;
        return cnt-1;
    }
    void init()
    {
        cnt=0;
        rt=newnode();
    }
    void ins(string s,int id)
    {
        int len=s.size();
        int now=rt;
        for(int i=0;i<len;i++)
        {
            if(nxt[now][s[i]-'a']==-1)
            {
                nxt[now][s[i]-'a']=newnode();
            }
            now=nxt[now][s[i]-'a'];
            sum[now]++;
        }
        if(ed[now])
        same[id]=ed[now];
        else
        same[id]=ed[now]=id;

    }
    void build()
    {
        queue<int>q;
        fail[rt]=rt;
        for(int i=0;i<26;i++)
        {
            if(nxt[rt][i]==-1)
            {
                nxt[rt][i]=rt;
            }
            else
            {
                fail[nxt[rt][i]]=rt;
                q.push(nxt[rt][i]);
            }
        }
        while(!q.empty())
        {
            int now=q.front();
            q.pop();
            for(int i=0;i<26;i++)
            {
                if(nxt[now][i]==-1)
                {
                    nxt[now][i]=nxt[fail[now]][i];
                }
                else
                {
                    fail[nxt[now][i]]=nxt[fail[now]][i];
                    q.push(nxt[now][i]);
                }
            }
        }
    }
    void query(string s)
    {
        int now=rt,top=0;
       for(int i=0;i<s.size();i++)
       {
          now=nxt[now][s[i]-'a'];
          int tt=now;
          while(tt)
          {
              if(ed[tt]&&!vis[now])
              {
              ans[ed[tt]]+=sum[now];
              //printf("%d %d\n",ed[tt],ans[ed[tt]]);
              }tt=fail[tt];
          }
          vis[now]=1;
       }
    }
};
Tire ac;
string s[300];
int main()
{
        int n;
        scanf("%d",&n);
        ac.init();
        for(int i=1;i<=n;i++)
        {
             cin>>s[i];
             ac.ins(s[i],i);
        }
        ac.build();
        for(int i=1;i<=n;i++)
            if(ac.same[i]==i)
            ac.query(s[i]);
        for(int i=1;i<=n;i++)
        {
            printf("%d\n",ac.ans[ac.same[i]]);
        }
    return 0;
}