cf888G. Xor-MST(Boruvka最小生成樹 Trie樹)

自為風月馬前卒發表於2018-10-14

題意

題目連結

給出(n)點，每個點有一個點權(a[i])，相鄰兩點之間的邊權為(a[i] oplus a[j])，求最小生成樹的值

Sol

非常interesting的一道題，我做過兩種這類題目，一種是直接打表找規律，另一種就像這種用Boruvka演算法加一些騷操作來搞。

首先，把所有元素扔到Trie樹裡面，這樣對於Trie樹上的每一層(對應元素中的每一位)共有兩種情況:

  1. 全為0或全為1

  2. 一部分為0另一部分為1

對於第一種情況，我們無需考慮，因為任意點相鄰產生的貢獻都是0，對於第二種情況，需要找到一條最小的邊來連線鏈各個集合，這可以在Trie樹上貪心實現

另外還有一個小Trick，我們把元素從小到大排序，這樣Trie樹上每個節點對應的區間就都是連續的

實現的時候可以從底往上update，也可以從上往下dfs

時間複雜度：(O(nlognlog_{max(a_i)}))

本來以為這題要寫一年，結果寫+調只用了1h不到？

#include<bits/stdc++.h>
#define Pair pair<int, int>
#define fi first
#define se second
#define MP(x, y) make_pair(x, y)
#define LL long long
using namespace std;
const int MAXN = 2e5 + 10, B = 31, INF = 1e9 + 7;
inline LL read() {
    char c = getchar(); LL x = 0, f = 1;
    while(c < `0` || c > `9`) {if(c == `-`) f = -1; c = getchar();}
    while(c >= `0` && c <= `9`) x = x * 10 + c - `0`, c = getchar();
    return x * f;
}
int N, a[MAXN], L[MAXN * 30], R[MAXN * 30], ch[MAXN * 30][2], tot = 0;
void insert(int val, int pos) {
    int now = 0;
    for(int i = B; ~i; i--) {
        int x = val >> i & 1;
        if(!ch[now][x]) ch[now][x] = ++tot, L[tot] = pos, R[tot] = pos;//???ʵ???֮?????λ 
        L[now] = min(L[now], pos); R[now] = max(R[now], pos);
        now = ch[now][x];
    }
}
int Query(int val, int now, int NowBit) {
    LL ans = 1 << (NowBit + 1);
    for(int i = NowBit; ~i; i--) {
        int x = val >> i & 1;
        if(ch[now][x]) now = ch[now][x];
        else ans |= 1 << i, now = ch[now][x ^ 1];
    }
    return ans;
}
LL dfs(int x, int NowBit) {
    LL res = 0;
    if(NowBit == 0)
        return (ch[x][0] && ch[x][1]) ? (a[L[ch[x][0]]] ^ a[R[ch[x][1]]]) : 0;
    //if(NowBit == 0) return ;
    if(ch[x][0] && ch[x][1]) {
        int tmp = INF;
        for(int i = L[ch[x][0]]; i <= R[ch[x][0]]; i++) tmp = min(tmp, Query(a[i], ch[x][1], NowBit - 1));
        res += tmp;
    }
    if(ch[x][0]) res += dfs(ch[x][0], NowBit - 1);
    if(ch[x][1]) res += dfs(ch[x][1], NowBit - 1);
    return res;
}
int main() {
    N = read();
    for(int i = 1; i <= N; i++) a[i] = read();
    sort(a + 1, a + N + 1);
    L[0] = INF; R[0] = 0;
    for(int i = 1; i <= N; i++) insert(a[i], i);
    cout << dfs(0, B);
    return 0;
}

相關文章