wannafly 挑战赛9 B 数一数（kmp）

链接：https://www.nowcoder.com/acm/contest/71/B

时间限制：C/C++ 1秒，其他语言2秒

空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

设s,t为两个字符串，定义f(s,t) = t的子串中，与s相等的串的个数。如f("ac","acacac")=3, f("bab","babab")=2。现在给出n个字符串，第i个字符串为s_i。你需要对，求出，由于答案很大，你只需要输出对 998244353取模后的结果。

输入描述:

第一行一个整数n。
接下来n行每行一个仅由英文字母构成的非空字符串，第i个字符串代表s

_i

。

输出描述:

共n行，第i行输出

对 998244353取模的结果。

输入例子:

1
BALDRSKYKirishimaRain

输出例子:

1

-->

示例1

输入

1
BALDRSKYKirishimaRain

输出

1

备注:

1 ≤ n ≤ 1e6，所有字符串的总长度不超过2e6
//////////////////////////////////////////////////////////////////////////////////////////

////////////////////////////////////////////////////////////////////////////////////////////////////////////////

 #include <bits/stdc++.h>
 #define mst(a,b)    memset((a),(b), sizeof a)
 #define lowbit(a)   ((a)&(-a))
 #define IOS         ios::sync_with_stdio(0);cin.tie(0);
 using namespace std;
 typedef long long ll;
 typedef unsigned long long ull;
 typedef pair<int,int> pii;
 const int mod=;
 const int maxn=1e6+;
 string str[maxn];
 int d[maxn];
 bool cmp(const int&a,const int&b){
     return str[a].length()<str[b].length();
 }
 char x[maxn<<];int nx[maxn<<],len;
 char y[maxn<<];
 int ans[maxn];
 bool zero;
 ll nn=;
 void kmp(int pos){
     if(zero)return;
     int sz=str[pos].size();
     if(sz<len){nn=;zero=true;return;}
     for(int i=;i<sz;++i)y[i]=str[pos][i];
     int i,j;
     int ans=;
     i=j=;
     while(i<sz){
         while(-!=j&&y[i]!=x[j])j=nx[j];
         ++i;++j;
         if(j>=len){
             ++ans;
             j=nx[j];
         }
     }
     nn=nn*ans%mod;
 }
 void init(int pos){
     len=str[pos].length();
     for(int i=;i<len;++i)x[i]=str[pos][i];
     int i,j;
     j=nx[]=-;
     i=;
     while(i<len){
         while(-!=j&&x[i]!=x[j])j=nx[j];
         if(x[++i]==x[++j])nx[i]=nx[j];
         else nx[i]=j;
     }
 }
 int main() {
 #ifdef local
     freopen("inpp","r",stdin);
 //    freopen("outpp","w",stdout);
 #endif
     int n;scanf("%d",&n);
     for(int i=;i<=n;++i)cin>>str[i],d[i]=i;
     sort(d+,d++n,cmp);
     init(d[]);
     for(int i=;i<=n;++i)kmp(d[i]);
     ans[d[]]=nn;
     for(int i=;i<=n;++i){
         if(str[d[i]].length()==len)ans[d[i]]=nn;
         else ans[d[i]]=;
     }
     for(int i=;i<=n;++i)printf("%d\n",ans[i]);
     return ;
 }