字典树详解:浅谈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
*/