fjut 1107 第八集 你明明自己也生病了，却还是要陪着我（字典树）

题意：

首先要写一个添加字符串操作，比如说我输入“1 abc”就是添加字符串abc。然后写查询城市名称操作比如说我输入“2 abcabc”就是查询abcabc是否是所有添加过的字符串中前后缀组合（或本身就是某单词的前缀或后缀），输出“YES”或“NO”。
字符串不会超过100个字符！操作总数不超过50000个。

解析：

利用Trie构建两个字典树，前缀树和后缀树，先将字符串正着和插入前缀树中，再将字符串翻转插入后缀树中。然后每次查找每个字符前缀和后缀最长能抵达的位置，分别记为len1，len2，记当前字符串的长度为len，如果$len1+len2 >= len$，则表面存在字符串的前缀和后缀能构成当前字符串。

$my$ $code$

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxnode = 500005;
const int sigma_size = 26;

struct Trie {
    int ch[maxnode][sigma_size];
    int sz;
    Trie() {sz = 1; memset(ch[0], 0, sizeof(ch[0]));}
    void init() {sz = 1; memset(ch[0], 0, sizeof(ch[0]));}

    int idx(char c) { return c - 'a'; }

    void insert(char *s) {
        int u = 0, n = strlen(s);
        for(int i = 0; i < n; i++) {
            int c = idx(s[i]);
            if(!ch[u][c]) {
                memset(ch[sz], 0, sizeof(ch[sz]));
                ch[u][c] = sz++;
            }
            u = ch[u][c];
        }
    }

    int find(char *s) {
        int u = 0, n = strlen(s);
        int ret = 0;
        for(int i = 0; i < n; i++) {
            int c = idx(s[i]);
            if(!ch[u][c])
                return ret;
            u = ch[u][c];
            ret++;
        }
        return ret;
    }
} pre, suf;

int oper;
char str[105];
int n;

int main() {
    while(scanf("%d", &n) != EOF) {
        pre.init(), suf.init();
        int len, len1, len2;
        while(n--) {
            scanf("%d%s", &oper, str);
            len = strlen(str);
            if(oper == 1) {
                pre.insert(str);
                reverse(str, str+len);
                suf.insert(str);
            }else {
                len1 = pre.find(str);
                reverse(str, str+len);
                len2 = suf.find(str);
                printf("%s\n", (len1 + len2 >= len) ? "YES" : "NO");
            }
        }
    }
    return 0;
}