luogu P3264 [JLOI2015]管道连接

首先最小斯坦纳树求出连接所有重要情报站的最小代价。

然后状压每个频道做一遍子集DP就完事了。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
#include<ctime> 

using namespace std;

const int N=1009,INF=1<<29;
priority_queue <pair<int,int> > q;
int n,m,p,head[N],cnt,Max,P[N],num,f[N][2050],vis[N],point[N],dp[2050],fuck[N];
struct Edge
{
	int nxt,to,w;
}g[N*10];

void add(int from,int to,int w)
{
	g[++cnt].nxt=head[from];
	g[cnt].to=to;
	g[cnt].w=w;
	head[from]=cnt;
}

void init()
{
	scanf("%d %d %d",&n,&m,&p);
	for (int i=1,x,y,z;i<=m;i++)
	{
		scanf("%d %d %d",&x,&y,&z);
		add(x,y,z),add(y,x,z);
	}
	Max=1<<p;
	for (int i=1;i<=n;i++)
		for (int j=0;j<Max;j++)
			f[i][j]=INF;
	for (int i=1,x,y;i<=p;i++)
	{
		scanf("%d %d",&x,&y),num=max(num,x);
		P[x]|=(1<<i-1);
		fuck[x]=(1<<y-1);
		f[y][1<<i-1]=0;
		point[i]=y;
	}
}

void Dijkstra(int s)
{
	memset(vis,0,sizeof(vis));
	while(!q.empty())
	{
		int x=q.top().second;q.pop();
		if(vis[x]) continue;
		vis[x]=1;
		for (int i=head[x];i;i=g[i].nxt)
		{
			int v=g[i].to;
			if(f[v][s]>f[x][s]+g[i].w)
			{
				f[v][s]=f[x][s]+g[i].w;
				q.push(make_pair(-f[v][s],v));
			}
		}
	}
}

int get(int s)
{
	for (int i=1;i<=p;i++)
		if((1<<i-1)&s)
			return point[i];
	return -1;
}

void work()
{
//	double c1=clock();
	for (int s=1;s<Max;s++)
	{
		for (int i=1;i<=n;i++)
		{
			for (int sub=s&(s-1);sub;sub=(sub-1)&s)
				f[i][s]=min(f[i][s],f[i][s^sub]+f[i][sub]);
			if(f[i][s]!=INF)
				q.push(make_pair(-f[i][s],i));
		}
		Dijkstra(s);
	}
//	printf("time:%.2lf
",(double)(clock()-c1)/CLOCKS_PER_SEC);
	for (int i=0;i<1<<num;i++)
		dp[i]=INF;
	for (int i=1;i<=num;i++)
		dp[P[i]]=f[get(P[i])][P[i]];
	for (int i=1;i<1<<num;i++)
	{
		int tmp=0,pos;
		for (int j=1;j<=num;j++)
			if((1<<j-1)&i)
				pos=j,tmp|=P[j];
		dp[i]=f[get(P[pos])][tmp];
//		printf("%d %d %d
",get(P[pos]),P[pos],f[get(P[pos])][P[pos]]);
		for (int sub=(i-1)&i;sub;sub=(sub-1)&i)
			dp[i]=min(dp[sub]+dp[i^sub],dp[i]);
	}
	printf("%d
",dp[(1<<num)-1]);
}

int main()
{
	init();
	work();
	return 0;
}