結果

問題 No.2761 Substitute and Search
ユーザー 👑 binapbinap
提出日時 2024-04-30 21:37:02
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
MLE  
実行時間 -
コード長 2,994 bytes
コンパイル時間 5,270 ms
コンパイル使用メモリ 283,524 KB
実行使用メモリ 793,428 KB
最終ジャッジ日時 2024-05-07 04:52:50
合計ジャッジ時間 52,216 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 750 ms
8,320 KB
testcase_05 AC 172 ms
8,192 KB
testcase_06 MLE -
testcase_07 MLE -
testcase_08 AC 399 ms
29,060 KB
testcase_09 AC 716 ms
28,312 KB
testcase_10 MLE -
testcase_11 MLE -
testcase_12 MLE -
testcase_13 MLE -
testcase_14 AC 478 ms
28,852 KB
testcase_15 MLE -
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
#include<atcoder/all>
#define rep(i,n) for(int i=0;i<n;i++)
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef vector<int> vi;
typedef vector<long long> vl;
typedef vector<vector<int>> vvi;
typedef vector<vector<long long>> vvl;
typedef long double ld;
typedef pair<int, int> P;

ostream& operator<<(ostream& os, const modint& a) {os << a.val(); return os;}
template <int m> ostream& operator<<(ostream& os, const static_modint<m>& a) {os << a.val(); return os;}
template <int m> ostream& operator<<(ostream& os, const dynamic_modint<m>& a) {os << a.val(); return os;}
template<typename T> istream& operator>>(istream& is, vector<T>& v){int n = v.size(); assert(n > 0); rep(i, n) is >> v[i]; return is;}
template<typename U, typename T> ostream& operator<<(ostream& os, const pair<U, T>& p){os << p.first << ' ' << p.second; return os;}
template<typename T> ostream& operator<<(ostream& os, const vector<T>& v){int n = v.size(); rep(i, n) os << v[i] << (i == n - 1 ? "\n" : " "); return os;}
template<typename T> ostream& operator<<(ostream& os, const vector<vector<T>>& v){int n = v.size(); rep(i, n) os << v[i] << (i == n - 1 ? "\n" : ""); return os;}

template<typename T> void chmin(T& a, T b){a = min(a, b);}
template<typename T> void chmax(T& a, T b){a = max(a, b);}

struct Trie{
	using MP = unordered_map<char, int>;
	vector<MP> to;
	vector<int> cnt;
	vector<set<int>> list;
	Trie(int l) : to(1), cnt(1) {
		list.resize(l + 1);
		list[0].insert(0);
	}
	int add(const string& s) {
		int v = 0;
		int level = 0;
		for(char c : s){
			if(!to[v].count(c)){
				to[v][c] = to.size();
				to.push_back(MP());
				cnt.push_back(0);
			}
			v = to[v][c];
			cnt[v]++;
			level++;
			list[level].insert(v);
		}
		return v;
	}
	int find(const string& s){
		int v = 0;
		for(char c : s){
			if(!to[v].count(c)) return 0;
			v = to[v][c];
		}
		return cnt[v];
	}
	void dfs(int a1, int b1){
		for(auto [c, a2] : to[a1]){
			if(!to[b1].count(c)){
				to[b1][c] = to.size();
				to.push_back(MP());
				cnt.push_back(0);
			}
			int b2 = to[b1][c];
			dfs(a2, b2);
		}
		cnt[b1] += cnt[a1];
		cnt[a1] = 0;
	};
	void merge(int k, char c, char d){
		for(int v : list[k]){
			if(!to[v].count(c)) continue;
			if(to[v].count(c) && !to[v].count(d)){
				to[v][d] = to[v][c];
				to[v].erase(c);
				continue;
			}
			if(to[v].count(c) && to[v].count(d)){
				// merge naively
				// TODO: merge-tech
				int a = to[v][c];
				int b = to[v][d];
				if(cnt[a] <= cnt[b]){
					dfs(a, b);
					to[v].erase(c);
				}else{
					dfs(b, a);
					to[v][d] = to[v][c];
					to[v].erase(c);
				}
				continue;
			}
		}
	}
};

int main(){
	int n, l, q;
	cin >> n >> l >> q;
	vector<string> s(n);
	cin >> s;
	Trie trie(l);
	rep(i, n) trie.add(s[i]);
	rep(_, q){
		int t;
		cin >> t;
		if(t == 1){
			int k;
			char c, d;
			cin >> k >> c >> d;
			trie.merge(k - 1, c, d);
		}
		if(t == 2){
			string t;
			cin >> t;
			cout << trie.find(t) << "\n";
		}
	}
	return 0;
}
0