zoukankan      html  css  js  c++  java
  • 联考20200725 T1 String


    分析:
    发现字符集大小只有8,时间限制是9s
    对于每种字符(c)构建生成函数(f(x)=sum_{i=0}[S_i==c]x^i)
    (S_2)的生成函数翻转与(S_1)做卷积,得到的([|S_2|-1,|S_1|-1])的位置的值就是(S_1)中下标从([0,|S1|-|S2|])开始的字符相同数量
    数据规模(10^6),需要比较优秀的NTT实现
    复杂度(O(8nlogn))

    #include<cstdio>
    #include<cstring>
    #include<cmath>
    #include<algorithm>
    #include<vector>
    #include<iostream>
    #include<map>
    #include<string>
    
    #define maxn 3000005
    #define MOD 998244353
    #define Poly vector<int>
    
    using namespace std;
    
    inline int getint()
    {
    	int num=0,flag=1;char c;
    	while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    	while(c>='0'&&c<='9')num=num*10+c-48,c=getchar();
    	return num*flag;
    }
    
    int n,m,K;
    char S1[maxn],S2[maxn];
    int Id[128];
    int A[8][maxn],B[8][maxn];
    int ans[maxn];
    int Wl,Wl2,w[maxn<<2];
    int rev[maxn<<2];
    
    inline int upd(int x){return x<MOD?x:x-MOD;}
    inline int ksm(int num,int k)
    {
    	int ret=1;
    	for(;k;k>>=1,num=1ll*num*num%MOD)if(k&1)ret=1ll*ret*num%MOD;
    	return ret;
    }
    
    inline void init(int N)
    {
    	Wl=w[0]=1;
    	while((Wl<<1)<=N)Wl<<=1;
    	w[1]=ksm(3,(MOD-1)/(Wl<<1)),Wl2=Wl<<1;
    	for(int i=2;i<=Wl2;i++)w[i]=1ll*w[i-1]*w[1]%MOD;
    }
    
    inline void NTT(int *a,int opt,int N)
    {
    	for(int i=0;i<N;i++)if(i<rev[i])swap(a[i],a[rev[i]]);
    	for(int i=1,B=Wl;i<N;i<<=1,B>>=1)
    		for(int j=0,t=i<<1;j<N;j+=t)for(int k=0,x=0;k<i;k++,x+=B)
    		{
    			int v=1ll*a[i+j+k]*w[opt==1?x:Wl2-x]%MOD;
    			a[i+j+k]=upd(a[j+k]-v+MOD),a[j+k]=upd(a[j+k]+v);
    		}
    	if(!~opt)for(int i=0,Inv=ksm(N,MOD-2);i<N;i++)a[i]=1ll*a[i]*Inv%MOD;
    }
    
    int main()
    {
    	K=getint();
    	scanf("%s%s",S1,S2);
    	Id['s']=0,Id['y']=1,Id['f']=2,Id['a']=3,Id['k']=4,Id['n']=5,Id['o']=6,Id['i']=7;
    	n=strlen(S1),m=strlen(S2);reverse(S2,S2+m);
    	K=m-K;
    	for(int i=0;i<n;i++)A[Id[S1[i]]][i]=1;
    	for(int i=0;i<m;i++)B[Id[S2[i]]][i]=1;
    	int len=1;
    	while(len<n+m)len<<=1;
    	for(int i=0;i<len;i++)rev[i]=(rev[i>>1]>>1)|(i&1?len>>1:0);
    	init(n+m);
    	for(int k=0;k<8;k++)
    	{
    		NTT(A[k],1,len),NTT(B[k],1,len);
    		for(int i=0;i<len;i++)A[k][i]=1ll*A[k][i]*B[k][i]%MOD;
    		NTT(A[k],-1,len);
    		for(int i=m-1;i<n;i++)ans[i]+=A[k][i];
    	}
    	int num=0;
    	for(int i=m-1;i<n;i++)num+=(ans[i]>=K);
    	printf("%d
    ",num);
    }
    

  • 相关阅读:
    物理机连接虚拟机中的sqlserver
    Vue.js
    拆分时间段
    System.Threading.Timer
    浏览器被恶心页面占用
    sqlserver超时时间已到
    几年没写CSS
    C#生成高清缩略图
    抽奖概率算法
    html 页面实现指定位置的跳转
  • 原文地址:https://www.cnblogs.com/Darknesses/p/13377156.html
Copyright © 2011-2022 走看看