Wannafly挑战赛9 B 数一数

最新推荐文章于 2022-10-13 16:29:59 发布

还是太年轻

最新推荐文章于 2022-10-13 16:29:59 发布

阅读量193

点赞数

CC 4.0 BY-SA版权

分类专栏：竞赛 KMP

本文链接：https://blog.youkuaiyun.com/albertluf/article/details/79325085

竞赛同时被 2 个专栏收录

87 篇文章

订阅专栏

KMP

9 篇文章

订阅专栏

找出最短的字符串s,则对于其他字符串t(除去与s相等的情况)
都有f(s,t)=0;因为要求所有的结果相乘，因此最后一定是0.
因此只需求出最短的字符串即可。不要忘记考虑存在多个
最短且相等的字符串的情况(若存在多个最短且不想等的字符串
则最后结果一定为0)

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+7;
const int mod=998244353;
string t[maxn];
int len,next1[2*maxn],id;
void NEXT()
{
    int k=-1,s=0;
    memset(next1,-1,sizeof(next1));
    while(s<len)
    {
        if(k==-1||t[id][k]==t[id][s])
            s++,k++,next1[s]=k;
        else k=next1[k];
    }
}
long long KMP(int i)
{
    int q=0,s=0,L=t[i].length();
    long long ans=0;
    while(s<L)
    {
        if(q==-1||t[id][q]==t[i][s])
            s++,q++;
        else q=next1[q];
        if(q==len)
        {
            ans++;
            q=next1[q];
        }
    }
    return ans;
}
int main()
{
    int n,i,j;
    len=2*1e6+7;
    scanf("%d",&n);
    for(i=0; i<n; i++)
    {
        cin>>t[i];
        if(t[i].length()<len)
        {
            len=t[i].length();
            id=i;
        }
    }
    NEXT();
    long long s=1;
    for(i=0;i<n;i++)
    {
        if(i==id)continue;
        s=(s*KMP(i))%mod;
        if(s==0)break;
    }
    for(i=0;i<n;i++)
    {
        if(t[i].length()==len)printf("%lld\n",s);
        else printf("0\n");
    }
    return 0;
}