結果

問題 No.430 文字列検索
ユーザー lynmisakura
提出日時 2022-03-21 16:47:46
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 419 ms / 2,000 ms
コード長 2,793 bytes
コンパイル時間 2,405 ms
コンパイル使用メモリ 207,180 KB
最終ジャッジ日時 2025-01-28 10:55:30
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

// code by lynmisakura. wish to be accepted!
#include<bits/stdc++.h>

using namespace std;

#ifdef LOCAL
#include "/Users/debug.h"
#else
#define debug(...) 42
#endif

using ll = long long;
using ull = unsigned long long;
using ld = long double;
using vi = vector<int>;
using vvi = vector<vi>;
using vl = vector<ll>;
using vvl = vector<vl>;
using pi = pair<int, int>;
using pl = pair<ll, ll>;
using vpi = vector<pi>;
using vpl = vector<pl>;

#define REP(i, N) for(int i = 0;i < N;i++)

#define mp make_pair
#define pb push_back
#define eb emplace_back
#define all(x) (x).begin(),(x).end()

template<class T>
void init(vector<T> &a, int N) {
  a.resize(N);
}

template<class T>
void init(vector<T> &a, int N, T val) {
  a.assign(N, val);
}

template<class T>
bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return true;
  } else { return false; }
}

template<class T>
bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return true;
  } else { return false; }
}

template<class T>
T extgcd(T a, T b, T &x, T &y) {
  T d = a;
  if (b != 0) {
    d = extgcd(b, a % b, y, x);
    y -= (a / b) * x;
  } else {
    x = 1;
    y = 0;
  }
  return d;
}

template<class T>
T modinv(T a, T m) {
  T x, y;
  extgcd(a, m, x, y);
  return (m + x % m) % m;
}

template<class S, int64_t P, int64_t B = 31>
class rolling_hash {

public:
  rolling_hash(vector<S> &a) : a(a) {
    init();
  }

  rolling_hash(std::string &s, char c = 'a') {
    a.resize((int)s.length());
    for (int i = 0; i < (int)s.length(); i++) {
      a[i] = s[i] - c + 1;
    }
    init();
  }

  void init() {
    N = a.size();
    hash.assign(N + 1, 0);
    b_inv.assign(N + 1, 0);

    long b_pow = 1;
    for (int i = 0; i < N; i++) {
      hash[i + 1] = (hash[i] + a[i] * b_pow) % P;
      b_pow = (b_pow * B) % P;
    }

    b_inv[0] = 1;
    b_inv[1] = modinv(B, P);
    for (int i = 2; i <= N; i++) {
      b_inv[i] = (b_inv[i - 1] * b_inv[1]) % P;
    }
  }

  int64_t f(int l, int r) {
    int64_t res = (hash[r] - hash[l] + P) % P;
    res = (res * b_inv[l]) % P;
    return res;
  }
  int64_t f_all(){
    return f(0,N);
  }

private:

  int N;
  vector<S> a;
  vector<int64_t> hash;
  vector<int64_t> b_inv;
};

int main() {
  
  string s;
  cin >> s;
  int len = s.length();

  rolling_hash<ll,998244353,31> rh_1(s,'A');
  rolling_hash<ll,int(1e9)+7,31> rh_2(s,'A');
  
  map<pl,int> m;
  REP(i,len)for(int j = 1;j <= 10 && i + j <= len;j++){
    ll h_1 = rh_1.f(i,i + j);
    ll h_2 = rh_2.f(i,i + j);
    m[mp(h_1,h_2)]++;
  }
  
  ll ans = 0;
  int q;
  cin >> q;
  while(q--){
    string c;
    cin >> c;
    rolling_hash<ll,998244353,31> rh_3(c,'A');
    rolling_hash<ll,int(1e9)+7,31> rh_4(c,'A');
    ll h_3 = rh_3.f_all();
    ll h_4 = rh_4.f_all();
    ans += m[mp(h_3,h_4)];
  }
  cout << ans << '\n';

}
0