牛客：数一数（kmp）

链接：题目
来源：牛客网

时间限制：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个字符串为si。你需要对∀1≤i≤n\forall 1 \leq i \leq n∀1≤i≤n，求出∏j=1nf(si,sj)\prod_{j=1}^n {f(s_i,s_j)}∏j=1n​f(si​,sj​)，由于答案很大，你只需要输出对 998244353取模后的结果。
输入描述:
第一行一个整数n。
接下来n行每行一个仅由英文字母构成的非空字符串，第i个字符串代表si。
输出描述:
共n行，第i行输出∏j=1nf(si,sj)\prod_{j=1}^n {f(s_i,s_j)}∏j=1n​f(si​,sj​)对 998244353取模的结果。
示例1
输入

1
BALDRSKYKirishimaRain

输出

1

备注:
1 ≤ n ≤ 1e6，所有字符串的总长度不超过2*1e6
思路：
这个题可以发现除了最短的能和其他的计算出结果外别的都不能，因为一个长的字串是不可能在一个短的字串相匹配，因此只需要算最短的那个，剩下的就是kmp这个算法了，这个算法那就是可以计算在另一个字符串中一个字符串的个数，可以看下方的资料，也可以看代码解释进行理解
资料：kmp详解

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
const ll N=3000001;
const ll mod=998244353; 
int n;
string a[N];
ll Next[N];//Next[i]表示s中以i结尾的非前缀子串与s的前缀能够匹配的最大位
void getnext(string s)//自己和自己匹配 
{
 ll len=s.size();
 Next[0]=-1;
 for(ll i=0,k=-1;i<len;)
 {
  if(k==-1||s[k]==s[i])
  {
   i++;
   k++;
   Next[i]=k;//记录已经匹配的字符数 
  }
  else 
  {
   k=Next[k];//如果不匹配的话，找出还匹配有几个字符，通过逆向寻找 
  }
 }
}
ll kmp(string s1,string s2)
{
 ll res=0;
 int j=0,i=0;
 while(i<s1.size())
 {
  if(j==-1||s1[i]==s2[j])i++,j++;//如果匹配成功j++; 
  else
  {
   j=Next[j];//否则j=到i这个字符还匹配有几个，通过Next数组 
  }
  if(j==s2.size())//如果够了结果加一 
  {
   res++;
  }
 }
 return res;
}
int main()
{
    ios::sync_with_stdio(false);
 cin>>n;
 ll mx=1e10,id=1;
 for(int i=1;i<=n;i++)
 {
  cin>>a[i];
  if(mx>a[i].size())
  {
   mx=a[i].size();
   id=i;
  }//找出最短的 
 }
 ll ans=1;
 getnext(a[id]);
 for(int i=1;i<=n;i++)
 { 
  ans=(ans*kmp(a[i],a[id]))%mod;
 }//用最短的计算出结果 
 for(int i=1;i<=n;i++)
 {
  if(a[i]!=a[id])cout<<"0"<<"\n";//不是最短的都是零 
  else cout<<ans<<"\n";
 }
 return 0;
}