#285054 (C++14) No.430 文字列検索

• 問題: No.430 文字列検索
• ユーザー: nii
• 提出日時: 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
• コンパイル時間: 928 ms
• コンパイル使用メモリ: 82,340 KB
• 実行使用メモリ: 6,220 KB
• 最終ジャッジ日時: 2024-11-10 00:19:45
• 合計ジャッジ時間: 1,577 ms
• ジャッジサーバーID （参考情報）: judge1 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	11 ms; 6,220 KB
testcase_02	AC	6 ms; 5,248 KB
testcase_03	AC	6 ms; 5,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	2 ms; 5,248 KB
testcase_06	AC	2 ms; 5,248 KB
testcase_07	AC	2 ms; 5,248 KB
testcase_08	AC	3 ms; 5,248 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	2 ms; 5,248 KB
testcase_11	AC	9 ms; 5,248 KB
testcase_12	AC	9 ms; 5,740 KB
testcase_13	AC	11 ms; 5,612 KB
testcase_14	AC	9 ms; 5,248 KB
testcase_15	AC	7 ms; 5,248 KB
testcase_16	AC	7 ms; 5,248 KB
testcase_17	AC	8 ms; 5,248 KB

ソースコード

#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;
}