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

• 問題: No.430 文字列検索
• ユーザー: pekempey
• 提出日時: 2017-02-22 07:15:27
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 9 ms / 2,000 ms
• コード長: 990 bytes
• コンパイル時間: 812 ms
• コンパイル使用メモリ: 87,644 KB
• 実行使用メモリ: 6,752 KB
• 最終ジャッジ日時: 2023-08-29 01:13:53
• 合計ジャッジ時間: 1,916 ms
• ジャッジサーバーID （参考情報）: judge13 / judge12

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 4,376 KB
testcase_01	AC	9 ms; 6,752 KB
testcase_02	AC	5 ms; 4,376 KB
testcase_03	AC	5 ms; 4,380 KB
testcase_04	AC	1 ms; 4,376 KB
testcase_05	AC	2 ms; 4,380 KB
testcase_06	AC	1 ms; 4,380 KB
testcase_07	AC	1 ms; 4,380 KB
testcase_08	AC	3 ms; 4,376 KB
testcase_09	AC	2 ms; 4,380 KB
testcase_10	AC	2 ms; 4,376 KB
testcase_11	AC	8 ms; 4,884 KB
testcase_12	AC	8 ms; 5,092 KB
testcase_13	AC	9 ms; 5,088 KB
testcase_14	AC	7 ms; 4,872 KB
testcase_15	AC	7 ms; 4,940 KB
testcase_16	AC	7 ms; 4,944 KB
testcase_17	AC	7 ms; 4,972 KB

ソースコード

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <array>

using namespace std;

int main() {
	string s;
	cin >> s;
	int m;
	cin >> m;
	vector<array<int, 26>> succ(1);
	vector<int> cnt(1);
	for (int i = 0; i < m; i++) {
		string c;
		cin >> c;
		int u = 0;
		for (char ch : c) {
			ch -= 'A';
			if (succ[u][ch] == 0) {
				succ.push_back({});
				cnt.push_back(0);
				succ[u][ch] = succ.size() - 1;
			}
			u = succ[u][ch];
		}
		cnt[u]++;
	}
	queue<int> q;
	vector<int> link(succ.size());
	q.push(0);
	while (!q.empty()) {
		int u = q.front();
		q.pop();
		for (int i = 0; i < 26; i++) {
			int v = succ[u][i];
			if (v == 0) {
				succ[u][i] = succ[link[u]][i];
			} else {
				link[v] = u == 0 ? 0 : succ[link[u]][i];
				q.push(v);
			}
		}
		cnt[u] += cnt[link[u]];
	}
	int u = 0;
	int ans = 0;
	for (char c : s) {
		u = succ[u][c - 'A'];
		ans += cnt[u];
	}
	cout << ans << endl;
}