1. 程式人生 > 實用技巧 >2020牛客暑期多校訓練營(第五場) Graph

2020牛客暑期多校訓練營(第五場) Graph

本題和cf 888G是一樣的

利用分治演算法

#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>

using namespace std;
const int maxn = 2e5+11;
typedef long long ll;

ll list[maxn];
ll tree[maxn*33][3];

int cnt=0;

ll add(ll x){//把數值插入字典樹
    int root = 0;
    for(int i=29;i>=0;i--){
        int t = (x>>i)&1;
        if(!tree[root][t]) tree[root][t] = ++cnt;
        root = tree[root][t];//向下爬
    }
    return 0;
}

ll find(ll x){//查詢字典樹裡面和x異或最小的數值
    int root = 0;
    ll an =0 ;
    for(int i= 29;i>=0;i--){
        ll t = (x>>i)&1;
        if(tree[root][t]) root = tree[root][t];
        else{
            root = tree[root][t^1];
            an |= (1<<i);
        }
    }
    return an;
}

ll ans = 0;

int dfs(int l,int r,int dep){
    if(dep==-1||l>=r) return 0;

    int mid = -100;
    for(int i = l;i<=r;i++){
        if((list[i]>>dep)&1) break;
        mid = i;
    }

    if(mid == -100 || mid == r){//全是1
        dfs(l,r,dep-1);
    }
    else{
        dfs(l,mid,dep-1);
        dfs(mid+1,r,dep-1);
    }

    if(mid==-100||mid==r) return 0;//dep位置全是0 或者全是1

    for(int i = l;i<=mid;i++){
        add(list[i]);
    }


    long long cns = 1e15;

    for(int i = mid+1;i<=r;i++){
        long long a = find(list[i]);
        cns = min(cns,a);
    }

    ans += 1LL*cns;
    
    for(int i=0;i<=cnt;i++){
        tree[i][0] = tree[i][1] = 0;
    }
    cnt = 0;
    return 0;
}
struct Node{
    int p;
    int len;
    Node(int aa,int bb):p(aa),len(bb){}
};

vector<Node>G[maxn];
void add(int x,int y,int len){
    G[x].push_back(Node(y,len));
}
int dfs2(int x,int fa,int dep){
    list[x] = dep;
    for(int i=0;i<G[x].size();i++){
        int p = G[x][i].p;
        int ln = G[x][i].len;
        if(p == fa) continue;
        dfs2(p,x,(dep^ln));
    }
    return 0;
}

int main(){
    ios::sync_with_stdio(false);
    int n;
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int x,y,len;
        scanf("%d%d%d",&x,&y,&len);
        x++;
        y++;
        add(x,y,len);
        add(y,x,len);
    }
    dfs2(1,-1,0);
    sort(list+1,list+1+n);
    ans = 0;
    dfs(1,n,29);
    printf("%lld\n",ans);
    return 0;
}