BZOJ5337: [TJOI2018]str

BZOJ5337: [TJOI2018]str

https://lydsy.com/JudgeOnline/problem.php?id=5337

分析:

• 这题数据范围实在不明确，我以为总串长是\(10^7\)级别的，一直在想一个和总串长相关的线性做法...然后膜了题解。
• 这玩意，设\(f[i][j]\)表示匹配前\(i\)个串，当前已经匹配到\(s\)的第\(j\)个字符上的方案数，转移只需要用哈希判断两串相等就行了。

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define base 10010257
#define N 10050
#define mod 1000000007
ull mi[N],h[N];
char w[N],s[N];
int K,n;
ll f[105][N];
inline void upd(ll &x,ll y) {
    x=x+y; if(x>=mod) x-=mod;
}
ull geth(int l,int r) {
    return h[r]-h[l-1]*mi[r-l+1];
}
int main() {
    scanf("%d%s",&K,w+1);
    n=strlen(w+1);
    int i,j;
    for(mi[0]=i=1;i<=n;i++) mi[i]=mi[i-1]*base;
    for(i=1;i<=n;i++) h[i]=h[i-1]*base+w[i];
    for(i=0;i<n;i++) f[0][i]=1;
    for(i=0;i<K;i++) {
        int x;
        scanf("%d",&x);
        while(x--) {
            scanf("%s",s+1);
            int m=strlen(s+1);
            ull nh=0;
            for(j=1;j<=m;j++) nh=nh*base+s[j];
            for(j=0;j<=n-m;j++) if(f[i][j]) {
                if(geth(j+1,j+m)==nh) upd(f[i+1][j+m],f[i][j]);
            }
        }
    }
    ll ans=0;
    for(i=1;i<=n;i++) upd(ans,f[K][i]);
    printf("%lld\n",ans);
}