mst

https://www.zybuluo.com/ysner/note/1245941

题面

给一个(n)点完全图,点权均小于(2^m)。定义边权等于两端点点权的与和(即(a_i&b_i))。求最大生成树边权和。

  • (10pts nleq8000)
  • (15pts m=1)
  • (25pts mleq12)
  • (55pts mleq15)
  • (90pts nleq10^5,mleq18)
  • (100pts nleq5*10^6,mleq20)

解析

(10pts)算法

(Kruskal)搞复杂度为(O(n^2logn)),会(TLE)
可以用(O(n^2))(Prim)算法。

for(rg int i=1;i<=n;i++) Dis[i]=-Inf;
			Dis[1]=0;   ll Ans=0;
			for(rg int i=1;i<=n;i++)
			{   int Now=0,Maxd=-Inf;
				for(rg int j=1;j<=n;j++)
					if(!Blue[j]&&Dis[j]>Maxd) Now=j,Maxd=Dis[j];
				Ans+=Maxd,Blue[Now]=1;
				for(rg int j=1;j<=n;j++)
					if(!Blue[j])  Dis[j]=Max(Dis[j],A[Now]&A[j]);
			}

值得注意的是初始值极小,永远在取(max)

(15pts)算法

说明点权只能为(0)(1)
(x)(1)点最多能连(x-1)条边。
(ans=x-1)

(25pts)算法

注意到如果数值(p)重复出现(x)次,答案加上((x-1)*p)(相同数值相互连边)后,就可把该数值视为唯一。
于是(n)降到(O(2^{12})=4096),又可以(Prim)啦。

(55pts)算法

注意到若(a&b=b),在((a,b))间连边一定最优。于是可以找出每个点(b)的子集(复杂度(O(15n))),若存在,优先建边并统计答案。

然后由于剩下点互不包含,点数会降到(C_m^{m/2})(因为最多的情况是选所有有(m/2)(1)的二进制数,而这些数因为(1)数量相同,肯定互不包含),似乎又可以暴力了。

(90pts)算法

这个算法很神仙。

论如何求集合中选一个数与当前值进行位运算的(max)
(当然先(orz)CJrank1yyb)

如果是暴力的话,我们的方法有两种:

  • 对于当前数(x),暴力计算所有存在的数(a_i)中,(xigoplus i)的最大值,这样的复杂度是(O(2^{16}))
  • 对于每个可能出现的数维护一个当前所有数的最大值。即每次插入时,暴力计算所有的答案的最大值,这样子询问的时候可以(O(1))查询。

两种方法一种是插入(O(1)),询问(O(n)),另外一个是插入(O(n)),询问(O(1))
我们把两种东西结合一下,这样可以得到一个(O(nsqrt{n}))的方法。

我们对于每个数从中间分开,拆成前(8)个二进制位和后(8)个二进制位。
这样子我们可以预处理一个数组(t[i][j])
表示集合中一个前(8)位是(i)的数,后(8)位与(j)进行位运算的最大值。

因为位运算是可以按位贪心的,所以对于查询一个数(x),我们把它拆成(x=a×2^8+b)
每次先暴力(for)所有可能的前(8)位,找到与(a)能够构成最大值的那些数,然后对于找到的所有数的前八位(p),直接查(t[p][b])
因为前八位更大的数一定更大,那么影响结果的就只剩下后八位了,
把两个部分拼接起来就好了。
这样子暴力(for)前八位的复杂度是(O(2^8)),查找后面部分最大值的复杂度是(O(1))
所以这样子总的复杂度(O(2^8))
而对于集合中插入一个数,前(8)位唯一确定,每次只需要预处理后(8)位的结果。
时间复杂度还是(O(2^8)),总的复杂度还是(O(sqrt{n}))
最后记得反过来再跑一遍。

照搬该方法到本题,复杂度为(O(nsqrt{n}))
注意一下连边。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#define re register
#define il inline
#define ll long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define fp(i,a,b) for(re int i=a;i<=b;i++)
#define fq(i,a,b) for(re int i=a;i>=b;i--)
using namespace std;
const int mod=1e9+7,N=1e5+100;
int a[N],h[N],cnt,n,m,k,f[N],p=1,base,nxt[N],t[1<<11][1<<11],c[1<<11][1<<11],cp[N],tl[N],ma[N],fa[N];
ll ans;
il ll gi()
{
   re ll x=0,t=1;
   re char ch=getchar();
   while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
   if(ch=='-') t=-1,ch=getchar();
   while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
   return x*t;
}
il int find(re int x){return fa[x]?fa[x]=find(fa[x]):x;}
il void link(re int u,re int v)
{
  re int fu=find(u),fv=find(v);
  if(fu^fv)
    {
      nxt[tl[fv]]=h[fu];
      tl[fv]=tl[fu];
      fa[fu]=fv;
      ++p;
    }
}
il void solve()
{
  memset(f,0,sizeof(f));memset(t,-1,sizeof(t));memset(ma,-1,sizeof(ma));
  fp(i,1,n)
    if(!fa[i])
    {
    for(re int j=h[i];j;j=nxt[j])
      {
	re int x=a[j]>>k,y=a[j]&base;//x代表前一半位,y代表后一半位
	fp(z,0,(1<<m-k)-1)//枚举前一半位并统计答案
	  if(f[z])
	    {
	      re int ss=a[j]&t[z][y]|(z<<k&a[j]);//组合
	      if(ss>ma[i]) ma[i]=ss,cp[i]=c[z][y];
	    }
      }
  for(re int j=h[i];j;j=nxt[j])
    {
      re int x=a[j]>>k,y=a[j]&base;f[x]=1;//标记其存在
      fp(z,0,(1<<k)-1)//枚举后一半位并处理
	if((y&z)>t[x][z]) t[x][z]=y&z,c[x][z]=i;
    }
    }
  memset(f,0,sizeof(f));memset(t,0,sizeof(t));
  fq(i,n,1)
    if(!fa[i])
      {
    for(re int j=h[i];j;j=nxt[j])
      {
	re int x=a[j]>>k,y=a[j]&base;
	fp(z,0,(1<<m-k)-1)
	  if(f[z])
	    {
	      re int ss=a[j]&t[z][y]|(z<<k&a[j]);
	      if(ss>ma[i]) ma[i]=ss,cp[i]=c[z][y];
	    }
      }
  for(re int j=h[i];j;j=nxt[j])
    {
      re int x=a[j]>>k,y=a[j]&base;f[x]=1;
      fp(z,0,(1<<k)-1)
	if((y&z)>t[x][z]) t[x][z]=y&z,c[x][z]=i;
    }
      }
  fp(i,1,n)
    if(!fa[i])
      {
	re int u=find(i),v=find(cp[i]);
	if(u^v) ans+=ma[i],link(i,cp[i]);
      }
}
int main()
{
  freopen("mst.in","r",stdin);
  freopen("mst.out","w",stdout);
  n=gi();m=gi();k=m/2;base=(1<<k)-1;
  fp(i,1,n)
    {
      a[i]=gi();
      h[i]=tl[i]=i;
    }
  while(p<n) solve();
  printf("%lld
",ans);
  fclose(stdin);
  fclose(stdout);
  return 0;
}

(100pts)算法

倒序枚举用于连接的边权(p),用(a_p)表示(p)的最大超集(只多一个(1))所在联通块的代表元。枚举(p)的另一超集(只多一个(1)),若两者不在同一联通块,合并之。从大到小保证最优性。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#define re register
#define il inline
#define ll long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define fp(i,a,b) for(re int i=a;i<=b;i++)
#define fq(i,a,b) for(re int i=a;i>=b;i--)
using namespace std;
const int mod=1e9+7,N=1e5+100;
int a[1<<21],h[N],cnt,n,m,f[1<<21];
ll ans;
il ll gi()
{
   re ll x=0,t=1;
   re char ch=getchar();
   while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
   if(ch=='-') t=-1,ch=getchar();
   while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
   return x*t;
}
il int find(re int x){return f[x]?f[x]=find(f[x]):x;}
int main()
{
  freopen("mst.in","r",stdin);
  freopen("mst.out","w",stdout);
  n=gi();m=gi();
  fp(i,1,n)
    {
      re int x=gi();
      if(a[x]) ans+=x;
      a[x]=x;
    }
  fq(t,(1<<m)-1,0)
    {
      re int &u=a[t];
      for(re int i=0;!u&&i<m;i++) u=a[t|(1<<i)];
      if(!u) continue;
      fp(i,0,m-1)
	{
	  re int v=a[t|(1<<i)],fu=find(u),fv=find(v);
	  if(v&&fu!=fv) ans+=t,f[fv]=fu;
	}
    }
  printf("%lld
",ans);
  fclose(stdin);
  fclose(stdout);
  return 0;
}
原文地址:https://www.cnblogs.com/yanshannan/p/9457759.html