BZOJ 2595: [Wc2008]游览计划 斯坦纳树

学习斯坦纳树

学习了一发斯坦纳树，发现是个很神奇的东西

这题可以看作求把一个点集之中的关键点连接起来的一棵最小生成树，所以可以用斯坦纳树的姿势来做

题中景点数最多10个，可以使用状态压缩来保存状态

f[i][j][s]表示根为(i,j)并连通状态为s时的最小花费

本题中有两种转移方式

f[i][j][s]=min{f[i][j][s']+f[i][j][s-s']-a[i][j]} //合并时把重复的那个点减一次，因为有阶段性可以直接dp

f[i][j][s]=f[i'][j'][s]+a[i][j] //(i,j)与(i',j')有公共边即相邻，可以直接转移，用spfa即可(有些时候可以floyd，但是spfa比较稳定...)

至于输出方案的话搞一个pre数组记录一下就好了

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define N 11
#define INF 1e9
using namespace std;

int read()
{
	int a=0,f=1;char c=getchar();
	while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9'){a=a*10+c-'0';c=getchar();}
	return a*f;
}

const int dx[4]={0,1,-1,0},dy[4]={1,0,0,-1};

struct node{int x,y;};
struct path{
	int x,y,s;
	path(){}
	path(int a,int b,int c):x(a),y(b),s(c){}
}fa[N][N][1<<N];
queue<node>Q;
int n,m,k;
int a[N][N],mark[N][N],f[N][N][1<<N];

void spfa(int s)
{
	while(!Q.empty())
	{
		int x=Q.front().x,y=Q.front().y;
		Q.pop();
		for(int i=0;i<4;++i)
		{
			int nx=x+dx[i],ny=y+dy[i];
			if(nx<1||nx>n||ny<1||ny>m) continue;
			if(f[nx][ny][s]>f[x][y][s]+a[nx][ny])
			{
				f[nx][ny][s]=f[x][y][s]+a[nx][ny];
				fa[nx][ny][s]=path(x,y,s);
				if(!mark[nx][ny])
				{
					node tmp;
					tmp.x=nx,tmp.y=ny;
					Q.push(tmp);
					mark[nx][ny]=1;
				}
			}
		}
		mark[x][y]=0;
	}
}

bool vis[N][N];
void dfs(int x,int y,int s)
{
	vis[x][y]=1;
	path tmp=fa[x][y][s];
	if(tmp.x==0&&tmp.y==0) return;
	dfs(tmp.x,tmp.y,tmp.s);
	if(tmp.x==x&&tmp.y==y) dfs(tmp.x,tmp.y,s-tmp.s);
}

int main()
{
	n=read(),m=read();
	for(int i=1;i<=n;++i)
		for(int j=1;j<=m;++j)
			for(int s=0;s<1<<N;++s)
				f[i][j][s]=INF;
	for(int i=1;i<=n;++i)
		for(int j=1;j<=m;++j)
		{
			a[i][j]=read();
			if(!a[i][j]) f[i][j][1<<(k++)]=0;
		}
	int top=1<<k;
	for(int state=0;state<top;++state)
	{
		for(int i=1;i<=n;++i)
			for(int j=1;j<=m;++j)
			{
				for(int s=state&(state-1);s;s=state&(s-1))
					if(f[i][j][state]>f[i][j][s]+f[i][j][state-s]-a[i][j])
						f[i][j][state]=f[i][j][s]+f[i][j][state-s]-a[i][j],fa[i][j][state]=path(i,j,s);
				if(f[i][j][state]<INF)
				{
					node tmp;tmp.x=i,tmp.y=j;
					Q.push(tmp);
					mark[i][j]=1;
				}
			}
		spfa(state);
	}
	int x=0,y=0,flag=1;
	for(int i=1;i<=n;++i) if(flag)
		for(int j=1;j<=m;++j)
			if(!a[i][j])
			{
				x=i,y=j;
				flag=0;
				break;
			}
	dfs(x,y,top-1);
	printf("%d\n",f[x][y][top-1]);
	for(int i=1;i<=n;++i)
	{
		for(int j=1;j<=m;++j)
			if(!a[i][j]) putchar('x');
			else if(vis[i][j]) putchar('o');
			else putchar('_');
		puts("");
	}
	return 0; 
}