HYSBZ2595-游览计划（斯坦纳树）

[Wc2008]游览计划

Time Limit: 10 Sec   Memory Limit: 256 MBSec   Special Judge
Submit: 1790   Solved: 859
[ Submit][ Status][ Discuss]

Description

Input

第一行有两个整数，N和 M，描述方块的数目。 
接下来 N行， 每行有 M 个非负整数， 如果该整数为 0， 则该方块为一个景点；
否则表示控制该方块至少需要的志愿者数目。 相邻的整数用 （若干个） 空格隔开，
行首行末也可能有多余的空格。

Output

由 N + 1行组成。第一行为一个整数，表示你所给出的方案
中安排的志愿者总数目。 
接下来 N行，每行M 个字符，描述方案中相应方块的情况： 
z  ‘_’（下划线）表示该方块没有安排志愿者； 
z  ‘o’（小写英文字母o）表示该方块安排了志愿者； 
z  ‘x’（小写英文字母x）表示该方块是一个景点； 
注：请注意输出格式要求，如果缺少某一行或者某一行的字符数目和要求不
一致（任何一行中，多余的空格都不允许出现） ，都可能导致该测试点不得分。

Sample Input

4 4
0 1 1 0
2 5 5 1
1 5 5 1
0 1 1 0

Sample Output

6
xoox
___o
___o
xoox

HINT

 对于100%的数据，N,M,K≤10，其中K为景点的数目。输入的所有整数均在[0,2^16]的范围内

Source

Ljcc930提供SPJ

解题思路：斯坦纳树+路径保存

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <vector>
#include <bitset>
#include <functional>

using namespace std;

#define LL long long
const int INF = 0x3f3f3f3f;

struct node
{
	int x, y, z;
}pre1, nt1, pre[20][20][2000];
int n, m, tot;
int status;//表示0~n号节点都被选择时的状态+1  
int dis[20][20][2000], vis[20][20];
//dis[i][j]表示以i节点为根选择点集状态为j时的最小值；vis[i][j]表示i节点为点集j时是否在队列中  
int a[20][20], x[20][20];
int dir[4][2] = { {1,0},{0,1},{-1,0},{0,-1} };
queue<node> q;

void init()
{
	memset(vis, 0, sizeof vis);
	memset(pre, 0, sizeof pre);
	status = 1 << tot;
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= m; j++)
		{
			for (int k = 0; k < status; k++)
				dis[i][j][k] = INF;
			if (!a[i][j]) dis[i][j][x[i][j]] = 0;
		}
}

void SPFA(int sta)
{
	while (!q.empty())
	{
		pre1 = q.front();
		q.pop();
		vis[pre1.x][pre1.y] = 0;
		for (int i = 0; i < 4; i++)
		{
			int xx = pre1.x + dir[i][0];
			int yy = pre1.y + dir[i][1];
			if (xx > n || yy > m || xx <= 0 || yy <= 0) continue;
			if (dis[pre1.x][pre1.y][sta] + a[xx][yy] < dis[xx][yy][sta])
			{
				dis[xx][yy][sta] = dis[pre1.x][pre1.y][sta] + a[xx][yy];
				pre[xx][yy][sta] = { pre1.x,pre1.y,sta };
				if (!vis[xx][yy])
				{
					vis[xx][yy] = 1;
					q.push({ xx,yy });
				}
			}
		}
	}
}

void Steiner_Tree()
{
	for (int i = 0; i < status; i++)
	{
		for (int j = 1; j <= n; j++)
		{
			for (int k = 1; k <= m; k++)
			{
				for (int p = i; p; p = (p - 1) & i)
				{
					int temp = dis[j][k][p] + dis[j][k][i - p] - a[j][k];
					if (temp < dis[j][k][i]) dis[j][k][i] = temp, pre[j][k][i] = { j,k,p };
				}
				if (dis[j][k][i] != INF)
				{
					nt1 = { j,k };
					q.push(nt1);
					vis[j][k] = 1;
				}
			}
		}
		SPFA(i);
	}
}

void dfs(int x, int y, int z)
{
	if (!x) return;
	vis[x][y] = 1;
	node temp = pre[x][y][z];
	dfs(temp.x, temp.y, temp.z);
	if (x == temp.x&&y == temp.y) dfs(x, y, z - temp.z);
}

int main()
{
	while (~scanf("%d %d", &n, &m))
	{
		tot = 0;
		for (int i = 1; i <= n; i++)
		{
			for (int j = 1; j <= m; j++)
			{
				scanf("%d", &a[i][j]);
				if (!a[i][j]) x[i][j] = 1 << tot, tot++;
			}
		}
		init();
		Steiner_Tree();
		int flag = 0;
		for (int i = 1; i <= n; i++)
		{
			for (int j = 1; j <= m; j++)
				if (!a[i][j])
				{
					printf("%d\n", dis[i][j][status - 1]);
					dfs(i, j, status - 1);
					for (int k = 1; k <= n; k++)
					{
						for (int p = 1; p <= m; p++)
						{
							if (!a[k][p]) printf("x");
							else if (vis[k][p]) printf("o");
							else printf("_");
						}
						printf("\n");
					}
					flag = 1; break;
				}
			if (flag) break;
		}
	}
	return 0;
}