結果

問題 No.2626 Similar But Different Name
ユーザー zeta7532zeta7532
提出日時 2024-01-11 01:41:28
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 281 ms / 3,000 ms
コード長 2,707 bytes
コンパイル時間 2,543 ms
コンパイル使用メモリ 215,532 KB
実行使用メモリ 76,440 KB
最終ジャッジ日時 2024-09-27 20:23:58
合計ジャッジ時間 9,136 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
using namespace std;
using ll = long long;
const ll mod = 998244353;
#define fi first
#define se second
#define rep(i,n) for(ll i=0;i<n;i++)
#define all(x) x.begin(),x.end()
#define faster ios::sync_with_stdio(false);cin.tie(nullptr)

const ll root = 3;

ll modpow(ll A,ll N){
    ll res=1;
    while(N){
        if(N&1) res=res*A%mod;
        A=A*A%mod;
        N>>=1;
    }
    return res;
}

ll modinv(ll A){return modpow(A,mod-2);}

void NTT(vector<ll> &A,const bool rev=false){
    vector<ll> B(A.size(),0);
    ll R=mod-1-(mod-1)/((ll)A.size());
    if(rev) R=(mod-1)/((ll)A.size());
    ll s=modpow(root,R);
    vector<ll> power((ll)A.size()/2+1,1);
    rep(i,(ll)A.size()/2) power[i+1]=(power[i]*s)%mod;
    ll cur=(ll)A.size()/2;
    for(ll i=1;i<(ll)A.size();i*=2){
        rep(j,cur){
            rep(k,i){
                ll p=A[k+i*j];
                ll q=A[k+i*j+(ll)A.size()/2];
                B[k+2*i*j]=(p+q)%mod;
                B[k+2*i*j+i]=(p+mod-q)*power[i*j]%mod;
            }
        }
        swap(A,B);
        cur/=2;
    }
    if(rev){
        s=modinv((ll)A.size());
        rep(i,(ll)A.size()) A[i]=(A[i]*s)%mod;
    }
}

vector<ll> convolution(vector<ll> A,vector<ll> B){
    ll size_=1;
    while(size_<=(ll)A.size()+(ll)B.size()-1) size_*=2;
    while(A.size()<size_) A.push_back(0);
    while(B.size()<size_) B.push_back(0);
    NTT(A);
    NTT(B);
    rep(i,size_) A[i]=(A[i]*B[i])%mod;
    NTT(A,true);
    return A;
}

int main() {
    ll N,M,K;
    cin >> N >> M >> K;
    string s,t;
    cin >> s >> t;
    vector<ll> po(500500,1);
    rep(i,500500-1) po[i+1]=po[i]*100%mod;
    vector<ll> S(N),T(M);
    vector<ll> A0(N),A1(N);
    vector<ll> B0(M),B1(M);
    rep(i,N){
        if('a'<=s[i]&&s[i]<='z') S[i]=(s[i]-'a')+1,A0[i]=0;
        if('A'<=s[i]&&s[i]<='Z') S[i]=(s[i]-'A')+1,A0[i]=1;
        S[i]=S[i]*po[i]%mod;
        A1[i]=1-A0[i];
    }
    rep(i,M){
        if('a'<=t[i]&&t[i]<='z') T[i]=(t[i]-'a')+1,B0[i]=0;
        if('A'<=t[i]&&t[i]<='Z') T[i]=(t[i]-'A')+1,B0[i]=1;
        T[i]=T[i]*po[i]%mod;
        B1[i]=1-B0[i];
    }
    reverse(all(B0));
    reverse(all(B1));
    vector<ll> C0=convolution(A0,B1);
    vector<ll> C1=convolution(A1,B0);
    vector<ll> cumS(N+1,0),cumT(M+1,0);
    rep(i,N) cumS[i+1]=(cumS[i]+S[i])%mod;
    rep(i,M) cumT[i+1]=(cumT[i]+T[i])%mod;
    ll ans=0;
    rep(i,N-M+1){
        bool ok=true;
        if((cumS[i+M]+mod-cumS[i])%mod!=cumT[M]*po[i]%mod) ok=false;
        if((C0[i+M-1]+C1[i+M-1])==0||K<(C0[i+M-1]+C1[i+M-1])) ok=false;
        if(ok) ans++;
    }
    cout << ans << endl;
    return 0;
}
0