結果

問題 No.430 文字列検索
ユーザー sigma425
提出日時 2019-10-01 13:28:13
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 23 ms / 2,000 ms
コード長 2,562 bytes
コンパイル時間 2,893 ms
コンパイル使用メモリ 211,580 KB
最終ジャッジ日時 2025-01-07 20:01:10
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define rep1(i,n) for(int i=1;i<=(int)(n);i++)
#define all(c) c.begin(),c.end()
#define pb push_back
#define fs first
#define sc second
#define chmin(x,y) x=min(x,y)
#define chmax(x,y) x=max(x,y)
using namespace std;
template<class S,class T> ostream& operator<<(ostream& o,const pair<S,T> &p){
	return o<<"("<<p.fs<<","<<p.sc<<")";
}
template<class T> ostream& operator<<(ostream& o,const vector<T> &vc){
	o<<"{";
	for(const T& v:vc) o<<v<<",";
	o<<"}";
	return o;
}
using ll = long long;
template<class T> using V = vector<T>;
template<class T> using VV = vector<vector<T>>;
constexpr ll TEN(int n) { return (n == 0) ? 1 : 10 * TEN(n-1); }

#ifdef LOCAL
#define show(x) cerr << "LINE" << __LINE__ << " : " << #x << " = " << (x) << endl
#else
#define show(x) true
#endif

struct Aho{
	using NP = Aho*;
	map<char,NP> nx;
	NP fail = nullptr, skip = nullptr;
	V<int> matched;		//string ids for this node

	void add(const string& s, int id, int i = 0){
		if(i == (int)s.size()){
			matched.pb(id);
			return;
		}
		if(nx[s[i]] == nullptr){
			nx[s[i]] = new Aho();
		}
		nx[s[i]] -> add(s,id,i+1);
	}
	template<class OP> NP count(OP op, int r){
		if(fail == nullptr) return this;	//root now
		for(int id: matched){
			op(id,r);
		}
		if(skip){
			skip -> count(op,r);
		}else{
			skip = fail -> count(op,r);
		}
		return matched.empty() ? skip : this;
	}

	template<class OP> void match(const string& s, OP op, int i = 0){
		if(i == (int)s.size()) return;
		if(nx.count(s[i]) && nx[s[i]] != nullptr){
			nx[s[i]] -> count(op,i+1);
			nx[s[i]] -> match(s,op,i+1);
		}else{
			if(fail == nullptr){	//root now
				match(s,op,i+1);
			}else{
				fail -> match(s,op,i);
			}
		}
	}

	static NP build(V<string> v){
		NP r = new Aho();
		rep(i,v.size()){
			r -> add(v[i],i);
		}
		queue<NP> que;
		que.push(r);
		assert(r -> fail == nullptr);
		while(!que.empty()){
			NP v = que.front(); que.pop();
			for(auto e: v->nx){
				auto c = e.fs;
				NP u = e.sc;
				NP fail = v -> fail;
				while(fail != nullptr && !fail->nx.count(c)){
					fail = fail -> fail;
				}
				u -> fail = (fail == nullptr) ? r : fail -> nx[c];
				que.push(u);
			}
		}
		return r;
	}
};

int main(){
	cin.tie(0);
	ios::sync_with_stdio(false);		//DON'T USE scanf/printf/puts !!
	cout << fixed << setprecision(20);
	
	string s;
	int N;
	cin >> s >> N;
	V<string> patterns(N);
	rep(i,N) cin >> patterns[i];
	auto AC = Aho::build(patterns);
	int cnt = 0;
	AC -> match(s,[&](int id,int r){
		cnt++;
	});
	cout << cnt << endl;
}
0