結果

問題 No.430 文字列検索
ユーザー momoyuumomoyuu
提出日時 2022-09-30 17:38:46
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 350 ms / 2,000 ms
コード長 2,997 bytes
コンパイル時間 3,015 ms
コンパイル使用メモリ 253,356 KB
実行使用メモリ 26,692 KB
最終ジャッジ日時 2023-08-24 05:24:11
合計ジャッジ時間 4,962 ms
ジャッジサーバーID
(参考情報)
judge14 / judge11
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 350 ms
26,692 KB
testcase_02 AC 9 ms
4,376 KB
testcase_03 AC 11 ms
4,380 KB
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 1 ms
4,376 KB
testcase_06 AC 2 ms
4,380 KB
testcase_07 AC 2 ms
4,376 KB
testcase_08 AC 326 ms
26,268 KB
testcase_09 AC 2 ms
4,380 KB
testcase_10 AC 14 ms
5,800 KB
testcase_11 AC 69 ms
7,512 KB
testcase_12 AC 70 ms
7,420 KB
testcase_13 AC 69 ms
7,512 KB
testcase_14 AC 52 ms
5,940 KB
testcase_15 AC 40 ms
4,928 KB
testcase_16 AC 12 ms
4,380 KB
testcase_17 AC 10 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

using ll = long long;
using ull = unsigned long long;
using ld = long double;
template<class t> using vc = vector<t>;
template<class t> using vvc = vc<vc<t>>;
using pi = pair<int,int>;
using pl = pair<ll,ll>;
using vi = vc<int>;
using vvi = vvc<int>;
using vl = vc<ll>;
using vvl = vvc<ll>;

#define rep(i,a,b) for (int i = (int)(a); i < (int)(b); i++)
#define irep(i,a,b) for (int i = (int)(a); i > (int)(b); i--)
#define all(a) a.begin(),a.end()
#define print(n) cout << n << '\n'
#define pritn(n) print(n)
#define printv(n,a) {copy(all(n),ostream_iterator<a>(cout," ")); cout<<"\n";}
#define printvv(n,a) {for(auto itr:n) printv(itr,a);}
#define rup(a,b) (a+b-1)/b
#define input(A,N) rep(i,0,N) cin>>A[i]
#define chmax(a,b) a = max(a,b)
#define chmin(a,b) a = min(a,b)

const int mod = 998244353;
struct mint{
    long long x;
    mint(long long x=0):x((x%mod+mod)%mod){}
    mint operator-() const{
        return mint(-x);
    }
    mint& operator+=(const mint& a){
        if((x+=a.x)>=mod)x-=mod;
        return *this;
    }
    mint& operator-=(const mint& a){
        if((x+=mod-a.x)>=mod)x-=mod;
        return *this;
    }
    mint& operator*=(const  mint& a){
        (x *= a.x) %= mod;
        return *this;
    }
    mint operator+(const mint& a) const{
        mint res(*this);
        return res+=a;
    }
    mint operator-(const mint& a) const{
        mint res(*this);
        return res-=a;
    }
    mint operator*(const mint& a) const{
        mint res(*this);
        return res*=a;
    }
    mint pow(long long n) const {
        assert(0 <= n);
        mint a = *this, r = 1;
        while (n) {
            if (n & 1) r *= a;
            a *= a;
            n >>= 1;
        }
        return r;
    }
    mint inv() const{
        return pow(mod-2);
    }
    mint& operator/=(const mint& a){
        return (*this)*=a.inv();
    }
    mint operator/(const mint& a) const {
        mint res(*this);
        return res/=a;
    }
    friend ostream& operator<<(ostream& os, const mint& m){
        os << m.x;
        return os;
    }
    bool operator==(const mint& a) const {
        return x == a.x;
    }
    bool operator<(const mint& a) const{
        return x < a.x;
    }
};

int main(){
    cout << fixed << setprecision(15);
    
    string s;
    cin>>s;
    int n = s.size();
    mint B(100);
    vc<mint> fac(n+1,1);
    rep(i,1,n+1)fac[i]=fac[i-1]*B;

    ll ans = 0;
    int q;
    cin>>q;
    vc<map<mint,int>> d(11);
    rep(i,0,q){
        string t;
        cin>>t;
        int m = t.size();
        mint h(0);
        rep(i,0,m){
            h *= B;
            h += mint(t[i]);
        }
        d[m][h]++;
    }
    rep(i,0,n){
        mint hash(0);
        for(int j = 1;j<=10;j++){
            hash *= B;
            int ni = i + j - 1;
            if(ni>=n)break;
            hash += mint(s[ni]);
            ans += d[j][hash];
        }
    }
    print(ans);
    
    //system("pause");
    return 0;
}
0