HDU-4825(数组维护字典树,更节省内存的写法)

叔叔传你过去

指针动态分配内存:
HDU-4825(数组维护字典树,更节省内存的写法)_第1张图片

数组分配内存:
HDU-4825(数组维护字典树,更节省内存的写法)_第2张图片

所以还是用数组嘛,虽然有点抽象,但简单易操作,而且代码短

#include
typedef long long ll;
using namespace std;
#define maxn 2000005
#define son 2
int ch[maxn][2],sz,val[maxn];//节点编号,字典树节点个数,值

void init()
{
    sz=1;
    memset(ch,0,sizeof(ch));
    memset(val,0,sizeof(val));
}

void insert(int x)
{
    int u=0;
    bool a[32];
    int xx=x;

    for(int i=0;i<32;i++) a[i]=x&1,x>>=1;

    for(int i=31;i>=0;i--){
        if(!ch[u][a[i]]) ch[u][a[i]]=sz++;
        u=ch[u][a[i]];
    }
    val[u]=xx;
}

int find(int x)
{
    int u=0;
    bool a[32];
    int xx=x;

    for(int i=0;i<32;i++) a[i]=x&1,x>>=1;

    for(int i=31;i>=0;i--){
        if(ch[u][!a[i]]) u=ch[u][!a[i]];
        else u=ch[u][a[i]];
    }
    return val[u];
}
int main()
{
    int T,c=0;
    scanf("%d",&T);
    for(int c=1;c<=T;c++){
    init();
    int n,m,x;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&x);
        insert(x);
    }

    printf("Case #%d:\n",c);
    for(int i=1;i<=m;i++){
        scanf("%d",&x);
       printf("%d\n",find(x));
    }

  }
   return 0;
}

你可能感兴趣的:(字典树)