子串(非NOIP的子串) - NTT - 哈希

题目大意:给两个字符串S和T,和一个置换p,问有多少S的子串,经过置换之后可以和T匹配。1e5
题解:考虑哈希,原来第i为的权值从BAS^i变成了根置换有关的东西,因此把BAS^p[i]倒过来和S做一个卷积即可,细节自行处理。
代码:

// luogu-judger-enable-o2
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define gc getchar()
#define lint long long
#define BAS 37
#define p 998244353
#define lint long long
#define N 1100010
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
inline int inn()
{
    int x,ch;while((ch=gc)<'0'||ch>'9');
    x=ch^'0';while((ch=gc)>='0'&&ch<='9')
        x=(x<<1)+(x<<3)+(ch^'0');return x;
}
int a[N],b[N],t[N],mi[N],r[N],hv[N];
inline int fast_pow(int x,int k,int ans=1)
{   for(;k;k>>=1,x=(lint)x*x%p) (k&1)?ans=(lint)ans*x%p:0;return ans;   }
inline int NTT(int *a,int n,int sgn)
{
    for(int i=0;i<n;i++) if(i<r[i]) swap(a[i],a[r[i]]);
    for(int i=2;i<=n;i<<=1)
    {
        int wn=fast_pow(3,sgn>0?(p-1)/i:p-1-(p-1)/i);
        for(int j=0,t=i>>1;j<n;j+=i)
            for(int k=0,w=1;k<t;k++,w=(lint)w*wn%p)
            {
                int x=a[j+k],y=(lint)w*a[j+k+t]%p;
                a[j+k]=(x+y)%p,a[j+k+t]=(x-y+p)%p;
            }
    }
    return 0;
}
inline int calc(int x,int l)
{   return (hv[x]-(lint)mi[l]*hv[x+l]%p+p)%p;   }
int main()
{
    int n=inn(),m=inn(),c=inn(),L,ms=m,ns=n;
    for(int i=0;i<n;i++) t[i]=inn()-1;
    for(int i=0;i<m;i++) a[i]=inn();
    for(int i=m-1;i>=0;i--) hv[i]=((lint)hv[i+1]*BAS+a[i])%p;
//  for(int i=0;i<m;i++) debug(i)sp,debug(hv[i])ln;
    for(int i=mi[0]=1;i<m;i++) mi[i]=(lint)mi[i-1]*BAS%p;
    for(int i=0;i<n;i++) b[i]=mi[t[n-i-1]];
//  for(int i=0;i<m;i++) cerr<<a[i]<<" ";cerr ln;
//  for(int i=0;i<n;i++) cerr<<b[i]<<" ";cerr ln;
    for(m+=n,n=1,L=0;n<=m;n<<=1,L++);
    for(int i=1;i<n;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(L-1));
    NTT(a,n,1),NTT(b,n,1),m=ms;
    for(int i=0;i<n;i++) a[i]=(lint)a[i]*b[i]%p;
    NTT(a,n,-1);int ninv=fast_pow(n,p-2);
    for(int i=0;i<n;i++) a[i]=(lint)a[i]*ninv%p;
//  for(int i=0;i<n;i++) cerr<<a[i]<<" ";cerr ln;
//  for(int i=0;i<m-ns+1;i++) debug(i)sp,debug(calc(i,ns))ln;
    for(int i=0;i<m-ns+1;i++) printf("%d",calc(i,ns)==a[i+ns-1]);
    return !printf("\n")+c-c;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值