結果

問題 No.430 文字列検索
ユーザー kenta255kenta255
提出日時 2019-08-12 11:41:42
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,792 bytes
コンパイル時間 1,999 ms
コンパイル使用メモリ 201,728 KB
実行使用メモリ 4,476 KB
最終ジャッジ日時 2023-10-13 11:40:48
合計ジャッジ時間 17,200 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,348 KB
testcase_01 WA -
testcase_02 AC 1,448 ms
4,348 KB
testcase_03 AC 1,439 ms
4,352 KB
testcase_04 AC 2 ms
4,352 KB
testcase_05 AC 1 ms
4,352 KB
testcase_06 AC 2 ms
4,352 KB
testcase_07 AC 2 ms
4,348 KB
testcase_08 AC 7 ms
4,352 KB
testcase_09 AC 1 ms
4,348 KB
testcase_10 AC 5 ms
4,348 KB
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 AC 1,441 ms
4,348 KB
testcase_16 AC 1,419 ms
4,348 KB
testcase_17 AC 1,432 ms
4,348 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define P pair<ll,ll>
#define FOR(I,A,B) for(ll I = ll(A); I < ll(B); ++I)
#define FORR(I,A,B) for(ll I = ll((B)-1); I >= ll(A); --I)
#define TO(x,t,f) ((x)?(t):(f))
#define SORT(x) (sort(x.begin(),x.end())) // 0 2 2 3 4 5 8 9
#define POSL(x,v) (lower_bound(x.begin(),x.end(),v)-x.begin()) //xi>=v  x is sorted
#define POSU(x,v) (upper_bound(x.begin(),x.end(),v)-x.begin()) //xi>v  x is sorted
#define NUM(x,v) (POSU(x,v)-POSL(x,v))  //x is sorted
#define REV(x) (reverse(x.begin(),x.end())) //reverse
ll gcd(ll a,ll b){if(a%b==0)return b;return gcd(b,a%b);}
ll lcm(ll a,ll b){ll c=gcd(a,b);return ((a/c)*(b/c)*c);}
#define NEXTP(x) next_permutation(x.begin(),x.end())
const ll INF=ll(1e18)+ll(7);
const ll MOD=1000000007LL;
#define out(a) cout<<fixed<<setprecision((a))




// http://perogram.hateblo.jp/entry/rolling_hash
vector<ll> rolling_hash_search(string s, string t){
	vector<ll> index_list;
	ll SL = s.size();
	ll TL = t.size();

	// b, hは互いに素 (gcd == 1)
	ll b = 1e6 + 7;
	ll h = 1e9 + 7;

	// 蟻本でいう、bのm乗
	ll a = 1;
	FOR(i, 0, TL){
		a *= b;
		a %= h;
	}
	
	// 先頭のローリングハッシュ
	ll s_hash = 0;
	FOR(i, 0, TL){
		s_hash = s_hash * b + s[i];
		s_hash %= h;    
	}

	ll t_hash = 0;
	FOR(i, 0, TL){
		t_hash = t_hash * b + t[i];
		t_hash %= h;    
	}

	// sの始点を1つずつ進めながらハッシュ値をチェック
	FOR(i, 0, SL - TL + 1){
		if(s_hash==t_hash){
			index_list.push_back(i);
		}

		if(i+TL<SL){
			s_hash = s_hash * b - s[i] * a + s[i+TL];
			s_hash %= h;
		}
	}

	return index_list;
}

int main(){
	string s,t;
	ll M,ans=0;
	cin >> s >> M;
	FOR(i,0,M){
		cin >> t;
		ans += rolling_hash_search(s, t).size();
	}
	cout << ans << endl;
	
	return 0;
}
0