hdu 1619 Unidirectional TSP 记忆化搜索

来源:互联网 发布:日系rpg手游 知乎 编辑:程序博客网 时间:2024/06/11 13:06

题意:一个n*m的矩阵,从第一列任意位置处出发,求走到最右端的最小权重值。往右走的规则为,对角线和水平方向都可以,如果超过最上面,则认为与下面相邻,如果超过最下面,则认为与最上面相邻.

有些类似树塔,每个位置都有一个dp[I][j],代表从当前点走到最右方需要的最少步数,据说是《算法竞赛入门经典》中的例题,打印路径为书上的代码,自己写的太长了

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int maxn = 110;int map[maxn][maxn],vis[maxn][maxn];int dp[maxn][maxn],fa[maxn][maxn];int n,m,first;int Min(int a,int b,int c){int d = a<b?a:b;if(c>d)return d;else return c;}int find_min(int x,int y){int a,b,c;if(x<0)x=n-1;else if(x>=n)x=0;if(dp[x][y])return dp[x][y];a=b=c=0;b = find_min(x+1,y+1);a = find_min(x,y+1);c = find_min(x-1,y+1);dp[x][y] += (Min(a,b,c)+map[x][y]);return dp[x][y];}void printpath(int x,int y){if(y!=m){if(first){printf("%d",x+1);first=0;}elseprintf(" %d",x+1);int r[3] = {x-1,x,x+1};if(x-1<0)r[0]=n-1;if(x+1>=n)r[2]=0;sort(r,r+3);int maxv = 1<<30;int s;for(int k=0;k<3;k++){if(maxv>dp[r[k]][y+1]){maxv=dp[r[k]][y+1];s=r[k];}}printpath(s,y+1);}}int main(){while(scanf("%d%d",&n,&m)!=EOF){for(int i=0;i<n;i++)for(int j=0;j<m;j++)scanf("%d",&map[i][j]);memset(dp,0,sizeof(dp));first=1;for(int i=0;i<n;i++)dp[i][m-1]=map[i][m-1];for(int i=0;i<n;i++)find_min(i,0);int ans=(1<<30);int s;for(int i=0;i<n;i++)if(ans>dp[i][0])ans=dp[i][0];for(int i=0;i<n;i++)if(ans==dp[i][0]){s=i;break;}printpath(s,0);printf("\n%d\n",ans);}} 

0 0
原创粉丝点击