結果

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

ソースコード

diff # 

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




const int NUMC = 26;
struct Trie {
	vector<vector<int>> V;
	vector<int> C;
	void create(vector<string> S) {
		V.clear();
		V.push_back(vector<int>(NUMC + 1));
		C.push_back(0);
		sort(S.begin(), S.end());
		for (string s : S) {
			int cur = 0;
			for (char _c : s) {
				int c = _c - 'A';
				if (V[cur][c + 1] == 0) {
					V.push_back(vector<int>(NUMC + 1));
					V[cur][c + 1] = V.size() - 1;
					C.push_back(0);
				}
				cur = V[cur][c + 1];
			}
			C[cur]++;
		}
	}
	int next(int cu, char c) {
		assert(cu < V.size());
		return V[cu][c - 'A' + 1];
	}
};
//-----------------------------------------------------------------
string S;
int M;
vector<string> C;
Trie trie;
int dp[50101][1010];
//-----------------------------------------------------------------
int main() {
	cin >> S >> M;
	rep(i, 0, M) {
		string s;
		cin >> s;
		C.push_back(s);
	}

	trie.create(C);

	int ans = 0;
	int N = S.length();
	int NN = trie.V.size();

	rep(i, 0, N) {
		dp[i][0] = 1;
		rep(j, 0, NN) if(0 < dp[i][j]) dp[i + 1][trie.next(j, S[i])] += dp[i][j];
		rep(j, 0, NN) if (0 < trie.C[j]) ans += dp[i + 1][j];
	}

	cout << ans << endl;
}
0