BZOJ 4006: [JLOI2015]管道连接

最小斯坦纳树森林

可以这样理解题意：在一幅n个点，m条双向边的图中，有k个关键点，每个关键点有一个颜色。现在要求将一些边连起来，使得任意一种颜色的所有关键点相连。连接每条边的花费已给出，问最小花费为多少。

分析：

显然，此时的答案是由大于等于1棵最小斯坦纳树组成的最小斯坦纳树森林，答案<=一棵最小斯坦纳树的答案。

设f[i][p]表示：当前位置为i，连接的关键点的位置为p的最小花费。（这一部与最小斯坦纳树一模一样）

设dp[p]表示：将所有颜色在集合p中的关键点连接的最小花费。

至于实现，通过离散化预处理等简单操作即可。

#include <bits/stdc++.h>
using namespace std;
const int N=1e3+5,M=3e3+5,K=11;
int n,m,k,u,v,w,x,y,max_statue,inf,tot;
int f[N][1<<K],ki[K],dp[1<<K];
bool vis[N];
int cnt,head[N];
struct edge{int next,to,w;}e[M<<1];
struct number{int id,pos;}num[K];
struct node
{
	int id,dis;
	inline bool operator < (const node &x) const
	{
		return x.dis<dis;
	}
};

inline bool cmp(number a,number b)
{
	return a.id<b.id;
}

inline void add(int u,int v,int w)
{
	cnt++;
	e[cnt].next=head[u];
	e[cnt].to=v;
	e[cnt].w=w;
	head[u]=cnt;
}

priority_queue<node>q;
inline void dij(int p)
{
	memset(vis,false,sizeof(vis));
	while (q.size())
	{
		node u=q.top(); q.pop();
		if (vis[u.id]) continue;
		vis[u.id]=true;
		int x=u.id;
		for (register int i=head[x]; i; i=e[i].next)
		if (f[e[i].to][p]>f[x][p]+e[i].w)
		{
			f[e[i].to][p]=f[x][p]+e[i].w;
			q.push((node){e[i].to,f[e[i].to][p]});
		}
	}
}

int main(){
	scanf("%d%d%d",&n,&m,&k);
	for (register int i=1; i<=m; ++i)
	{
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w); add(v,u,w);
	}
	memset(f,60,sizeof(f)); inf=f[0][0];
	for (register int i=1; i<=k; ++i) 
	{
		scanf("%d%d",&x,&y);
		num[i].id=x; num[i].pos=y;
		ki[x]|=1<<(i-1);
		f[y][1<<(i-1)]=0;
	}
	max_statue=(1<<k)-1;
	for (register int p=1; p<=max_statue; ++p)
	{
		for (register int i=1; i<=n; ++i)
		{
			for (register int pp=p; pp; pp=p&(pp-1)) f[i][p]=min(f[i][p],f[i][pp]+f[i][p^pp]);
			if (f[i][p]!=inf) q.push((node){i,f[i][p]});		
		}
		dij(p);
	}
	
	sort(num+1,num+k+1,cmp);
	for (register int i=1; i<=k; ++i) 
	if (num[i].id!=num[i-1].id) tot++,num[tot].id=num[i].id,num[tot].pos=num[i].pos; 
	
	memset(dp,60,sizeof(dp));
	dp[0]=0;
	max_statue=(1<<tot)-1;
	for (register int p=1; p<=max_statue; ++p)
	{
		int now=0;
		for (register int i=1; i<=tot; ++i) if ((1<<(num[i].id-1))&p) now|=ki[i];
		for (register int i=1; i<=n; ++i) dp[p]=min(dp[p],f[i][now]);
		for (register int pp=p; pp; pp=p&(pp-1)) dp[p]=min(dp[p],dp[pp]+dp[p^pp]);
	}
    printf("%d\n",dp[max_statue]);
return 0;
}