字典树的两道经典例题:最大异或对 最长异或路径

P10471 最大异或对 The XOR Largest Pair - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

模板题,从最高位开始建立01字典树,查询时也从高位开始查,贪心思想遇到不一样的就走,没有就走一样的,就能保证走完的路径是这个数能组成的最大异或和。注意trie的空间范围:

#include<bits/stdc++.h>
using namespace std;
const int N=1e7+10;
int n,trie[N][2],a[100001],tot,ans,anss;
void insert(int x)
{
	bitset<32> s;
	s=x;
	int p=0;
	for(int i=30;i>=0;i--)
	{
		int ch=s[i];
		if(!trie[p][ch]) trie[p][ch]=++tot;
		p=trie[p][ch];
	}
}
int search(int x)
{
	bitset<32> s;
	s=x;
	int p=0;
	ans=0;
	for(int i=30;i>=0;i--)
	{
		int ch=s[i];
		if(trie[p][1-ch]) 
		{
			p=trie[p][1-ch];
			ans+=(1<<i);
		}
		else p=trie[p][ch];
	}
	return ans;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		insert(a[i]);
	}
	for(int i=1;i<=n;i++)
	{ 
		anss=max(anss,search(a[i]));
	}
	cout<<anss;
}

P4551 最长异或路径 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

根据异或的基本性质不难发现,x到y的异或和就是x到根节点的异或和异或y到根节点的异或和,求出所有节点的异或和,从高位插入字典树中,再从高位查询,和上道题一样,遇到不一样的就走。

#include<bits/stdc++.h>
using namespace std;
const int N=100001,M=1e7+10;
int n,d[N],trie[M][2],anss;
int ver[N*2],Next[N*2],edge[N*2],head[N],tot;
void add(int x,int y,int z)
{
	ver[++tot]=y,edge[tot]=z;
	Next[tot]=head[x],head[x]=tot;
}
void dfs(int x,int fa)
{
	for(int i=head[x];i;i=Next[i])
	{
		int y=ver[i],z=edge[i];
		if(y==fa) continue;
		d[y]=d[x]^z;
		dfs(y,x);
	}
}
void insert(int x)
{
	int p=0;
	for(int i=30;i>=0;i--)
	{
		int ch=(x>>i)&1;
		if(!trie[p][ch]) trie[p][ch]=++tot;
		p=trie[p][ch];
	}
}
int search(int x)
{
	int p=0,ans=0;
	for(int i=30;i>=0;i--)
	{
		int ch=(x>>i)&1;
		if(trie[p][1-ch])
		{
			ans+=(1<<i);
			p=trie[p][1-ch];
		}
		else p=trie[p][ch];
	}
	return ans;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z),add(y,x,z);
	}
	dfs(1,-1);
	for(int i=1;i<=n;i++)
	{
		insert(d[i]);
		anss=max(anss,search(d[i]));
	}
	printf("%d",anss);
}
求1~n数组的最大异或和也可以使用类似于求一段区间最大异或和的方法。具体步骤如下: 1. 将1~n数组中的所有数以二进制形式插入到字典树中。 2. 对于每个数,从高位到低位依次匹配字典树上的节点,如果当前位为1,就往字典树的右子树走,否则就往左子树走。匹配完整个二进制数后,我们可以得到一个最大异或值。 3. 对于1~n数组,我们可以将其中的相邻两个数看作一段区间,然后使用类似于求一段区间最大异或和的方法求出最大异或和。 时间复杂度为O(nlogC),其中n为数组长度,C为数的范围。以下是求1~n数组的最大异或和的C++代码: ```c++ #include <iostream> using namespace std; const int MAXN = 100010; const int MAXBITS = 30; struct TrieNode { int cnt; int children[2]; } trie[MAXN * MAXBITS]; int root, node_cnt; void insert(int x) { int p = root; for (int i = MAXBITS - 1; i >= 0; i--) { int idx = (x >> i) & 1; if (!trie[p].children[idx]) { trie[p].children[idx] = ++node_cnt; } p = trie[p].children[idx]; trie[p].cnt++; } } int query(int x) { int p = root, res = 0; for (int i = MAXBITS - 1; i >= 0; i--) { int idx = (x >> i) & 1; if (trie[trie[p].children[idx ^ 1]].cnt > 0) { res += (1 << i); p = trie[p].children[idx ^ 1]; } else { p = trie[p].children[idx]; } } return res; } int main() { int n; cin >> n; root = 1; node_cnt = 1; int pre = 0, ans = 0; for (int i = 1; i <= n; i++) { int x; cin >> x; insert(pre); pre ^= x; ans = max(ans, query(pre)); } cout << ans << endl; return 0; } ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值