之前在做一道关于字符串匹配的题时,用到了字典树,但那时是用指针实现的,这次又遇到需要使用字典树这一结构的题,向学姐要了她的板子,学习了用数组实现的方法,对于解题而言,更为简短快速。
因为题目要求最大异或和,因此用的是01字典树,在字典树的基础上稍作修改。
以下为字典树和01字典树的普遍实现:
字典树
#include<iostream>
#include<algorithm>
using namespace std;
struct Trie
{
static const int N = 101010 , M = 26;
int node[N][M],cnt[N],root,L; //cnt记录对应节点的字符串个数
void init()
{
fill_n(cnt,N,0);
fill_n(node[N-1],M,0);
L = 0;
root = newnode();
}
int newnode()
{
fill_n(node[L],M,0);
return L++;
}
void add(char *s)
{
int p = root;
for(int i=0;s[i];++i)
{
int c = s[i] - ‘a‘;
if(!node[p][c])
node[p][c] = newnode();
p = node[p][c];
}
++cnt[p];
}
};
01字典树(可用于求异或和最大问题,long long型开64倍,int型开32倍)
#include<algorithm>
using namespace std;
struct Trie_01
{
static const int maxn=1e5+10,N = 32*maxn,M = 2;
int node[N][M],value[N],rt,L; //value记录对应节点的值,用于返回
void init()
{
fill_n(node[N-1],M,0);
fill_n(value,N,0);
L = 0;
rt = newnode();
}
int newnode()
{
fill_n(node[L],M,0);
return L++;
}
void add(int x)
{
int p = rt;
for (int i=31;i>=0;--i)
{
int idx = (x>>i)&1;
if (!node[p][idx])
{
node[p][idx] = newnode();
}
p = node[p][idx];
value[p]=min(value[p],x);
}
}
int query(int x)
{
int p = rt;
for (int i=31;i>=0;--i)
{
int idx = (x>>i)&1;
if (node[p][idx^1])
p = node[p][idx^1];
else
p = node[p][idx];
}
return value[p];
}
};
原文地址:https://www.cnblogs.com/orangee/p/9090287.html
时间: 2024-10-25 21:52:50