【每日一题】9月17日题目精讲(01字典树)

The XOR-longest Path

https://ac.nowcoder.com/acm/problem/50349

利用异或的性质转化,再用字典树维护。首先我们知道树上两点必定有且只有一条简单路径,并且他们的关系有两种情况
1.他们具有祖孙关系,对于这种情况,我们记f[i]表示根节点到i的异或路径,那么f[i]xor f[j]即为i,j的异或路径
2.他们不具有祖孙关系,那么我们假如已知他们的LCA,根据第一种情况,所以异或路径值为(f[i]⨁f[lca])⨁(f[lca]⨁[j]),而异或操作满***换律和结合律,因此即为f[i]⨁f[j]。我们已经得出了树上两点间的异或路径的求法。只有处理出f数组,题目就相当于从f数组中选出两个数使它们的异或值最大,用字典树即可。

#include<iostream>
#include<string>
#include<math.h>
#include<algorithm>
#include<vector>
#include<queue>
#include<bits/stdc++.h>
typedef long long ll;
#define INF 0x3f3f3f3f
ll gcd(ll a, ll b)
{
    return b ? gcd(b, a % b) : a;
}
ll lcm(ll a, ll b) {
    return a * b / (gcd(a, b));
}
#define PII pair<int,int>
using namespace std;
const int maxn = 2e6 + 10, mod = 1e9 + 7;
int qmi(int a, int k, int p)        //快速幂模板
{
    int res = 1;
    while (k)
    {
        if (k & 1) res = (ll)res * a % p;
        k >>= 1;
        a = (ll)a * a % p;
    }
    return res;
}
///////////////////////////////////////////////////////////
struct Edge
{
    int v, w, next;
}edge[maxn];
int head[maxn],t;
int ch[maxn][3], f[maxn];
inline void Add_edge(int u, int v, int w)
{
    edge[++t].next = head[u];
    head[u] = t;
    edge[t].v = v;
    edge[t].w = w;
}
void dfs(int u, int fa) {
    for (int i = head[u]; i; i = edge[i].next) {
        int v = edge[i].v;
        if (v == fa) {
            continue;
        }
        f[v] = f[u] ^ edge[i].w;//f数组存储着异或
        dfs(v, u);//接着遍历子节点
    }
}
void insert(int x)//01字典树插入模板
{
    int u=1;
    for(int i=1<<30;i;i>>=1)
    {
        int num=(x&i)?1:0;
        if(!ch[u][num])ch[u][num]=++t;
        u=ch[u][num];
    }
}
int find(int x)//01字典树查询模板
{
    int u=1,ans=0;
    for(int i=1<<30;i;i>>=1)
    {
        int num=(x&i)?0:1;
        if(ch[u][num])
        {
            u=ch[u][num];
            ans+=i;
        }
        else u=ch[u][!num];
    }
    return ans;
}
int main() {
    int n;
    cin >> n;
    memset(f, 0, sizeof(f));
    for (int i = 1; i <= n - 1; i++) {
        int u, v, w;
        cin >> u >> v >> w;
        Add_edge(u, v, w);
        Add_edge(v, u, w);
    }
    dfs(1, 0);
    t = 1;
    for (int i = 1; i <= n; i++) {
        insert(f[i]);
    }
    int ans = 0;
    for (int i = 1; i <= n; i++) {
        ans = max(ans, find(f[i]));
    }
    cout << ans << endl;
    return 0;
}
/*
4
1 2 3
2 3 4
2 4 6
*/
全部评论

相关推荐

10-11 17:45
门头沟学院 Java
走吗:别怕 我以前也是这么认为 虽然一面就挂 但是颇有收获!
点赞 评论 收藏
分享
如题,字节跳动怎么才能看到自己的面评,找hr说看不到
SoulStar:自己应该看不到,这个是字节比较保密的信息,之前有mentor加我,说他能看到,但是不能给我说,给我说了他可能就要被辞退了
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务