题意:
给定一个m x n矩阵,要求从上到下寻找一条最短路径,这条最短路径顺序必须是从小到下,每一层的时候向左向右都行。
思路:
申请数组如下:
dp [ i ] [ j ] :抵达第i层第j个点的最小消耗。
pre [ i ] [ j ]:抵达第i层第j个点的前一个点,即保存路径。因为每个点的pre点只有三种情况,上左右,所以当 pre [ i ] [ j ]==j时,即代表是上,j-1是左。j+1是右。
转移方程:
先是从左到右:dp [ i ][ j ]=Min(dp[ i-1 ][ j ],dp[ i ][ j-1])+w;
再是从右到左:dp [ i ][ j ]=Min(dp[ i ][ j ],dp[ i ][ j+1] + w );
这样的时间复杂度为O(m*n);
挺不错的一道题目。
还是一次AC,今天人品不错。
#include<iostream>
#include<vector>
#include<string>
#include<queue>
#include<cmath>
#include<algorithm>
#define llong long long
#define Min(a,b) (a<b?a:b)
#define Max(a,b) (a>b?a:b)
#define Abs(a) ((a)>0?(a):-(a))
#define Mod(a,b) (((a)-1+(b))%(b)+1)
using namespace std;
int n,m;
const int N=505;
const int M=105;
const int inf=2099999999;
int dp[M][N];
int a[M][N];
int pre[M][N];
void myprint(int i,int j)
{
if(i==1)
{
printf("%d\n",j);
return;
}
if(pre[i][j]==j)
{
myprint(i-1,j);
}
else
{
myprint(i,pre[i][j]);
}
printf("%d\n",j);
}
int main()
{
scanf("%d%d",&m,&n);
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
{
scanf("%d",a[i]+j);
}
for(int j=1;j<=n;j++)
dp[1][j]=a[1][j];
for(int i=1;i<=m;i++)
dp[i][0]=dp[i][n+1]=inf;
for(int i=2;i<=m;i++)
{
for(int j=1;j<=n;j++)
{
if(dp[i-1][j]<dp[i][j-1])
{
dp[i][j]=dp[i-1][j]+a[i][j];
pre[i][j]=j;
}
else
{
dp[i][j]=dp[i][j-1]+a[i][j];
pre[i][j]=j-1;
}
}
for(int j=n;j>=1;j--)
{
if(dp[i][j]>dp[i][j+1]+a[i][j])
{
dp[i][j]=dp[i][j+1]+a[i][j];
pre[i][j]=j+1;
}
}
}
int ans=inf,k=0;
for(int i=1;i<=n;i++)
if(dp[m][i]<ans)
{
ans=dp[m][i];
k=i;
}
myprint(m,k);
return 0;
}