uoj176 新年的繁荣

链接:http://uoj.ac/problem/176
对于这种边权难以直接维护的都直接考虑brouvka算法。
显然,我们要做的是实现一个可以查询&x最大的数据结构。
可以先对于所有权值建立一颗01-trie树。
考虑在trie树查询答案的过程,可以考虑一个从高位到低位的贪心。
当x的第i位为1时,最优策略一定是能走1就走1。
当x的第i位为0时,既可以走0也可以走1。
因此可以用一个类似线段树合并的方式,把每一个trie树上的节点右儿子合并到左儿子上。
然后直接按照brouvka算法的套路搞一下就行了。

#include<iostream>
#include<cctype>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<ctime>
#include<set>
#include<cstdlib>
#include<algorithm>
#define N 220000
#define L 200000
#define M 8000000
#define eps 1e-7
#define inf 1e9+70
#define db double
#define ll long long
#define ldb long double
using namespace std;
inline int read()
{
	char ch=0;
	int x=0,flag=1;
	while(!isdigit(ch)){ch=getchar();if(ch=='-')flag=-1;}
	while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
	return x*flag;
}
int v[N],f[N];
int find(int x){if(x!=f[x])f[x]=find(f[x]);return f[x];}
void merge(int x,int y){f[find(x)]=find(y);}
struct node{int x,y;}w[N],p[N];
bool operator<(node a,node b)
{
	if(a.y!=b.y)return a.y<b.y;
	else return a.x<b.x;
}
int n,m,a[N];
struct Trie
{
	#define lson lc[o]
	#define rson rc[o]
	int size,root,lc[M],rc[M],maxv[M],mixv[M];
	int newnode()
	{
		int o=++size;
		lson=rson=0;maxv[o]=mixv[o]=-inf;
		return o;
	}
	void upd(int x,int y)
	{
		if(!x||!y)return;
		if(maxv[x]==maxv[y])mixv[x]=max(mixv[x],mixv[y]);
		if(maxv[x]<maxv[y])mixv[x]=max(mixv[x],max(maxv[x],mixv[y]));
		if(maxv[x]>maxv[y])mixv[x]=max(mixv[x],maxv[y]);
		maxv[x]=max(maxv[x],maxv[y]);
	}
	void pushup(int o)
	{
		maxv[o]=mixv[o]=-inf;
		upd(o,lson);upd(o,rson);
	}
	void insert(int &o,int t,int x,int id)
	{
		if(!o)o=newnode();
		if(!t)
		{
			if(maxv[o]>id)mixv[o]=max(mixv[o],id);
			if(maxv[o]<id)mixv[o]=maxv[o],maxv[o]=id;
			return;
		}
		if(!((1<<(t-1))&x))insert(lson,t-1,x,id);
		else insert(rson,t-1,x,id);
		pushup(o);
	}
	int merge(int x,int y)
	{
		if(!x||!y)return x|y;
		int o=newnode();
		lson=merge(lc[x],lc[y]);
		rson=merge(rc[x],rc[y]);
		maxv[o]=mixv[o]=-inf;
		upd(o,x);upd(o,y);
		return o;
	}
	void solve(int o,int t)
	{
		if(!t||!o)return;
		if(lson)solve(lson,t-1);
		if(rson)solve(rson,t-1);
		lson=merge(lson,rson);
	}
	node query(int o,int t,int x,int id)
	{
		if(!t)return (node){(id==maxv[o])?mixv[o]:maxv[o],0};
		if((1<<(t-1))&x)
		{
			if((!rson)||(id==maxv[rson]&&mixv[rson]==-inf))
			return query(lson,t-1,x,id);
			else
			{
				node ans=query(rson,t-1,x,id);
				ans.y+=(1<<(t-1));
				return ans;
			}
		}
		else
		return query(lson,t-1,x,id);
	}
}T;
int main()
{
	n=read(),m=read();
	for(int i=1;i<=n;i++)a[i]=read();
	for(int i=1;i<=n;i++)f[i]=i;
	ll ans=0;
	for(int i=0;i<M-1;i++)T.lc[i]=T.rc[i]=0,T.maxv[i]=T.mixv[i]=-inf;
	int tot=0;
	while(true)
	{
		tot++;
		int cnt=0;
		for(int i=1;i<=n;i++)v[i]=0,f[i]=find(i);
		for(int i=1;i<=n;i++)
		{
			v[f[i]]=max(v[f[i]],i);
			if(i==f[i])cnt++;
		}
		for(int i=1;i<=n;i++)f[i]=v[f[i]];
		if(cnt==1)break;
		for(int i=0;i<=T.size;i++)T.lc[i]=T.rc[i]=0,T.maxv[i]=T.mixv[i]=-inf;
		T.size=T.root=0;
		for(int i=1;i<=n;i++)T.insert(T.root,m,a[i],find(i));
		T.solve(T.root,m);
		
		for(int i=1;i<=n;i++)w[i]=(node){-1,-1};
		
		for(int x=1;x<=n;x++)
		{
			int id=find(x);
			node o=T.query(T.root,m,a[x],id);
			w[id]=max(w[id],o);
		}
		
		int tot=0;
		for(int i=1;i<=n;i++)
		if(w[i].x!=-1)
		{
			int x=w[i].x;
			if(find(i)!=find(w[x].x)||find(i)<find(w[i].x))
			{
				ans+=w[i].y;
				p[++tot]=(node){find(i),find(w[i].x)};
			}
		}
		for(int i=1;i<=tot;i++)merge(p[i].x,p[i].y);
	}
	printf("%lld",ans);
	return 0;
}
posted @ 2019-01-18 00:05  Creed-qwq  阅读(400)  评论(0编辑  收藏  举报