字典树

字典树详解:浅谈Trie树(字典树)
基本操作:插入,查找,删除
例题:查单词
就是找以某个字符串为前缀的单词的数量.
Sample Input
banana
band
bee
absolute
acm

ba
b
band
abc
Sample Output
2
3
1
0

字典树这个数据结构是用二维tree数组维护的,tree[i][j]表示编号为i的结点的第j个孩子 的编号,另外还有两个重要的一维数组val,sum
val储存的是权值信息,用结点的编号访问,可以用来查找单词是否存在
sum可以用来解决有相同前缀和问题,还有删除操作

树的构建(插入):
图中除标注外其余的结点在sum中对应的值都是1

在这里插入图片描述

int tree[manx][27],cou=1,sum[manx],root;
void iinsert()
{
    int len=strlen(str),root=0;//根节点的编号为0
    for(int i=0; i<len; i++)
    {
        int pos=str[i]-'a';
        if(!tree[root][pos])//如果之前没有从root到pos的边
            tree[root][pos]=cou++;
        root=tree[root][pos];
        sum[root]++;
    }
}

删除:

int del()
{
    root=0;
    int len=strlen(str),flag=1;
    for(int i=0; i<len; i++)
    {
        int pos=str[i]-'a';
        sum[root]--;
        root=tree[root][pos];
    }
    sum[root]--;
}

查找:

int ffind()
{
    root=0;
    int len=strlen(str),flag=1;
    for(int i=0; i<len; i++)
    {
        int pos=str[i]-'a';
        if(!tree[root][pos]||!sum[tree[root][pos]])//不存在,或已被删除完
        {
            flag=0;
            break;
        }
        root=tree[root][pos];
    }
    int ans=sum[root];//得到单词数量
    if(!flag)
        ans=0;
    return ans;
}

相关问题:求异或值最大(或最小)    ⟹    \implies 字典树+贪心

具体操作:先用给定的序列构造一棵01字典树,tree[manx][2],并把序列中的值存在叶子结点上(这时候val数组就要起作用了)(因为序列的值都是按32位,从最高位开始存,所以都存的32位,自然权值就都在叶子节点上了)。

然后开始贪心找,要异或结果最大,就要找与它每一位都相反的数,如果没有的话就只能找和它相同的了。

嗯,只能从最高位开始操作,和一个数的大小关系一样的道理。

例2:(求异或值最大)Xor Sum
例3:(求异或值最小+删除操作)Perfect Security

例3代码:

#include <iostream>
#include <string.h>
#include<stdio.h>
#include<math.h>
#include<algorithm>
#include<vector>
#include<queue>
typedef long long LL;
using namespace std;
const int manx=1e7+10;
const int INF=0x3f3f3f3f;
int cou,tree[manx][2],val[manx],ans,sum[manx];
void iinsert(int x)
{
     int root=0;
     for(int i=29;i>=0;i--)
     {
          int net=(x>>i)&1;
          if(!tree[root][net])tree[root][net]=++cou;
          root=tree[root][net];
          sum[root]++;
     }
     val[root]=x;
}
void ffind(int x)
{
     int root=0;
     for(int i=29;i>=0;i--)
     {
          int net=(x>>i)&1;
          if(tree[root][net]&&sum[tree[root][net]])
          //如果这一位上存在与它相反的数字且未被全部删除
               root=tree[root][net];
          else
               root=tree[root][net^1];
          sum[root]--;
     }
     ans=val[root];
}
int main()
{
     int n,x,a[300010];
     while(scanf("%d",&n)!=EOF)
     {
          cou=0;
          memset(tree,0,sizeof(tree));
          memset(sum,0,sizeof(sum));
          for(int i=1;i<=n;i++)
               scanf("%d",&a[i]);
          for(int i=1;i<=n;i++)
          {
               scanf("%d",&x);
               iinsert(x);
          }
          for(int i=1;i<n;i++)
          {
               ffind(a[i]);
               printf("%d ",a[i]^ans);
          }
          ffind(a[n]);
          printf("%d\n",a[n]^ans);
     }
}

最开始写的递归(现在感觉好麻烦,还是用sum数组操作好hhh),想在在回溯的时候删除路径(将tree[i][j]置为0),但是没想到会有相同数字的情况,导致删除一个数的时候,就把和它相同的数都删除了,一直卡在第8组数据,后面加了一个数组村下个每个数出现的次数,就过了

#include <iostream>
#include <string.h>
#include<stdio.h>
#include<math.h>
#include<algorithm>
#include<vector>
#include<queue>
typedef long long LL;
using namespace std;
const int manx=1e7+10;
const int INF=0x3f3f3f3f;
int cou,tree[manx][2],val[manx],vs[manx],ans,sum[manx];
void iinsert(int x)
{
    int root=0;
    for(int i=29; i>=0; i--)
    {
        vs[root]=0;
        int net=(x>>i)&1;
        if(!tree[root][net])
            tree[root][net]=++cou;
        root=tree[root][net];
    }
    val[root]=x;
    sum[root]++;
}
void ffind(int x,int root,int pos,int proot,int pnet)
{
    if(pos==-1)
    {
        sum[root]--;
        if(sum[root]==0)
            tree[proot][pnet]=0;
        ans=val[root];
        return;
    }
    int net=(x>>pos)&1;
    if(tree[root][net])
        ffind(x,tree[root][net],pos-1,root,net);
    else
        ffind(x,tree[root][net^1],pos-1,root,net^1);
    if(!tree[root][0]&&!tree[root][1])
        tree[proot][pnet]=0;
}
int main()
{
    int n,x,a[300010];
    while(scanf("%d",&n)!=EOF)
    {
        memset(tree,0,sizeof(tree));
        memset(vs,1,sizeof(vs));
        memset(sum,0,sizeof(sum));
        for(int i=1; i<=n; i++)
            scanf("%d",&a[i]);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&x);
            iinsert(x);
        }
        for(int i=1; i<n; i++)
        {
            ffind(a[i],0,29,0,0);
            printf("%d ",a[i]^ans);
        }
        ffind(a[n],0,29,0,0);
        printf("%d\n",a[n]^ans);
    }
}
/*
5
1 2 3 4 1
1 1 2 3 4
*/

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值