結果

問題 No.430 文字列検索
ユーザー HaarHaar
提出日時 2019-05-17 07:37:11
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 290 ms / 2,000 ms
コード長 2,726 bytes
コンパイル時間 1,820 ms
コンパイル使用メモリ 179,948 KB
実行使用メモリ 27,228 KB
最終ジャッジ日時 2023-10-17 07:10:05
合計ジャッジ時間 3,532 ms
ジャッジサーバーID
(参考情報)
judge15 / judge14
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 290 ms
27,228 KB
testcase_02 AC 12 ms
4,524 KB
testcase_03 AC 13 ms
4,524 KB
testcase_04 AC 2 ms
4,348 KB
testcase_05 AC 2 ms
4,348 KB
testcase_06 AC 1 ms
4,348 KB
testcase_07 AC 2 ms
4,348 KB
testcase_08 AC 269 ms
26,964 KB
testcase_09 AC 2 ms
4,348 KB
testcase_10 AC 16 ms
6,008 KB
testcase_11 AC 82 ms
8,220 KB
testcase_12 AC 80 ms
8,220 KB
testcase_13 AC 81 ms
8,220 KB
testcase_14 AC 63 ms
6,636 KB
testcase_15 AC 50 ms
5,844 KB
testcase_16 AC 17 ms
4,524 KB
testcase_17 AC 15 ms
4,524 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define FOR(v, a, b) for(int v = (a); v < (b); ++v)
#define FORE(v, a, b) for(int v = (a); v <= (b); ++v)
#define REP(v, n) FOR(v, 0, n)
#define REPE(v, n) FORE(v, 0, n)
#define REV(v, a, b) for(int v = (a); v >= (b); --v)
#define ALL(x) (x).begin(), (x).end()
#define RALL(x) (x).rbegin(), (x).rend()
#define ITR(it, c) for(auto it = (c).begin(); it != (c).end(); ++it)
#define RITR(it, c) for(auto it = (c).rbegin(); it != (c).rend(); ++it)
#define EXIST(c,x) ((c).find(x) != (c).end())
#define LLI long long int
#define fst first
#define snd second
#define popcount __builtin_popcount

#ifdef DEBUG
#include <misc/C++/Debug.cpp>
#else
#define dump(x) ((void)0)
#endif

#define gcd __gcd

using namespace std;
template <class T> constexpr T lcm(T m, T n){return m/gcd(m,n)*n;}

template <typename I> void join(ostream &ost, I s, I t, string d=" "){for(auto i=s; i!=t; ++i){if(i!=s)ost<<d; ost<<*i;}ost<<endl;}
template <typename T> istream& operator>>(istream &is, vector<T> &v){for(auto &a : v) is >> a; return is;}
template <typename T, typename U> istream& operator>>(istream &is, pair<T,U> &p){is >> p.first >> p.second; return is;}

template <typename T, typename U> bool chmin(T &a, const U &b){return (a>b ? a=b, true : false);}
template <typename T, typename U> bool chmax(T &a, const U &b){return (a<b ? a=b, true : false);}
template <typename T, size_t N, typename U> void fill_array(T (&a)[N], const U &v){fill((U*)a, (U*)(a+N), v);}

template <typename T> class RollingHash{
private:
  T str;
  LLI base, mod;
  int sl;
  vector<LLI> shash, pow;
public:
  RollingHash(const T &str, LLI base, LLI mod): str(str), base(base), mod(mod), sl(str.size()), shash(sl+1), pow(sl+1){
    pow[0] = 1;
    shash[0] = 0;
    FORE(i,1,sl){
      shash[i] = (shash[i-1]*base+str[i-1]) % mod;
      pow[i] = pow[i-1]*base % mod;
    }
  }

  LLI hash(int i, int j){ // [i, j)
    return (shash[j] - shash[i] * pow[j-i] + mod*mod) % mod;
  }

  LLI hash(const T &p){
    LLI phash = 0;
    REP(i,(int)p.size()) phash = (phash*base+p[i]) % mod;
    return phash;
  }
  
  vector<int> find(const T &p){
    vector<int> ret;
    int pl = p.size();
    LLI phash = hash(p);
    REPE(i,sl-pl) if(hash(i,i+pl) == phash) ret.push_back(i);
    return ret;
  }
};


int main(){
  cin.tie(0);
  ios::sync_with_stdio(false);

  string S;
  int M;
  while(cin >> S >> M){
    vector<string> C(M); cin >> C;

    RollingHash<string> rh(S, 37, (LLI)1e9+7);

    int L = S.size();

    map<LLI, int> mp;
    REP(i,L){
      FORE(j,1,10){
	if(i+j<=L){
	  ++mp[rh.hash(i,i+j)];
	}
      }
    }
    
    int ans = 0;

    REP(i,M){
      ans += mp[rh.hash(C[i])];
    }
    
    cout << ans << endl;
  }
  
  return 0;
}
0