[计蒜客 阿里巴巴的手机代理商（中等）] 字典树

[计蒜客 阿里巴巴的手机代理商（中等）] 字典树

分类：Data Structure Trie

1. 题目链接

[计蒜客 阿里巴巴的手机代理商（中等）]

2. 题意描述

阿里巴巴的手机代理商正在研究 infra 输入法的新功能。他们需要分析单词频率以改进用户输入法的体验。于是需要你在系统内核里面写一个 API。 API 有如下功能：

1. 添加操作:添加操作格式为insert barty 8 ，意思为插入barty这个单词，这个单词词频为 88 次。注意如果再次添加insert barty 8操作时，就会将词频增加为 1616 次。（不会出现词频 \le 0≤0 的情况）。
2. 删除操作:删除操作格式为delete barty，意思为删除所有barty这个单词。如果当前没有删除的词汇，输出”Empty”。
3. 查询操作:查询操作格式为query ty，意思为查询当前版本以ty结尾的单词词频总和。
4. 修改操作:修改操作格式为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;
}