1、算法思路
Tire树是用来方便的存储和查找字符串(或者其他类型的数据)的数据结构。
一般来说题目会限定好所要求的数据范围,例如小写字母、大写字母等,如果实在过多,可以使用二进制编码进行存储,例如对于所有整数,可以将其二进制表示视为一个字符串进行存储。
举一个例子就可以明白Tire的含义,接下来我们展示用Tire存储以下字符串:
- abcbc
- abc
- afgrd
- ood
第一个字符串存储如下图:
标记上星星代表他是一个完整的字符串。我们继续添加:
这样就完成了Tire树的插入,查找也只需要从上到下进行检索,如果能找到带星星的目标,则证明之前存储过相关的字符串。
2、例题
维护一个字符串集合,支持两种操作:
I x 向集合中插入一个字符串 x;
Q x 询问一个字符串在集合中出现了多少次。
共有 N 个操作,输入的字符串总长度不超过 105,字符串仅包含小写英文字母。
输入格式
第一行包含整数 N,表示操作数。
接下来 N 行,每行包含一个操作指令,指令为 I x 或 Q x 中的一种。
输出格式
对于每个询问指令 Q x,都要输出一个整数作为结果,表示 x 在集合中出现的次数。
每个结果占一行。
数据范围
1≤N≤2∗10^4
输入样例:
5
I abc
Q abc
Q ab
I ab
Q ab
输出样例:
1
0
1
#include<iostream>
#include<cstring>
using namespace std;
const int N = 1e5+ 10;
//idx 跟单链表中的作用类似,记录下当前我们用到了哪个位置
//son中son[0]是根节点,查找都是从根节点的子节点开始
//cnt记录每个字符串的总共的数量
int son[N][26],cnt[N],idx;
void push(string str)
{
int p = 0;
for(int i = 0; str[i]; i ++)
{
//如果没有就增加一个节点
if(son[p][str[i] - 'a'] == 0)
{
son[p][str[i] - 'a'] = ++ idx;
p = idx;
}else{
p = son[p][str[i] - 'a'];
}
}
cnt[p] ++;
}
void query(string str)
{
int p = 0;
for(int i = 0; str[i]; i ++)
{
if(son[p][str[i] - 'a'] == 0)
{
cout << "0" << endl;
return;
}else{
p = son[p][str[i] - 'a'];
}
}
cout << cnt[p] << endl;
}
int main()
{
int n;
cin >> n;
while(n --)
{
char op[2];
string str;
cin >> op >> str;
if(op[0] == 'I')
{
push(str);
}else{
query(str);
}
}
}
输入格式
第一行输入一个整数 N。
第二行输入 N 个整数 A1~AN。
输出格式
输出一个整数表示答案。
数据范围
1≤N≤10^5,
0≤Ai<2^31
输入样例:
3
1 2 3
输出样例:
3
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010, M = 3100010;
int n;
int a[N], son[M][2], idx;
void insert(int x)
{
int p = 0;
for (int i = 30; i >= 0; i -- )
{
int &s = son[p][x >> i & 1];
if (!s) s = ++ idx;
p = s;
}
}
int search(int x)
{
int p = 0, res = 0;
for (int i = 30; i >= 0; i -- )
{
int s = x >> i & 1;
if (son[p][!s])
{
res += 1 << i;
p = son[p][!s];
}
else p = son[p][s];
}
return res;
}
int main()
{
scanf("%d", &n);
for (int i = 0; i < n; i ++ )
{
scanf("%d", &a[i]);
insert(a[i]);
}
int res = 0;
for (int i = 0; i < n; i ++ ) res = max(res, search(a[i]));
printf("%d\n", res);
return 0;
}
//题解来源说明
作者:yxc
链接:https://www.acwing.com/activity/content/code/content/45284/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
这个题就是将数字的二进制表示当成一个“字符串”进行处理。