結果

問題 No.430 文字列検索
コンテスト
ユーザー nii
提出日時 2018-09-14 20:24:18
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 10 ms / 2,000 ms
コード長 1,451 bytes
コンパイル時間 866 ms
コンパイル使用メモリ 82,744 KB
実行使用メモリ 6,348 KB
最終ジャッジ日時 2024-11-10 00:19:47
合計ジャッジ時間 1,412 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

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 count(string S){
		int v=0;
		int ans=0;
		for(int i=0;i<S.size();i++){
			v=su[v][S[i]-'A'];
			ans+=ac[v];
		}
		return ans;
	}
	//
	vector<int> where(string S){
		int v=0;
		vector<int> ans;
		for(int i=0;S.size();i++){
			v=su[v][S[i]-'A'];
			if(ac[v]){
				ans.push_back(i-S.size()+1);
			}
		}
		return ans;
	}
	//
};
//
int main(){
	string S;
	int Q;
	cin>>S>>Q;
	//
	AhoCorasick aho('A','Z');
	while(Q--){
		string C;
		cin>>C;
		aho.add(C);
	}
	//
	aho.build();
	//
	cout<<aho.count(S)<<endl;
	return 0;
}
0