Trie树 模板

普通Trie：

struct TRIE{
    int trie[MAXN][],tot,end[MAXN];
    TRIE(){tot=;}
    void insert(char *s){//s为要插入的字符串
        int len=strlen(s);
        int u=;//1为根节点
        for(int i=;i<len;i++){
            int c=s[i]-'a';//'a'有时需换成'A'或'0'
            if(!trie[u][c])//没有共同前缀，建立一个新的
                trie[u][c]=++tot;//tot为总点数
            u=trie[u][c];//继续向下插入单词
        }
        end[u]=true;//标记是一个出现过的单词（图中涂红色）
    }
    //查找单词是否是词典中某单词的前缀
    bool find1(char *s){//s为要查找的字符串
        int len=strlen(s);
        int u=;//1为根节点
        for(int i=;i<len;i++){
            int c=s[i]-'a';//'a'有时需换成'A'或'0'
            if(!trie[u][c])//单词没有出现，直接返回false
                return false;
            u=trie[u][c];//继续向下查找单词
        }
        //如果扫描完了这个单词
        return true;//是某个单词的前缀
    }
    //查找单词是否在词典中出现过
    bool find2(char *s){//s为要查找的字符串
        int len=strlen(s);
        int u=;//1为根节点
        for(int i=;i<len;i++){
            int c=s[i]-'a';//'a'有时需换成'A'或'0'
            if(!trie[u][c])//单词没有出现，直接返回false
                return false;
            u=trie[u][c];//继续向下查找单词
        }
        //如果扫描完了这个单词
        return end[u];//如果出现过，返回true；如果没有出现过（是前缀），返回false
    }
}Trie;

01Trie：

struct TRIE_01{
    int trie[MAXN*32][2],tot,end[MAXN*32];
    TRIE_01(){tot=1;}
    void insert(int x){
        int root=1;
        for(int i=32;i>=0;i--){
            int t=((x>>i)&1);
            if(!trie[root][t]) trie[root][t]=++tot;
            root=trie[root][t];
        }
		end[root]=x;
    }
	int query(int x){ //查询所有数中和 x异或结果最大的数
		int root=1;
		for(int i=32;i>=0;i--){
		    int v=(x>>i)&1;
		    //利用贪心策略，优先寻找和当前位不同的数
		    if(trie[root][v^1]) root=trie[root][v^1];
		    else root=trie[root][v];
		}
		return end[root]; //返回结果
	}
}Trie_01;

它还可以有平衡树的作用：

题目就是普通平衡树

#include <cstdio>
#include <algorithm>
#include <cstring>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 100010 * 33
using namespace std;
int root=1,tot=1,sumv[maxn],n,opt,x,ch[maxn][2];
void ins(int val,int c){
    val += (int)1e7;
    for(int i=31,p=root,t;i>=0;--i){
        t=(val>>i)&1;
        if(!ch[p][t]) ch[p][t]=++tot;
        p=ch[p][t]; sumv[p]+=c;
    }
}
int rank(int val){
    val += (int)1e7;
    int res=0,p=root;
    for(int i=31;i>=0;--i){
        int t=(val>>i)&1;
        if(t) res += sumv[ch[p][0]];
        p=ch[p][t];
    }
    return res; struct TRIE_01{
    int trie[MAXN*32][2],tot,end[MAXN*32];
    TRIE_01(){tot=1;}
    void insert(int x){
        int root=1;
        for(int i=32;i>=0;i--){
            int t=((x>>i)&1);
            if(!trie[root][t]) trie[root][t]=++tot;
            root=trie[root][t];
        }
		end[root]=x;
    }
	int query(int x){ //查询所有数中和 x异或结果最大的数
		int root=1;
		for(int i=32;i>=0;i--){
		    int v=(x>>i)&1;
		    //利用贪心策略，优先寻找和当前位不同的数
		    if(trie[root][v^1]) root=trie[root][v^1];
		    else root=trie[root][v];
		}
		return end[root]; //返回结果
	}
}Trie_01;

}
int kth(int val){
    int k=root,res=0;
    for(int i=31;i>=0;--i){
        if(val>sumv[ch[k][0]]) res|=(1<<i),val-=sumv[ch[k][0]],k=ch[k][1];
        else k=ch[k][0];
    }
    res-=(1e7); return res;
}
int main(){
    //setIO("input");
    scanf("%d",&n);
    while(n--){
        scanf("%d%d",&opt,&x);
        if(opt==1) ins(x,1);
        else if(opt==2) ins(x,-1);
        else if(opt==3) printf("%d\n",rank(x)+1);
        else if(opt==4) printf("%d\n",kth(x));
        else if(opt==5) printf("%d\n",kth(rank(x)));
        else if(opt==6) printf("%d\n",kth(rank(x+1)+1));
    }
    return 0;
}