結果

問題 No.430 文字列検索
コンテスト
ユーザー はまやんはまやん
提出日時 2017-02-24 03:23:01
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 14 ms / 2,000 ms
コード長 2,830 bytes
コンパイル時間 1,759 ms
コンパイル使用メモリ 182,624 KB
実行使用メモリ 7,920 KB
最終ジャッジ日時 2024-11-10 00:14:57
合計ジャッジ時間 2,441 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=a;i<b;i++)





#define CMAX 26
#define OFFSET 'A'
struct Node { int nxt[CMAX + 1]; int exist; vector<int> accept;
Node() : exist(0){ memset(nxt, -1, sizeof(nxt)); }};
struct Trie {
	vector<Node> nodes; int root;
	Trie() : root(0){ nodes.push_back(Node()); }
	void update_direct(int node, int id) { nodes[node].accept.push_back(id); }
	void update_child(int node, int child, int id) { ++nodes[node].exist; }
	void add(const string &str, int str_index, int node_index, int id)
	{
		if (str_index == str.size()) update_direct(node_index, id);
		else {
			const int c = str[str_index] - OFFSET;
			if (nodes[node_index].nxt[c] == -1) {
				nodes[node_index].nxt[c] = (int)nodes.size();
				nodes.push_back(Node());
			}
			add(str, str_index + 1, nodes[node_index].nxt[c], id);
			update_child(node_index, nodes[node_index].nxt[c], id);
		}
	}
	void add(const string &str, int id) { add(str, 0, 0, id); }
	void add(const string &str) { add(str, nodes[0].exist); }
	int size() { return (nodes[0].exist); }
	int nodesize() { return ((int)nodes.size()); }
};

struct Aho_Corasick : Trie {
	static const int FAIL = CMAX; vector<int> correct;
	Aho_Corasick() : Trie() {}
	void build() {
		correct.resize(nodes.size());
		rep(i, 0, nodes.size()) correct[i] = (int)nodes[i].accept.size();

		queue<int> que;
		rep(i, 0, CMAX + 1) {
			if (~nodes[0].nxt[i]) {
				nodes[nodes[0].nxt[i]].nxt[FAIL] = 0;
				que.emplace(nodes[0].nxt[i]);
			} else nodes[0].nxt[i] = 0;
		}
		while (!que.empty()) {
			Node &now = nodes[que.front()];
			correct[que.front()] += correct[now.nxt[FAIL]];
			que.pop();
			for (int i = 0; i < CMAX; i++) {
				if (now.nxt[i] == -1) continue;
				int fail = now.nxt[FAIL];
				while (nodes[fail].nxt[i] == -1) {
					fail = nodes[fail].nxt[FAIL];
				}
				nodes[now.nxt[i]].nxt[FAIL] = nodes[fail].nxt[i];

				auto &u = nodes[now.nxt[i]].accept;
				auto &v = nodes[nodes[fail].nxt[i]].accept;
				vector<int> accept;
				set_union(begin(u), end(u), begin(v), end(v), back_inserter(accept));
				u = accept;
				que.emplace(now.nxt[i]);
			}
		}
	}

	int match(const string &str, vector< int > &result, int now = 0) {
		result.assign(size(), 0);
		int count = 0;
		for (auto &c : str) {
			while (nodes[now].nxt[c - OFFSET] == -1) now = nodes[now].nxt[FAIL];
			now = nodes[now].nxt[c - OFFSET];
			count += correct[now];
			for (auto &v : nodes[now].accept) ++result[v];
		}
		return count;
	}
};
//-----------------------------------------------------------------
string S;
int M;
int cnt[5010];
//-----------------------------------------------------------------
int main() {
	cin >> S >> M;
	Aho_Corasick ac;
	rep(i, 0, M) {
		string s;
		cin >> s;
		ac.add(s);
	}

	ac.build();
	vector<int> cnt(M);
	cout << ac.match(S, cnt) << endl;
}
0