結果

問題 No.430 文字列検索
ユーザー niinii
提出日時 2018-09-14 14:10:43
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 11 ms / 2,000 ms
コード長 1,213 bytes
コンパイル時間 956 ms
コンパイル使用メモリ 82,484 KB
実行使用メモリ 5,980 KB
最終ジャッジ日時 2023-09-18 12:41:33
合計ジャッジ時間 2,242 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,356 KB
testcase_01 AC 11 ms
5,980 KB
testcase_02 AC 6 ms
4,376 KB
testcase_03 AC 5 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,376 KB
testcase_07 AC 2 ms
4,380 KB
testcase_08 AC 3 ms
4,380 KB
testcase_09 AC 2 ms
4,380 KB
testcase_10 AC 1 ms
4,376 KB
testcase_11 AC 10 ms
5,284 KB
testcase_12 AC 10 ms
5,420 KB
testcase_13 AC 10 ms
5,372 KB
testcase_14 AC 8 ms
4,924 KB
testcase_15 AC 7 ms
4,436 KB
testcase_16 AC 7 ms
4,376 KB
testcase_17 AC 7 ms
4,464 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
//
struct AhoCorasick{
	//
	char from,to;
	vector<vector<int>> su;
	vector<int> wa,ac;
	//
	AhoCorasick(char from='a',char to='z'):from(from),to(to),su(1,vector<int>(to-from+1)),ac(1){}
	//
	void add(string S){
		int now=0;
		for(int i=0;i<S.size();i++){
			char c=S[i];
			c-=from;
			if(su[now][c]==0){
				su[now][c]=su.size();
				su.push_back(vector<int>(to-from+1));
				ac.push_back(0);
			}
			now=su[now][c];
		}
		ac[now]++;
	}
	//
	void build(){
		queue<int> que;
		for(int i=0;i<=to-from;i++){
			if(su[0][i]!=0){
				que.push(su[0][i]);
			}
		}
		wa.resize(su.size());
		while(!que.empty()){
			int now=que.front();
			que.pop();
			for(int i=0;i<=to-from;i++){
				int nex=su[now][i];
				if(nex!=0){
					que.push(nex);
					wa[nex]=su[wa[now]][i];
				}else{
					su[now][i]=su[wa[now]][i];
				}
			}
			ac[now]+=ac[wa[now]];
		}
	}
	//
};
//
int main(){
	string S;
	int Q;
	cin>>S>>Q;
	//
	AhoCorasick aho('A','Z');
	while(Q--){
		string C;
		cin>>C;
		aho.add(C);
	}
	//
	aho.build();
	//
	int v=0;
	int ans=0;
	for(int i=0;i<S.size();i++){
		v=aho.su[v][S[i]-'A'];
		ans+=aho.ac[v];
	}
	cout<<ans<<endl;
	return 0;
}
0