結果
問題 | No.430 文字列検索 |
ユーザー |
![]() |
提出日時 | 2016-10-02 22:35:16 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 1,798 bytes |
コンパイル時間 | 1,464 ms |
コンパイル使用メモリ | 112,076 KB |
実行使用メモリ | 20,992 KB |
最終ジャッジ日時 | 2024-11-10 00:01:21 |
合計ジャッジ時間 | 2,619 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 13 WA * 1 |
ソースコード
#define _USE_MATH_DEFINES #include <algorithm> #include <cstdio> #include <functional> #include <iostream> #include <cfloat> #include <climits> #include <cstdlib> #include <cstring> #include <cmath> #include <map> #include <queue> #include <set> #include <sstream> #include <stack> #include <string> #include <time.h> #include <vector> #include <random> #include <list> using namespace std; typedef long double ld; typedef long long ll; typedef unsigned long long ull; typedef pair<int, int> i_i; typedef pair<ll, int> ll_i; typedef pair<int, ll> i_ll; typedef pair<double, int> d_i; typedef pair<ll, ll> ll_ll; typedef pair<double, double> d_d; struct edge { int u, v; ll w; }; #define rep(i, N) for (int i = 0; i < (int)(N); i++) #define pb push_back int INF = INT_MAX / 10; ll MOD = 1000000007; ll _MOD = 1000000009; double EPS = 1e-10; int mod_inv(ll x, int M) { int t = M - 2, y = 1; for (; t; t>>=1) { if (t & 1) y = y * x % M; x = x * x % M; } return y; } struct rolling_hash { int M; vector<int> a, b; rolling_hash(string s, int _M) { M = _M; int N = s.length(); a.resize(N + 1); ll x = 1; for (int i = 0; i < N; i++) { a[i + 1] = (a[i] + x * s[i]) % M; x = x * 256 % M; } ll inv = mod_inv(256, M); b.resize(N + 1); b[0] = 1; for (int i = 0; i < N; i++) b[i + 1] = b[i] * inv % M; } int get(int l, int r) { return ((ll)(a[r] - a[l]) * b[l] % M + M) % M; } }; int main() { string s; cin >> s; int N = s.length(); rolling_hash a(s, _MOD); vector<map<int, int> > mp(11); for (int d = 1; d <= 10; d++) for (int l = 0; l + d <= N; l++) mp[d][a.get(l, l + d)]++; int M; cin >> M; ll ans = 0; while (M--) { string t; cin >> t; rolling_hash b(t, _MOD); ans += mp[t.length()][b.get(0, t.length())]; } cout << ans << endl; }