首页 > 代码库 > poj - 3764 - The xor-longest Path(Trie)
poj - 3764 - The xor-longest Path(Trie)
题意:一棵 n 个结点的树,树边有权值w(0 <= w < 2^31),求最两结点间的最大异或。
题目链接:http://poj.org/problem?id=3764
——>>取0为根,预处理出所有结点到根的异或xOr[i]。那么结点 a 与结点 b 之间的路径异或就是xOr[a] ^ xOr[b]。。
权值 w 最多31位,于是,将每个xOr的二进制表示从高位到低位插入到 01 Trie中(0为0,非0为1)。。
查询时从高位开始贪心。。
很好的题目,第一次不在字母上使用Trie。。
#include <cstdio> #include <cstring> #include <algorithm> using std::max; const int MAXN = 100000 + 10; const int MAX_NODE = 3000000 + 10; const int MAX_X = 2; const int MAX_BIT = 30; struct EDGE { int to; int w; int nxt; } edge[MAXN << 1]; int n; int hed[MAXN], ecnt; int xOr[MAXN]; int ch[MAX_NODE][MAX_X], sz; void Init() { ecnt = 0; memset(hed, -1, sizeof(hed)); xOr[0] = 0; sz = 1; memset(ch[0], 0, sizeof(ch[0])); } void AddEdge(int u, int v, int w) { edge[ecnt].to = v; edge[ecnt].w = w; edge[ecnt].nxt = hed[u]; hed[u] = ecnt++; } void Read() { int u, v, w; for (int i = 0; i < n - 1; ++i) { scanf("%d%d%d", &u, &v, &w); AddEdge(u, v, w); AddEdge(v, u, w); } } void Dfs(int u, int fa) { for (int e = hed[u]; e != -1; e = edge[e].nxt) { int v = edge[e].to; if (v != fa) { xOr[v] = xOr[u] ^ edge[e].w; Dfs(v, u); } } } void Insert(int x) { int u = 0; for (int i = MAX_BIT; i >= 0; --i) { int cur = ((1 << i) & x) > 0 ? 1 : 0; if (!ch[u][cur]) { memset(ch[sz], 0, sizeof(ch[sz])); ch[u][cur] = sz++; } u = ch[u][cur]; } } void Insert() { for (int i = 0; i < n; ++i) { Insert(xOr[i]); } } int Query(int x) { int ret = 0, u = 0; for (int i = MAX_BIT; i >= 0; --i) { int cur = ((1 << i) & x) > 0 ? 1 : 0; if (ch[u][!cur]) { ret |= (1 << i); u = ch[u][!cur]; } else { u = ch[u][cur]; } } return ret; } void Query() { int ret = 0; for (int i = 0; i < n; ++i) { ret = max(ret, Query(xOr[i])); } printf("%d\n", ret); } int main() { while (scanf("%d", &n) == 1) { Init(); Read(); Dfs(0, -1); Insert(); Query(); } return 0; }
poj - 3764 - The xor-longest Path(Trie)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。