結果

問題 No.430 文字列検索
ユーザー sigma425sigma425
提出日時 2019-10-01 13:28:13
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 17 ms / 2,000 ms
コード長 2,562 bytes
コンパイル時間 2,119 ms
コンパイル使用メモリ 219,796 KB
実行使用メモリ 7,060 KB
最終ジャッジ日時 2024-11-10 00:35:15
合計ジャッジ時間 2,798 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 17 ms
7,060 KB
testcase_02 AC 5 ms
5,248 KB
testcase_03 AC 5 ms
5,248 KB
testcase_04 AC 1 ms
5,248 KB
testcase_05 AC 1 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,888 KB
testcase_12 AC 10 ms
6,400 KB
testcase_13 AC 10 ms
6,400 KB
testcase_14 AC 8 ms
5,680 KB
testcase_15 AC 6 ms
5,248 KB
testcase_16 AC 6 ms
5,248 KB
testcase_17 AC 6 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

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