欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-06-04 12:46:58
...

叔叔传你过去

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

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

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

#include<bits/stdc++.h>
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;
}
相关标签: 字典树