题目大意是给定一个矩阵,每次向右走一列,可以从第一列的任意一行开始,最后一列的任意一行结束,每次可以走下一列的正右,右上和右下,求一条路径,是路径上权值和最小,并且打印路径,路径序列要字典序最小。
我们用m存储原式矩阵,定义状态dp(i,j)为从第i行第j列走出矩阵所需最小和,path(i,j)为从第i行第j列走出矩阵的最优路径中,第j+1列应该走的行号。那么状态转移方程为:
dp(i,j)=Min{dp(i-1,j+1),dp(i,j+1),dp(i+1,j+1)}+m[i][j]。其中i-1和i+1需单独判断,并且预处理dp矩阵的最后一列的下一列为0。从后向前迭代,记录路径后从前向后输出即可。
#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
long long dp[15][110];
int path[15][110],m[15][110];
long long Min(long long a,long long b,long long c,int pa,int pb,int pc,int& pos)
{
int ans;
if(a>b){
ans=b;pos=pb;
}
else if(a==b){
ans=a;
if(pa>pb) pos=pb;
else pos=pa;
}
else{
ans=a;pos=pa;
}
if(ans>c){
ans=c;pos=pc;
}
else if(ans==c&&pos>pc)
pos=pc;
return ans;
}
int main()
{
//freopen("in.txt","r",stdin);
int r,c;
while(scanf("%d%d",&r,&c)!=EOF){
for(int i=1;i<=r;i++){
dp[i][c+1]=0;
for(int j=1;j<=c;j++) scanf("%lld",&m[i][j]);
}
for(int j=c;j>0;j--)
for(int i=1;i<=r;i++){
long long a,b,tc;
int pa,pb,pc;
if(i-1==0) {a=dp[r][j+1];pa=r;}
else {a=dp[i-1][j+1];pa=i-1;}
b=dp[i][j+1];pb=i;
if(i+1>r) {tc=dp[1][j+1];pc=1;}
else {tc=dp[i+1][j+1];pc=i+1;}
dp[i][j]=Min(a,b,tc,pa,pb,pc,path[i][j])+m[i][j];
}
long long Min=0x7FFFFFFF,pos;
for(int i=1;i<=r;i++) if(Min>dp[i][1]){
Min=dp[i][1];pos=i;
}
int j=1;
while(j!=(c+1)){
cout<<pos;
pos=path[pos][j];
j++;
if(j!=(c+1)) cout<<" ";
else cout<<endl;
}
cout<<Min<<endl;
}
return 0;
}