题目大意:给一个N*M的矩形,矩形的每个点都有一个值(-1表示不能通过),表示到这个点需要用的花费。现在给定K个宝物的坐标,要求从任意位置进入矩阵取完这些宝物,再从任意位置出来需要的最少花费。
解题思路:先求出每个宝物到其他宝物点的距离与离开边界的距离 再用状态压缩dp求最佳路线(TSP)
#include<stdio.h>
#include<iostream>
#include<string>
#include<string.h>
#include<math.h>
#include<functional>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int maxn = 215;
const int inf = 1<<30;//0x7f;
typedef __int64 LL;
int n,m,k;
int map[maxn][maxn],dist[maxn][maxn]; //dist[i][j]表示到点(i,j)的距离
int out[25],dis[25][25],dp[1<<14][25]; //out[i]表示第i个点离开边界的最小距离 dis[i][j]表示第i个宝物到第j个宝物的距离 dp[i][j]中i表示已走点的状态 j表示最后个点的位置(dp[(二进制)0..011][1]表示走了前2个点停在第一个点)
bool vis[maxn][maxn];
int xs[] = { 0,1,0,-1 };
int ys[] = { 1,0,-1,0 };
struct node
{
int x,y;
}tre[25];
typedef pair<int,int>pii;
void Dijkstra( int sx,int sy,int r )
{
priority_queue<pii,vector<pii>,greater<pii> >que;
for( int i = 0; i < n; i ++ )
for( int j = 0; j < m; j ++ )
dist[i][j] = inf;
dist[sx][sy] = 0;
que.push( make_pair(0,sx*m+sy) );
while( !que.empty() )
{
pii u = que.top(); que.pop();
int x = u.second/m;
int y = u.second%m;
if( x == 0 || x == n-1 || y == 0 || y == m-1 )
out[r] = min( out[r],dist[x][y] );
for( int i = 0; i < 4; i ++ )
{
int xx = x + xs[i];
int yy = y + ys[i];
if( xx >= 0 && xx < n && yy >= 0 && yy < m && map[xx][yy] != -1 )
{
if( dist[xx][yy] > dist[x][y] + map[xx][yy] ){
dist[xx][yy] = dist[x][y] + map[xx][yy];
que.push( make_pair(dist[xx][yy],xx*m+yy) );
}
}
}
}
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("data.txt","r",stdin);
#endif
int cas,u,v,w;
scanf("%d",&cas);
while( cas -- ){
scanf("%d%d",&n,&m);
for( int i = 0; i < n; i ++ )
for( int j = 0; j < m; j ++ )
scanf("%d",&map[i][j]);
scanf("%d",&k);
for( int i = 0; i < k; i ++ )
scanf("%d%d",&tre[i].x,&tre[i].y);
for( int i = 0; i < k; i ++ ){ //求每个宝物到其他宝物点的距离
out[i] = inf;
Dijkstra( tre[i].x,tre[i].y,i );
for( int j = 0; j < k; j ++ ){
dis[i][j] = dist[tre[j].x][tre[j].y];
}
}
for( int i = 0; i < (1<<k); i ++ )
for( int j = 0; j < k; j ++ )
dp[i][j] = inf;
for( int i = 0; i < k; i ++ )
{
dp[1<<i][i] = map[tre[i].x][tre[i].y] + out[i]; //初始化从边界到每个宝物点的距离
}
for( int i = 0; i < (1<<k); i ++ ) //枚举状态
{
for( int j = 0; j < k; j ++ ) //枚举最后所在点
{
if( i&(1<<j) == 0 || dp[i][j] == inf ) continue; //最后所在点与已走状态不符或dp[i][j]走不到
for( int e = 0; e < k; e ++ ) //枚举从j点到e点
{
if( i&(1<<e) ) continue;
dp[i|(1<<e)][e] = min(dp[i|(1<<e)][e],dp[i][j]+dis[j][e]);
}
}
}
int ans = inf;
for( int i = 0; i < k; i ++ ) //枚举最后从哪个点出去总距离最小
ans = min( ans,dp[(1<<k)-1][i] + out[i] );
printf("%d\n",ans);
}
return 0;
}