Trie树:是一种能够高效存储和查找字符串集合的数据结构
Trie字符串统计
思路:
(笔记来自AcWing 835. Trie字符串统计 - AcWing)
代码如下:
#include<iostream>
#include<cstdio>
#include<string>
#include<algorithm>
using namespace std;
const int N = 1e5 + 10;int son[N][26]; //trie树每个点的所有儿子(最多26个) //[N]父节点 []子节点
int cnt[N];//以当前点结尾的单词有多少个
int idx;//存的是当前已经用到了哪个下标(未用)
//idx=0 既是根节点 又是空结点(一个点没有子节点 会让它指向0)//1.insert函数: trie的插入:
void insert(string str)
{int p = 0; //根节点为0for (int i = 0; i < str.size(); i ++) //char字符串以'\0'结尾{int u = str[i] - 'a'; //将字符串的小写字母转换为0~25if (!son[p][u]) //如果p下面没有路径(没有子节点'u')找到u 则创建对应路径son[p][u] = ++ idx ;p = son[p][u]; //下次下移到这个结点开始寻找}cnt[p] ++; //以p节点结尾的 字符串有多少
}//query函数: 查询指定字符串出现次数
int query(string str)
{int p = 0;for (int i = 0; i < str.size(); i ++){int u = str[i] - 'a';if (!son[p][u]) //如果没有return 0;p = son[p][u];//下次下移到这个结点开始寻找}//指定字符串 循环结束return cnt[p]; //返回这个字符串的数量}int main()
{int n;cin >> n;while (n --){char ch;string s;cin >> ch >> s;if (ch == 'I')insert(s);else cout << query(s) << endl;}return 0;
}
最大异或对
活动 - AcWing
思路:
1.用Trie树存储每个数的二进制数形式
2.遍历这课树,每一位寻找与当前位u不同的数
3.若找到则向高位移动再加1(res*2 + 1),若找不到则向高位移动加0(res*2 + 0)
代码如下:
#include<iostream>
#include<algorithm>
using namespace std;
int const N=100010,M=31*N;int n;
int a[N];
int son[M][2],idx;
//M代表一个数字串二进制可以到多长void insert(int x)
{int p=0; //根节点for(int i=30;i>=0;i--){int u=x>>i&1; /取X的第i位的二进制数是什么 x>>k&1(前面的模板)if(!son[p][u]) son[p][u]=++idx; ///如果插入中发现没有该子节点,开出这条路p=son[p][u]; //指针指向下一层}
}
int search(int x)
{int p=0;int res=0;for(int i=30;i>=0;i--){ ///从最大位开始找int u=x>>i&1;if(son[p][!u]) 如果当前层有对应的不相同的数{ ///p指针就指到不同数的地址p=son[p][!u];res=res*2+1;///*2相当左移一位 然后如果找到对应位上不同的数res+1 例如 001} /// 010 else --->011 //刚开始找0的时候是一样的所以+0 到了0和1的时候原来0右移一位,判断当前位是同还是异,同+0,异+1{p=son[p][u];res=res*2+0;}}return res;
}
int main(void)
{cin.tie(0);cin>>n;idx=0;for(int i=0;i<n;i++){cin>>a[i];insert(a[i]);}int res=0;for(int i=0;i<n;i++){ res=max(res,search(a[i])); ///search(a[i])查找的是a[i]值的最大与或值}cout<<res<<endl;
}