結果

問題 No.430 文字列検索
ユーザー matcharate12matcharate12
提出日時 2024-08-21 18:14:43
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 435 ms / 2,000 ms
コード長 2,292 bytes
コンパイル時間 1,028 ms
コンパイル使用メモリ 104,136 KB
実行使用メモリ 28,032 KB
最終ジャッジ日時 2024-11-10 01:12:22
合計ジャッジ時間 3,191 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 435 ms
28,032 KB
testcase_02 AC 29 ms
5,248 KB
testcase_03 AC 30 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 403 ms
27,776 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 18 ms
5,760 KB
testcase_11 AC 111 ms
8,832 KB
testcase_12 AC 108 ms
8,832 KB
testcase_13 AC 110 ms
8,832 KB
testcase_14 AC 85 ms
7,424 KB
testcase_15 AC 68 ms
6,528 KB
testcase_16 AC 34 ms
5,248 KB
testcase_17 AC 33 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

//#include <bits/stdc++.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <iomanip>
#include <numeric>
#include <cmath>
#include <cstdint>
#include <stack>

#define rep(i,n) for(int i = 0; i < (n); i++) //[0,n)
#define srep(i,a,b) for(int i = (a); i < (b); i++) //[a,b)
#define all(A) (A).begin(),(A).end()
#define rall(A) (A).rbegin(),(A).rend()
using namespace std;
using ll = long long;
using ull = unsigned long long;
const ll inf = 1LL<<60;
const int iinf = (int)1e9+1;
const int mod9 = 998244353;
const int mod1 = (int)1e9+7;
struct Edge { int to; long long cost; int from; int i; };
bool compe(const Edge &e,const Edge &e2){ return e.cost < e2.cost; }
using Graph = vector<vector<int>>;
using EGraph = vector<Edge>;
using SGraph = vector<vector<ll>>;
template <class T>
int siz(T& a){return (int)a.size();}
double square(double a){ return 1.0*a*a; }
double torad(double x){ return x*(acos(-1)/180.0); }
ll cross(ll ax,ll ay,ll bx,ll by){ return ax*by-ay*bx; }
ll mceil(ll a,ll b){ return (a+b-1)/b; }

struct RollingHash {
    const int base = 1007;
    const int base2 = 2009;

    const int mod = (int)1e9+7;
    const int mod2 = (int)1e9+9;

    vector<ll> hash,pow,hash2,pow2;
    
    RollingHash(string &S){
        int n = siz(S);
        hash.assign(n+1,0); pow.assign(n+1,1);
        hash2.assign(n+1,0); pow2.assign(n+1,1);

        rep(i,n) (pow[i+1] = 1LL*base*pow[i]) %= mod;
        rep(i,n) (hash[i+1] = S[i]+hash[i]*base%mod);
        
        rep(i,n) (pow2[i+1] = 1LL*base2*pow2[i]) %= mod2;
        rep(i,n) (hash2[i+1] = S[i]+hash2[i]*base2%mod2);
    }

    pair<ll,ll> get(int l,int r){
        ll val = hash[r]-hash[l]*pow[r-l]%mod;
        ll val2 = hash2[r]-hash2[l]*pow2[r-l]%mod2;

        if(val < 0) val += mod;
        if(val2 < 0) val2 += mod2;

        return {val,val2};
    }
};

int main(){
    string S; cin >> S;
    int N; cin >> N;
    
    RollingHash rh(S);
    map<pair<ll,ll>,int> mp;
    
    rep(i,siz(S)){
        srep(j,1,10+1){
            if(i+j > siz(S)) break;
            mp[rh.get(i,i+j)]++;
        }
    }

    int ans = 0;
    while(N--){
        string C; cin >> C;
        RollingHash crh(C);

        ans += mp[crh.get(0,siz(C))];
    }

    cout << ans;
}

0