HDU 4568 SPFA + TSP

时间:2022-10-28 16:18:16

这道题是长沙邀请赛的题,当时是道签到题。

这种题还是很常见的,讲一下思路。

首先是预处理出每个宝藏之间的距离,还有到边的距离,直接对每个宝藏进行一次SPFA就可以了。

然后就是经典的求TSP的过程。

#include <set>
#include <map>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <iomanip>
#include <cstring>
#include <iostream>
#include <algorithm>
#define Max 2505
#define FI first
#define SE second
#define ll long long
#define PI acos(-1.0)
#define inf 1111111111
#define LL(x) ( x << 1 )
#define bug puts("here")
#define PII pair<int,int>
#define RR(x) ( x << 1 | 1 )
#define mp(a,b) make_pair(a,b)
#define mem(a,b) memset(a,b,sizeof(a))
#define infA(a) for (int i = 0 ; i <= n ; ++ i)a[i] = inf ;
#define REP(i,s,t) for( int i = ( s ) ; i <= ( t ) ; ++ i ) using namespace std; inline void RD(int &ret) {
char c;
do {
c = getchar();
} while(c < '0' || c > '9') ;
ret = c - '0';
while((c=getchar()) >= '0' && c <= '9')
ret = ret * 10 + ( c - '0' );
} inline void OT(int a) {
if(a >= 10)OT(a / 10) ;
putchar(a % 10 + '0') ;
}
#define N 222
#define K 15 int Map[N][N] ;
struct tru {
int x ,y ;
} tk[K] ;
int tt ;
int MM[K][K] ;
int dp[1 << K][K] ;
int dis[N][N] ;
bool vis[N][N] ;
PII qe[1111111] ;
int D[K] ;
int n , m ;
int mx[4] = {0 , 0 , 1 , -1} ;
int my[4] = {1 , -1 , 0 , 0} ;
int inmap(int x ,int y) {
if(x >= 0 && x < n && y >= 0 && y < m && Map[x][y] != -1)return 1 ;
return 0 ;
}
void init(int pos) {
for (int i = 0 ; i < n ; i ++ ) {
for (int j = 0 ; j < m ; j ++ ) {
dis[i][j] = inf ;
vis[i][j] = 0 ;
}
}
dis[tk[pos].x][tk[pos].y] = Map[tk[pos].x][tk[pos].y] == -1 ? inf : 0 ;
int h = 0 , t = 0 ;
qe[h ++ ] = mp(tk[pos].x ,tk[pos].y) ;
while(h > t) {
PII tp = qe[t ++ ] ;
vis[tp.FI][tp.SE] = 0 ;
if(tp.FI == 0 || tp.FI == n - 1 || tp.SE == 0 || tp.SE == m - 1) {
D[pos] = min(D[pos] , dis[tp.FI][tp.SE]) ;
}
for (int i = 0 ; i < 4 ; i ++ ) {
int tx = tp.FI + mx[i] ;
int ty = tp.SE + my[i] ;
if(inmap(tx , ty)) {
if(dis[tx][ty] > dis[tp.FI][tp.SE] + Map[tx][ty]) {
dis[tx][ty] = dis[tp.FI][tp.SE] + Map[tx][ty] ;
if(!vis[tx][ty]) {
vis[tx][ty] = 1 ;
qe[h ++ ] = mp(tx ,ty) ;
}
}
}
}
}
} int main() {
#ifndef ONLINE_JUDGE
freopen("in.txt","r",stdin) ;
freopen("out.txt","w",stdout) ;
#endif
int T ;
cin >> T ;
while ( T -- ) {
cin >> n >> m ;
REP(i , 0, n - 1 ) {
REP(j , 0 , m - 1)scanf("%d",&Map[i][j]) ;
}
cin >> tt ;
REP(i , 0 , tt - 1) {
RD(tk[i].x) ;
RD(tk[i].y) ;
}
REP(i , 0 , tt - 1) {
REP(j , 0 , tt - 1) {
MM[i][j] = inf ;
}
MM[i][i] = 0 ;
D[i] = inf ;
}
for (int i = 0 ; i < (1 << tt) ; i ++ ) {
for (int j = 0 ; j < tt ; j ++ )dp[i][j] = inf ;
} REP(i , 0 , tt - 1) {
init(i) ;
REP(j , 0 , tt - 1) {
if(i == j)continue ;
MM[i][j] = min(MM[i][j] , dis[tk[j].x][tk[j].y]) ;//宝藏到宝藏之间的最近距离
}
dp[1 << i][i] = D[i] + Map[tk[i].x][tk[i].y] ;//该宝藏到达边的最近距离
}
for (int i = 0 ; i < 1 << tt ; i ++ ) {
for (int j = 0 ; j < tt ; j ++ ) {
if(i & (1 << j) == 0)continue ;
if(dp[i][j] == inf)continue ;
for (int k = 0 ; k < tt ; k ++ ) {
if(i & (1 << k))continue ;
int tp = i | (1 << k) ;
dp[tp][k] = min(dp[tp][k] , dp[i][j] + MM[j][k]) ;
}
}
}
int ans = inf ;
for (int i = 0 ; i < tt ; i ++ ) {
ans = min(ans , dp[(1 << tt) - 1][i] + D[i] ) ;
}
cout << ans << endl;
}
return 0 ;
}
/*
5
4 4
3 3 3 3
3 -1 -1 -1
3 -1 3 3
3 -1 -1 -1
1
2 2 5 5
1 1 1 1 1
1 -1 -1 -1 -1
1 -1 -1 -1 -1
1 -1 -1 -1 -1
1 -1 -1 -1 -1
2
4 0
0 4 3 3
3 2 3
5 4 3
1 4 2
1
1 1
3 3
3 2 3
5 4 3
1 4 2
2
1 1
2 2 */