hdu4085 Peach Blossom Spring(斯坦纳树/子集条件限制)

题目

n(4<=n<=50)个点，

k(1<=k<=5,2*k<=n)个房子，对应k个避难所

编号1到k的是房子，编号n-k+1到n的是避难所，一个房子只能对应一个避难所

m(0<=m<=1e3)条已经破坏的路，u,v,w代表修缮u和v之间的道路需要花费w

花费最少的代价，使得4个房子对应4个避难所

思路来源

https://www.cnblogs.com/staginner/archive/2012/08/02/2619857.html

题解

同刚做过的zoj3613，对于这种一对一一边k个的情形，先把2*k个都压位，

先用斯坦纳树处理好带树根的dp[i][state]，

不妨两个房子，分别叫a、b；两个避难所，分别叫c、d

①用ans[state]把树根消掉，处理好若state连通所需的最小代价，

此时，state=1111，只考虑两个房子两个避难所abcd四个统统连在一起的最小代价

②枚举两个合法子集sub1和sub2，状压，使新的合法状态state的值更小 

此刻，state=1111，新考虑的是一个房子一个避难所连在一起，另一个房子一个避难所连在一起，前后不一定要相连的最小代价

比如a和c连在一起，b和d连在一起，前后不相连；或者a和d连在一起，b在c连在一起，前后不相连；

由于状压，所以考虑了所有的配对情况，只保留了最小值

卧槽 这是南昌邀请赛原题啊 wtm 当时就在想那个不连在一起的怎么解决

代码

#include<bits/stdc++.h>
using namespace std;
/*
 *  Steiner Tree：求，使得指定K个点连通的生成树的最小总权值
 *  st[i] 表示顶点i的标记值，如果i是指定集合内第m(0<=m<K)个点，则st[i]=1<<m 
 *  endSt=1<<K,这里K用tot表示 
 *  dp[i][state] 表示以i为根，连通状态为state的生成树值
 */
const int N=55;
const int M=1e3+10;
const int K=11;
int p[N],s[N];
int dp[N][1<<K],st[N],endSt;
int ans[1<<K];
int t,n,m,k;
int f,r,tot,sum[K];//factory,resource,tot
int u,v,num,cost;
int head[N],nex[M<<1],V[M<<1],W[M<<1],cnt;
bool vis[N][1<<K];
queue<int> que;
void add(int u,int v,int w)
{
	V[++cnt]=v;
	W[cnt]=w;
	nex[cnt]=head[u];
	head[u]=cnt;
}
void initSteinerTree()
{
	num=cost=0;
	cnt=tot=0;
	f=r=0;
	memset(head,0,sizeof head);
	memset(dp,-1,sizeof dp);
	memset(ans,-1,sizeof ans);
	memset(st,0,sizeof st);
    for(int i=1;i<=n;i++)
    memset(vis[i],0,sizeof vis[i]);
    //对每个st[i]赋值 对endSt赋值 
}
void input()
{
  	for(int i=1;i<=k;++i)
  	st[i]=1<<(tot++);
  	for(int i=1;i<=k;++i)
  	st[n+1-i]=1<<(tot++);
    for(int i=1;i<=m;++i)
    {
    	scanf("%d%d%d",&u,&v,&cost);
    	add(u,v,cost);
    	add(v,u,cost);
    }
    //tot是总的处于集合中的点的数量 
    endSt=1<<tot;
    for(int i=1;i<=n;i++)
    if(st[i])dp[i][st[i]]=0;
}
void update(int &a,int x)
{
    a=(a>x || a==-1)? x : a;
}
void SPFA(int state)
{
    while(!que.empty())
	{
        int u=que.front();
        que.pop();
        vis[u][state]=false;
        for(int i=head[u];i;i=nex[i])
		{
            int v=V[i];//以v为根 不管v不是所选集合节点还是是 
            if(dp[v][st[v]|state]==-1 || dp[v][st[v]|state]>dp[u][state]+W[i])
			{
        	    dp[v][st[v]|state]=dp[u][state]+W[i];
    	        if(st[v]|state!=state || vis[v][state])continue; //只用当前state更新所有根节点i的dp[i][state]的最小值 
        	    vis[v][state]=true;
       	 	    que.push(v);
            }
        }
    }
}
void steinerTree()
{
    for(int j=1;j<endSt;j++)
	{
        for(int i=1;i<=n;i++)
		{
            if(st[i] && (st[i]&j)==0) continue;//i要么不是选中集合的点 也就是新加的点 要么是枚举的j集合的子集 
            for(int sub=(j-1)&j;sub;sub=(sub-1)&j)
			{
                int x=st[i]|sub,y=st[i]|(j-sub);//sub是j的子集p+以i为根 j-sub是p的补集+以j为根 
                if(dp[i][x]!=-1 && dp[i][y]!=-1)
                update(dp[i][j],dp[i][x]+dp[i][y]);//拆成两半枚举 
            }
            if(dp[i][j]!=-1)//对于j状态 可以以i为根 
            {
            	que.push(i);
				vis[i][j]=true;
			}
        }
        SPFA(j);
    }
}
bool ok(int state)
{
	//保证状态下 房子和避难所数量相同 
	int res=0;
	for(int i=0;i<k;++i)
	if(state>>i&1)res++;
	for(int i=k;i<tot;++i)
	if(state>>i&1)res--;
	return res==0;
}
void solve()
{
	for(int j=1;j<endSt;++j)
	{
		for(int i=1;i<=n;i++)//枚举根为i 状态j最小值为ans[j] 
		if(dp[i][j]!=-1) 
		update(ans[j],dp[i][j]);
		//printf("ans[%d]:%d\n",j,ans[j]); 
	}
	for(int j=1;j<endSt;++j)
	{
		if(!ok(j))continue;
       	for(int sub=(j-1)&j;sub;sub=(sub-1)&j)
		{
         	int x=sub,y=j-sub;//sub是j的子集p+以i为根 j-sub是p的补集+以j为根
			if(!ok(x)||!ok(y))continue;
			if(ans[x]!=-1 && ans[y]!=-1) 
           	update(ans[j],ans[x]+ans[y]);//拆成两半枚举
        }
	} 
	if(ans[endSt-1]==-1)puts("No solution");
	else printf("%d\n",ans[endSt-1]);
}
int main()
{
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d%d",&n,&m,&k);
		initSteinerTree();
		input();
		steinerTree();
		solve();
	}
	return 0;
}