【Wannafly挑战赛9-B】数一数

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

题目就不贴了。。

设res[i]为第i行的最终结果，可以想到，res[i]为0或不为0。长度不是最短的字符串res一定0，因为res是所有子串数量的乘积，对于每个字符串，只有长度最短的字符串乘过之后才会有结果，其余的res都是0。有多个长度最短的，结果都一样，总之...看代码就懂了（逃。之后就是KMp了（好长时间没复习kmp都忘了惹Σ(ﾟдﾟ;)

上代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int mod = 998244353;
const int N = 1e6+3;
string s[N];
char s1[N*2];
int id[N], Next[N*2];
LL res[N];
void make(string s, int m)
{
    int i, j;
    for(i=1, j=0; i<m; i++)
    {
        while(j > 0 && s[i] != s[j])
            j = Next[j-1];
        if(s[i] == s[j])
            j++;
        Next[i] = j;
    }
}
int kmp(string s1, string s2)
{
    int i, j, num = 0;
    int l1 = s1.size();
    int l2 = s2.size();
    Next[0] = 0;
    make(s1, l1);
    for(i = 0, j = 0; i < l2; i++)
    {
        while(j > 0 && s1[j] != s2[i])
            j = Next[j-1];
        if(s1[j] == s2[i])
            j++;
        if(j == l1)
        {
            num++;
            j = Next[j-1];
        }
    }
    return num;
}
int main()
{
    int n, i, j;
    cin>>n;
    for(i = 1; i <= n; i++)
    {
        scanf("%s", s1);
        s[i] = s1;
        id[i] = i;//用id标号来给长度从小到大排序
    }
    sort(id+1, id+1+n, [](int x, int y){return s[x]. size()<s[y].size();});//又学会一种偷懒+的写法(°∀°)ﾉ
    int m = s[id[1]].size();
    res[id[1]] = 1;
    for(j = 2; j <= n; j++)
        res[id[1]] = (res[id[1]]*kmp(s[id[1]], s[j])%mod)%mod;
    for(i = 2; i <= n; i++)
    {
        if(s[id[i]].size() != m) break;
        if(s[id[i]].size() == m) res[id[i]] = res[id[1]];
    }
    for(i = 1; i <= n; i++)
        printf("%lld
", res[i]);
    return 0;
}