蒟蒻的trie树专题

POJ 3630 Phone List: 模板

///meek

#include<bits/stdc++.h>
using namespace std;

using namespace std ;
typedef long long ll;
#define mem(a) memset(a,0,sizeof(a))
#define pb push_back
inline ll read()
{
    ll x=,f=;char ch=getchar();
    while(ch<''||ch>''){
        if(ch=='-')f=-;ch=getchar();
    }
    while(ch>=''&&ch<=''){
        x=x*+ch-'';ch=getchar();
    }return x*f;
}
//****************************************
const int  N=+;
#define mod 10000007

#define inf 10000007
#define maxn 10000

char a[N];
struct Trie {
       int ch[N][],sum[N];
       int siz;
       Trie() { siz=;
                mem(ch);mem(sum);
       }
       int idx(char c) {return c-'a';}
       void insertt(char *s) {
            int u=,n=strlen(s);
            for(int i=;i<n;i++) {
                int c=idx(s[i]);
                if(!ch[u][c]) {
                    ch[u][c]=siz++;
                }
                u=ch[u][c];sum[u]++;
            }
       }
       int ask(char *s) {
           int u=,n=strlen(s);
           for(int i=;i<n;i++) {
             int c=idx(s[i]);
             if(ch[u][c]) u=ch[u][c];
             else return ;
           }
           return sum[u];
       }
}trie;
int main() {
    while(gets(a)) {
        if(strlen(a)==)break;
        trie.insertt(a);
    }
    while(scanf("%s",a)!=EOF) {
        cout<<trie.ask(a)<<endl;
    }
    return ;
}

POJ3630

HDU 1004 Let the Balloon Rise：模板

///meek

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

using namespace std ;
typedef long long ll;
#define mem(a) memset(a,0,sizeof(a))
#define pb push_back
inline ll read()
{
    ll x=,f=;char ch=getchar();
    while(ch<''||ch>''){
        if(ch=='-')f=-;ch=getchar();
    }
    while(ch>=''&&ch<=''){
        x=x*+ch-'';ch=getchar();
    }return x*f;
}
//****************************************
const int  N=+;
#define mod 10000007

#define inf 10000007
#define maxn 10000

struct Trie{
    int ch[N][],sum[N],siz=,W=,ansi,ans;
    void init() {mem(ch),mem(sum),siz=;W=,ansi=;ans=-;}
    int idx(char c) {return c-'a';}
    void insertt(char *s) {
        int u=,n=strlen(s);
         W++;
        for(int i=;i<n;i++) {
            int c=idx(s[i]);
            if(!ch[u][c]) {
                ch[u][c]=siz++;
            }
            u=ch[u][c],sum[u]++;
            if(ans<sum[u]&&i==n-) {ans=sum[u];ansi=W;}
        }
    }
}trie;
int main() {
   char a[N][];int n;
    while(scanf("%d",&n)&&n) {
            trie.init();
        for(int i=;i<=n;i++) {
            scanf("%s",a[i]);
            trie.insertt(a[i]);
        }
        cout<<a[trie.ansi]<<endl;
    }
    return ;
}

HDU1004

HDU 1251统计难题 ：查找前缀个数

///meek

#include<bits/stdc++.h>
using namespace std;

using namespace std ;
typedef long long ll;
#define mem(a) memset(a,0,sizeof(a))
#define pb push_back
inline ll read()
{
    ll x=,f=;char ch=getchar();
    while(ch<''||ch>''){
        if(ch=='-')f=-;ch=getchar();
    }
    while(ch>=''&&ch<=''){
        x=x*+ch-'';ch=getchar();
    }return x*f;
}
//****************************************
const int  N=+;
#define mod 10000007

#define inf 10000007
#define maxn 10000

char a[N];
struct Trie {
       int ch[N][],sum[N];
       int siz;
       Trie() { siz=;
                mem(ch);mem(sum);
       }
       int idx(char c) {return c-'a';}
       void insertt(char *s) {
            int u=,n=strlen(s);
            for(int i=;i<n;i++) {
                int c=idx(s[i]);
                if(!ch[u][c]) {
                    ch[u][c]=siz++;
                }
                u=ch[u][c];sum[u]++;
            }
       }
       int ask(char *s) {
           int u=,n=strlen(s);
           for(int i=;i<n;i++) {
             int c=idx(s[i]);
             if(ch[u][c]) u=ch[u][c];
             else return ;
           }
           return sum[u];
       }
}trie;
int main() {
    while(gets(a)) {
        if(strlen(a)==)break;
        trie.insertt(a);
    }
    while(scanf("%s",a)!=EOF) {
        cout<<trie.ask(a)<<endl;
    }
    return ;
}

HDU1251

HDU 4825 Xor Sum : 给你n个数m个询问，每隔询问一个A，问你这n个数中与A异或值最大是多少，

将树转化成01 串的trie树，经典题型

///meek

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

using namespace std ;
typedef long long ll;
#define mem(a) memset(a,0,sizeof(a))
#define pb push_back
inline ll read()
{
    ll x=,f=;char ch=getchar();
    while(ch<''||ch>''){
        if(ch=='-')f=-;ch=getchar();
    }
    while(ch>=''&&ch<=''){
        x=x*+ch-'';ch=getchar();
    }return x*f;
}
//****************************************
const int  N=+;
#define mod 10000007

#define inf 10000007
#define maxn 10000

struct Trie{
   int ch[N][],sum[N],siz=,ans,W;
   void init(){mem(sum),mem(ch),siz=;}
   void insertt(int x) {
       int tmp=x;
        int each[],k=;mem(each);
        while(x) {
            each[k++]=x%;
            x/=;
        }int u=;
        for(int i=;i>=;i--) {
            int c=each[i];
            if(!ch[u][c]){
                ch[u][c]=siz++;
            }
            u=ch[u][c];if(i==)sum[u]=tmp;
        }
   }
   int ask(int x) {
       int u=;int tmp=x;
        int each[],k=;mem(each);
         while(x) {
            each[k++]=x%;
            x/=;
        }
        for(int i=;i>=;i--) {
            int g,c=each[i];if(c==)g=;else g=;
            if(ch[u][g]) {
                u=ch[u][g];
            }
            else u=ch[u][c];
            if(i==) return sum[u];
        }
   }
}trie;
int main() {

    int T=read(),n,m,z[N];
    int oo=;
    while(T--) {int a[N];
         trie.init();
        scanf("%d%d",&n,&m);
        for(int i=;i<=n;i++) {
            scanf("%d",&a[i]);
            trie.insertt(a[i]);
        }
        for(int i=;i<=m;i++) {
            scanf("%d",&z[i]);
        } printf("Case #%d:\n",oo++);
        for(int i=;i<=m;i++) {
            cout<<trie.ask(z[i])<<endl;
        }
    }
    return ;
}

HDU 4825

POJ 3764 The xor-longest Path:

题意：给你一个树，及n-1条边权，问你任意一条路径上最大异或边权值是多少

题解： 我们根据异或性质，对于u->v这条路径上异或值可以转化为 XOR（0->v）^XOR（0->u）

我们先dfs出根节点到任意节点的路径异或值，于是转化成任意两个数的 异或最大值了

///meek

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

using namespace std ;
typedef long long ll;
#define mem(a) memset(a,0,sizeof(a))
#define pb push_back
inline ll read()
{
    ll x=,f=;char ch=getchar();
    while(ch<''||ch>''){
        if(ch=='-')f=-;ch=getchar();
    }
    while(ch>=''&&ch<=''){
        x=x*+ch-'';ch=getchar();
    }return x*f;
}
//****************************************
const int  N=+;
#define mod 10000007

#define inf 10000007
#define maxn 10000

int a[N],head[N],t,n;
struct ss {
   int to,next,va;
}e[N*];
void init() {mem(head),t=;}
void add(int u,int v,int w) {e[t].to=v;e[t].va=w;e[t].next=head[u];head[u]=t++;}
void dfs(int x,int pre) {
     for(int i=head[x];i;i=e[i].next) {
        if(e[i].to==pre) continue;
        a[e[i].to]=e[i].va^a[x];
        dfs(e[i].to,x);
     }
}
struct Trie{
   int ch[N][],sum[N],siz,ans,W;
   void init(){mem(sum),mem(ch),siz=;}
   void insertt(int x) {
       int tmp=x;
        int each[],k=;mem(each);
        while(x) {
            each[k++]=x%;
            x/=;
        }int u=;
        for(int i=;i>=;i--) {
            int c=each[i];
            if(!ch[u][c]){
                ch[u][c]=siz++;
            }
            u=ch[u][c];if(i==)sum[u]=tmp;
        }
   }
   int ask(int x) {
       int u=;int tmp=x;
        int each[],k=;mem(each);
         while(x) {
            each[k++]=x%;
            x/=;
        }
        for(int i=;i>=;i--) {
            int g,c=each[i];if(c==)g=;else g=;
            if(ch[u][g]) {
                u=ch[u][g];
            }
            else u=ch[u][c];
            if(i==) return sum[u];
        }
   }
}trie;
int main() {

    while(scanf("%d",&n)!=EOF) {
        int u,v,w;init();
        trie.init();
        for(int i=;i<n;i++) {
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w),add(v,u,w);
        }
        memset(a,,sizeof(a));
        dfs(,-);
        for(int i=;i<n;i++) {
            trie.insertt(a[i]);
        }
        int ans=;
        for(int i=;i<n;i++) {
            ans=max(ans,a[i]^trie.ask(a[i]));
        }
        cout<<ans<<endl;
    }
    return ;
}

POJ 3764