数据结构--Trie树

Trie树查找和插入的时间复杂度均为O(n)。

 https://www.acwing.com/problem/content/837/

#include<iostream>
using namespace std;
const int N=1e5+10;//输入字符串的总长度为1e5+10
int son[N][26],cnt[N],idx;//son[i]表示字符串第i个字符是什么，cnt[i]表示字符以第i个字符结尾的字符串有多少个
char s[N];
void insert(char s[]){
    int p=0;
    for(int i=0;s[i];i++){
        int u=s[i]-'a';
        if(!son[p][u]){
            son[p][u]=++idx;
        }
        p=son[p][u];
    }
    cnt[p]++;
}
int query(char s[]){
    int p=0;
    for(int i=0;s[i];i++){
        int u=s[i]-'a';
        if(!son[p][u]){
            return 0;
        }
        p=son[p][u];
    }
    return cnt[p];
}
int main(void){
    int n;
    cin>>n;
    for(int i=0;i<n;i++){
        char op[2];
        cin>>op>>s;
        if(op[0]=='I'){
            insert(s);
        }else{
            cout<<query(s)<<endl;
        }
    }
    return 0;
}

 

 

 https://www.acwing.com/problem/content/145/

#include<iostream>
using namespace std;
const int N=1e5+10,M=N*31;
int a[N];
int son[M][2],idx;
void insert(int x){
    int p=0;
    for(int i=30;i>=0;i--){
        int u=(x>>i)&1;
        if(!son[p][u]) son[p][u]=++idx;
        p=son[p][u];
    }
}
int query(int x){
    int p=0;
    int res=0;
    for(int i=30;i>=0;i--){
        int u=(x>>i)&1;
        if(son[p][!u]) p=son[p][!u],res+=1<<i;
        else p=son[p][u];
    }
    return res;
}
int main(void){
    int n;
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
        insert(a[i]);
    }
    int res=0;
    for(int i=0;i<n;i++){
        res=max(res,query(a[i]));
    }
    cout<<res;
    return 0;
}