[计蒜客 阿里巴巴的手机代理商(中等)] 字典树
分类:Data Structure
Trie
1. 题目链接
2. 题意描述
阿里巴巴的手机代理商正在研究 infra 输入法的新功能。他们需要分析单词频率以改进用户输入法的体验。于是需要你在系统内核里面写一个 API。 API 有如下功能:
- 添加操作:添加操作格式为insert barty 8 ,意思为插入barty这个单词,这个单词词频为 88 次。注意如果再次添加insert barty 8操作时,就会将词频增加为 1616 次。(不会出现词频 \le 0≤0 的情况)。
- 删除操作:删除操作格式为delete barty,意思为删除所有barty这个单词。如果当前没有删除的词汇,输出”Empty”。
- 查询操作:查询操作格式为query ty,意思为查询当前版本以ty结尾的单词词频总和。
- 修改操作:修改操作格式为update ty tied,意思为将所有结尾是ty的单词更新为tied结尾,比如barty会变为 bartied。如果不存在ty结尾的单词,输出Empty。如果已经存在tied结尾的单词,那么说明存在 conflict。 不做合并,输出Conflict。如果既不存在ty结尾的单词,也已经存在以tied结尾的单词,则输出Empty。
3. 解题思路
建立一棵逆序的字典树。
但是update str1 str2
的时候,有一个case让我WA到怀疑人生。
当str1
是str2
的子串的时候。。。。
4. 实现代码
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1100000;
int T, n;
struct TNode {
unsigned long long cnt;
TNode *ch[26];
void init() {
cnt = 0;
for(int i = 0; i < 26; ++i) ch[i] = NULL;
}
} nd[MAXN], *rt;
int tot;
void init() {
tot = 0;
nd[tot].init();
rt = &nd[tot ++];
}
void insert(char s[], int len, unsigned long long cnt) {
TNode *ptr = rt;
for(int i = len - 1; i >= 0; --i) {
ptr->cnt += cnt;
assert(ptr->cnt >= 0);
int x = s[i] - 'a';
if(ptr->ch[x] == NULL) {
nd[tot].init();
ptr->ch[x] = &nd[tot ++];
assert(tot <= MAXN);
}
ptr = ptr->ch[x];
}
ptr->cnt += cnt;
assert(ptr->cnt >= 0);
}
unsigned long long query(char s[], int len, bool prefix = true) {
TNode *ptr = rt;
for(int i = len - 1; i >= 0; --i) {
int x = s[i] - 'a';
if(ptr->ch[x] == NULL) return 0;
ptr = ptr->ch[x];
}
unsigned long long ret = ptr->cnt;
if(prefix == false) {
for(int i = 0; i < 26; ++i) ret -= ptr->ch[i] == NULL ? 0 : ptr->ch[i]->cnt;
}
assert(ret >= 0);
return ret;
}
int delete_(char s[], int len) {
unsigned long long cnt = query(s, len, false);
if(cnt == 0) return 0;
insert(s, len, -cnt);
return 1;
}
int update(char s1[], int len1, char s2[], int len2) {
unsigned long long cnt1 = query(s1, len1);
if(cnt1 == 0) return 0;
unsigned long long cnt2 = query(s2, len2);
if(cnt2 > 0) return -1;
TNode *ptr1 = rt;
for(int i = len1 - 1; i >= 0; --i) {
ptr1->cnt -= cnt1;
int x = s1[i] - 'a';
ptr1 = ptr1->ch[x];
}
ptr1->cnt -= cnt1;
TNode temp = *ptr1;
for(int i = 0; i < 26; ++i) ptr1->ch[i] = NULL;
TNode *ptr2 = rt;
for(int i = len2 - 1; i >= 0; --i) {
ptr2->cnt += cnt1;
int x = s2[i] - 'a';
if(ptr2->ch[x] == NULL) {
nd[tot].init();
ptr2->ch[x] = &nd[tot ++];
assert(tot <= MAXN);
}
ptr2 = ptr2->ch[x];
}
ptr2->cnt += cnt1;
memcpy(ptr2->ch, temp.ch, sizeof(temp.ch));
return 1;
}
char oper[50], str[MAXN], str2[MAXN];
int main() {
#ifdef __LOCAL_WONZY__
freopen("input.txt", "r", stdin);
#endif
int ret;
unsigned long long cnt;
scanf("%d", &T);
while(T --) {
scanf("%d", &n);
init();
for(int i = 1; i <= n; ++i) {
scanf("%s", oper);
switch (oper[0]) {
case 'i':
scanf("%s %llu", str, &cnt);
insert(str, strlen(str), cnt);
break;
case 'd':
scanf("%s", str);
ret = delete_(str, strlen(str));
if(ret == 0) printf("Empty\n");
break;
case 'q':
scanf("%s", str);
cnt = query(str, strlen(str));
printf("%llu\n", cnt);
break;
case 'u':
scanf("%s %s", str, str2);
ret = update(str, strlen(str), str2, strlen(str2));
if(ret == 0) printf("Empty\n");
else if(ret == -1) printf("Conflict\n");
break;
}
}
}
return 0;
}