题意:给定nxm的矩形,每个格子值∈[0,100],从左上角出发,将所有非0的格子都要经过至少一次,最后回到原点,两个格子之间的距离为1,求最少需要多少步?(值大于0的格子最多10个)
题解:典型的货郎担(TSP)问题:有n个城市,两两之间有道路联通,给出城市i和j之间的距离,求一条经过每个城市1次,最后回到起点的路线,要求距离之和最短。
TSP问题在n小的时候可以用动态规划求解,也可以全排列
全排列的话就是将除起点外的点的访问顺序全部枚举出来,每次算出距离之和,最后统计最小值即可。不过全排列的时间复杂度是n!,这道题n小于10,所以可以用这种方法,但是如果n达到15,那肯定不行了,就可以考虑dp
dp就是将起点规定为编号0,dp[i][S]表示从起点i出发,将集合S中的城市全部走一遍,再回到0号城市的最短总距离,
那么递推方程就是 dp[i][S] = min(dp[j][S-{j}]+dist(i,j)) | j∈S
因为n比较小,所以这里的集合S可以用二进制的方法来表示
(注意位运算的优先级问题)
#include <cstdio>
#include <cstring>
#include <algorithm>
#define INF 100000000
using namespace std;
int n,m;
int ma[100][100];
int cnt; // number of valid cells
struct Position {
int row,col;
} pos[20];
int dp[20][1<<12];
// compute the distance between x and y
int distance(int x,int y) {
int r1 = pos[x].row;
int c1 = pos[x].col;
int r2 = pos[y].row;
int c2 = pos[y].col;
return abs(r1-r2)+abs(c1-c2);
}
int main() {
while(scanf("%d%d",&n,&m) == 2) {
pos[0] = (Position){0,0};
cnt = 1;
for(int i = 0;i < n;i++)
for(int j = 0;j < m;j++) {
scanf("%d",&ma[i][j]);
if(i==0&&j==0) continue;
if(ma[i][j]) {
pos[cnt++] = (Position){i,j};
}
}
for(int i = 0;i <20;i++)
for(int j = 0;j < (1<<12);j++)
dp[i][j] = INF;
for(int i = 0;i < cnt;i++) {
dp[i][0] = distance(0,i);
}
for(int i = 1;i < (1<<(cnt-1));i++ )
for(int j = 0;j < cnt;j++) {
for(int k = 1;k <cnt;k++) {
if((((1<<(k-1))) & i) == 0) continue;
dp[j][i] = min(dp[j][i],distance(j,k)+dp[k][i-(1<<(k-1))]);
}
}
printf("%d\n",dp[0][(1<<(cnt-1))-1]);
}
return 0;
}