Problem C

题目

Vjudge传送门
HDU传送门
Problem Description
度熊手上有一本神奇的字典，你可以在它里面做如下三个操作：

  1、insert : 往神奇字典中插入一个单词
  2、delete: 在神奇字典中删除所有前缀等于给定字符串的单词
  3、search: 查询是否在神奇字典中有一个字符串的前缀等于给定的字符串

Input
这里仅有一组测试数据。第一行输入一个正整数N(1≤N≤100000)，代表度熊对于字典的操作次数，接下来N行，每行包含两个字符串，中间中用空格隔开。第一个字符串代表了相关的操作（包括： insert, delete 或者 search）。第二个字符串代表了相关操作后指定的那个字符串，第二个字符串的长度不会超过30。第二个字符串仅由小写字母组成。
Output
对于每一个search 操作，如果在度熊的字典中存在给定的字符串为前缀的单词，则输出Yes 否则输出 No。
Sample Input

5
insert hello
insert hehe
search h
delete he
search hello

Sample Output

Yes
No

代码

#include<iostream>
#include<cstring>
#include<sstream>
using namespace std;
#define IOS ios::sync_with_stdio(false)
const int N = 1e6 + 10;
int tr[N][26], tot = 0;
int End[N];
void insert(string s) {
	int pos = 0; int len = s.length();
	for (int i = 0; i < len; i++) {
		int c = s[i] - 'a';
		if (tr[pos][c] == 0 || tr[pos][c] == 2) {
			tr[pos][c] = ++tot;
		}
		pos = tr[pos][c];
		End[pos]++;
	}
}
int query(string s) {
	int pos = 0, len = s.length();
	for (int i = 0; i < len; i++) {
		int c = s[i] - 'a';
		if (tr[pos][c] == 0) return false;
		pos = tr[pos][c];
	}
	return End[pos];
}
void Delete(string s){
	int pos = 0, len = s.length();
	for(int i = 0; i < len; i++){
		int c = s[i] - 'a';
		if(tr[pos][c] == 0) return;
		pos = tr[pos][c];
	}
	int cnt = End[pos];
	pos = 0;
	for(int i = 0; i < len; i++){
		int c = s[i] - 'a';
		pos = tr[pos][c];
		End[pos] -= cnt;
	}
	for(int i = 0; i < 26; i++) tr[pos][i] = 0;
}
int main() {
	int t; cin >> t; getchar();
	string str, a, b;
	while(t--){
		getline(cin, str);
		istringstream is(str);
		is >> a >> b;
		if(a == "insert"){
			insert(b);
		}else if(a == "search"){
			int ans = query(b);
			if(ans) cout << "Yes" << endl;
			else cout << "No" << endl;
		}else{
			Delete(b);
		}
	}
	return 0;
}