2020牛客多校第五场 Graph(异或最小生成树)

链接:https://ac.nowcoder.com/acm/contest/5670/B
来源:牛客网

题目描述
Mr. W got a new graph with N vertices and N - 1 edges. It’s a connected graph without cycles. Each edge should have an ugly value. To make the graph more beautiful, Mr. W hope you can help him modify it. You can delete or add one edge with an ugly value at a time and you can do it as many times as you want. But the following conditions should be satisfied at any time:

  1. The graph is connected.
  2. For each cycles in the graph, the XOR sum of all ugly values in the cycle is 0.
    Mr. W wants to know the minimum sum of ugly values of all edges in the graph.
    输入描述:
    The first line contains one integerN(2 \leq N \leq 100000)N(2≤N≤100000).
    Next N - 1 lines each contains three integers U, V, W (0 \leq U,V < N, 0 \leq W < 2^{30})U,V,W(0≤U,V 30
    ), denoting an edge between vertex U and V with ugly value W.
    输出描述:
    One integer, the minimum sum of ugly values of all edges in the graph.
    示例1
    输入
    复制
    6
    0 1 1
    1 2 4
    1 3 3
    0 4 5
    0 5 2
    输出
    复制
    7

题意:
给一张树,可以加边或者删边,但要满足环上边的权值异或为0,且图连通。求最后得到的最小边权和。

思路:
就是CF888G的变形,改一下就能过了。

模拟一下可以发现,不管怎么加边和删边,两个点的的路径异或值不变,也就是两点加边的值不变。只需要记录根节点到每个节点的路径异或值,就可以得到任意两点连边对应权值。

于是题目就变形成了每个点有一个权值,两点连边的边权为权值异或,求这个完全图的最小生成树。

而对于异或最小生成树,
这是一张完全图,当然是不能用kruskal或者prim。
但是最小生成树除了这两个算法,还有一个叫做boruvka的算法,大致思路是维护每个连通块连出的最短边,然后每个连通块择优,选上最短边。

用在本题上,我们可以建立01字典树,那么每次对于连出的0出边和1出边,各有一个连通块,要连通这两个连通块,则需要从左子树(右子树)中枚举一个点,然后再利用01字典树的性质,得到在另一半对应的最小值。

因为对于两个点异或值,肯定是高位相等的越多,值越小,所以从01字典树最高位往下走是符合boruvka算法的

对于每个分出两个子树的点都这样处理,这类似分治的过程,复杂度肯定是够的。

#include
#include
#include
#include 
#include 
#include 

using namespace std;

typedef long long ll;

const int INF = 0x3f3f3f3f;
const int maxn = 2e5 + 7;
int L[maxn * 40],R[maxn * 40];
int ch[maxn * 40][2],tot;
int n,root;
int a[maxn];

void Insert(int &now,int x,int dep) {
    if(!now) now = ++tot;
    L[now] = min(L[now],x);R[now] = max(R[now],x);
    if(dep < 0) return;
    int bit = a[x] >> dep & 1;
    Insert(ch[now][bit],x,dep - 1);
}

ll query(int now,int val,int dep) {
    if(dep < 0) return 0;
    int bit = (val >> dep) & 1;
    if(ch[now][bit]) return query(ch[now][bit],val,dep - 1);
    return query(ch[now][bit ^ 1],val,dep - 1) + (1 << dep);
}

ll dfs(int now,int dep) {
    if(dep < 0) return 0;
    if(R[ch[now][0]] && R[ch[now][1]]) {
        ll mi = INF;
        for(int i = L[ch[now][0]];i <= R[ch[now][0]];i++) {
            mi = min(mi,query(ch[now][1],a[i],dep - 1));
        }
        return dfs(ch[now][0],dep - 1) + mi + dfs(ch[now][1],dep - 1) + (1 << dep);
    }
    if(R[ch[now][0]]) return dfs(ch[now][0],dep - 1);
    if(R[ch[now][1]]) return dfs(ch[now][1],dep - 1);
    return 0;
}

void solve() {
    sort(a + 1,a + 1 + n);
    memset(L,0x3f,sizeof(L));
    for(int i = 1;i <= n;i++) Insert(root, i, 30);
    printf("%lld\n",dfs(root,30));
}

int head[maxn],nex[maxn],to[maxn],val[maxn],TOT;

void add(int x,int y,int z) {
    to[++TOT] = y;
    nex[TOT] = head[x];
    val[TOT] = z;
    head[x] = TOT;
}

void DFS(int x,int fa,ll sum) {
    a[x] = sum;
    for(int i = head[x];i;i = nex[i]) {
        int v = to[i],w = val[i];
        if(v == fa) continue;
        DFS(v,x,sum ^ w);
    }
}

int main()
{
    scanf("%d",&n);
    for(int i = 1;i < n;i++) {
        int x,y,z;scanf("%d%d%d",&x,&y,&z);
        x++;y++;
        add(x,y,z);add(y,x,z);
    }
    DFS(1,-1,0);
    solve();
    return 0;
}

你可能感兴趣的:(#,字典树,#,codeforces,#,生成树)