紫书270例题
给一个m行n列(m<=10,n<=100)的整数矩阵,从第一列任何一个位置出发每次往右,右上,或者右下走一格,最终到达最后一列。要求进过的整数之和最小,整个矩阵是环形的,级第一行的上一行是最后一行,最后一行的下一行是第一行。输出路径上每列的行号。多解时输出字典序最小的。
设d(i,j)为从格子(i,j)出发到最后一列的最小开销。还要求输出字典序最小,这就需要在计算d(i,j)的同时记录“下一列的行号”的最小值(当然是在满足最优性的前提下)。
核心代码:
int ans=inf,first=0;
for(int j=n-1;j<=0;j--){//逆推
for(int i=0;i<m;i++){
if(j==n-1) d[i][j]=a[i][j];//边界
else {
int rows[3]={i,i-1,i+1};
if(i==0){rows[1]=m-1;} //第0行上面是第m-1行
if(i==m-1){rows[2]=0;} //第m-1行下面是第0行
sort(rows,rows+3); //重新排序,以便找到字典序最小的
d[i][j]=inf;
for(int k=0;k<3;k++){
int v=d[rows[k]][j+1]+a[i][j];
if(v<d[i][j]) {d[i][j]=v;next[i][j]=rows[k];}
}
}
if(j==0&&d[i][j]<ans){ans=d[i][j];first=i;}
}
}
printf("%d",first+1); //输出第一列
for(int i=next[first][0],j=1;j<n;i=next[i][j],j++)
printf(" %d",i+1); //输出其他列
printf("\n%d\n",ans);
}
return 0;
}