可以从上一层,左层和右层过来,求和最小的路径。
两遍dp,从两端各走一遍
状态方程:
dp[i][j]=min(dp[i][j],dp[i][j+1]+a[i][j]);
dp[i][j]=min(dp[i][j],dp[i][j-1]+a[i][j]);
这题要注意最小值要开到2*10^9,否则会wa,我存路径的数组一开始也开小了 导致re,输出路径不能取长的,否则不小心的话还会ole,以后多注意这类问题吧
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
int main()
{
int n,m;
int a[111][555],p[111][555],vis[51111];
int dp[111][555];
scanf("%d%d",&n,&m);
memset(dp,0,sizeof(dp));
memset(p,0,sizeof(p));
for(int i=1; i<=n; i++)
for(int j=1; j<=m; j++)
scanf("%d",&a[i][j]);
for(int i=1; i<=n; i++)
{
for(int j=1; j<=m; j++)
{
dp[i][j]=dp[i-1][j]+a[i][j];
}
for(int j=2; j<=m; j++)
{
int v=1;
if(dp[i][j]==dp[i][j-1]+a[i][j])
{
v=0;
}
else
dp[i][j]=min(dp[i][j],dp[i][j-1]+a[i][j]);
if((dp[i][j]==(dp[i][j-1]+a[i][j]))&&v)
{
p[i][j]=j-1;
}
}
for(int j=m-1; j>=1; j--)
{
int v=1;
if(dp[i][j]==dp[i][j+1]+a[i][j])
{
v=0;
}
else
dp[i][j]=min(dp[i][j],dp[i][j+1]+a[i][j]);
if((dp[i][j]==(dp[i][j+1]+a[i][j]))&&v)
{
p[i][j]=j+1;
}
}
}
int min=2000000000;
int ans;
for(int i=1; i<=m; i++)
{
if(min>dp[n][i])
{
min=dp[n][i];
ans=i;
}
}
int i=n,j=ans;
int t=0;
vis[t++]=ans;
while(i>1)
{
int w=p[i][j];
if(w==0)
{
i--;
w=j;
}
else
{
j=w;
}
vis[t++]=w;
}
for(i=t-1; i>=0; i--)
{
printf("%d\n",vis[i]);
}
return 0;
}