CF888G Xor-MST

题目链接

problem

给出n个点,每个点有权值,求最小生成树。定义一条边的代价为所连接两点的权值异或值。

solution

考虑分治,根据最高位为0还是1分为两部分。然后分别求最小生成树。合并的时候就将最高位为0的一部分插入到trie中,然后从最高位为1的一部分中查询。

注意对trie的清空。

code

/*
* @Author: wxyww
* @Date:   2019-08-18 19:36:04
* @Last Modified time: 2019-08-18 20:20:03
*/
#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#include<ctime>
using namespace std;
typedef long long ll;
const int N = 200100,INF = 1e9;
ll read() {
	ll x=0,f=1;char c=getchar();
	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 tree[N * 31][2],a[N];
int n,tot;

int get(int x) {
	for(int i = 29;i >= 0;--i) printf("%d",x >> i & 1);
	puts("");
}
void add(int x) {
	// get(x);
	int p = 0;
	for(int i = 29;i >= 0;--i) {
		int k = x >> i & 1;
		if(!tree[p][k]) tree[p][k] = ++tot;
		p = tree[p][k];
	}

}

int query(int x) {
	// get(x);
	int p = 0,ret = 0;
	for(int i = 29;i >= 0;--i) {
		int k = x >> i & 1;
		if(!tree[p][1] && !tree[p][0]) return INF;
		if(!tree[p][k]) {
			ret += 1 << i;
			p = tree[p][k ^ 1];
		}
		else p = tree[p][k];
		
	}
	// printf("%d\n",ret);
	return ret;
}
ll solve(int l,int r,int pos) {
	if(l > r) return 0;
	// printf("%d %d %d\n",l,r,pos);
	if(l == r || pos == -1) {
		add(a[l]);
		return 0;
	}

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


	ll ret = 0;
	ret += solve(l,mid,pos - 1);

	int mn = INF;

	for(int i = mid + 1;i <= r;++i) mn = min(mn,query(a[i]));

	if(mn != INF) ret += mn;

	for(int i = 0;i <= tot;++i) tree[i][0] = tree[i][1] = 0;
	tot = 0;

	ret += solve(mid + 1,r,pos - 1);

	// for(int i = mid + 1;i <= r;++i) add(a[i]);
	for(int i = l;i <= r;++i) add(a[i]);
	// printf("%d\n",ret);
	// printf("%d %d %d %d\n",l,r,pos,ret);
	return ret;

}
int main() {
	// freopen("1.in","r",stdin);
	n = read();
	for(int i = 1;i <= n;++i) a[i] = read();

	sort(a + 1,a + n + 1);

	// for(int i = 1;i <= n;++i) get(a[i]);

 	// puts("");

	cout<<solve(1,n,29);

	return 0;
}
posted @ 2019-08-18 20:23  wxyww  阅读(...)  评论(...编辑  收藏