题目大意
传送门
给一个m行n列的整数矩阵,从第一列任何一个位置出发依次往右、右上、右下走一格,最终到达最后一列。
要求经过的整数之和最小。矩阵的第一行的上一行是最后一行,最后一行的下一行是第一行。
输出路径上每行的行号。多个解的时候输出字典序小的。
样例
input
5 6
3 4 1 2 8 6
6 1 8 2 7 4
5 9 3 9 9 5
8 4 1 3 2 6
3 7 2 8 6 4
5 6
3 4 1 2 8 6
6 1 8 2 7 4
5 9 3 9 9 5
8 4 1 3 2 6
3 7 2 1 2 3
2 2
9 10 9 10
output
1 2 3 4 4 5
16
1 2 1 5 4 5
11
1 1
19
解释
思路
d[i][j]表示为从格子(i,j)出发到最后一列的最小开销。
则d[x][n]都为edge[x][n],这个是初始态。
对状态做初始化:
for (int i = 0; i < m; i++)
for (int j = 0; j < n; j++)
{
if (j == n-1) dp[i][j] = edge[i][j];
else dp[i][j] = INF;
}
递推公式:
for (int j = n - 2; j >= 0; j--)
for (int i = 0; i < m; i++)
for (int k = -1; k <= 1; k++)
{
dp[i][j] = min(dp[i][j], dp[(i+k+m)%m][j + 1] + edge[i][j]);
}
代码
#include <stdio.h>
#define MAXN 105
#define INF 1<<30
#define min(a,b) (a<b)? a:b
int edge[MAXN][MAXN];
int dp[MAXN][MAXN];
int m, n;
int main(){
while (~scanf("%d %d",&m,&n))
{
for (int i = 0; i < m; i++)
for (int j = 0; j < n; j++)
scanf("%d", &edge[i][j]);
for (int i = 0; i < m; i++)
for (int j = 0; j < n; j++){
if (j == n-1) dp[i][j] = edge[i][j];
else dp[i][j] = INF;
}
for (int j = n - 2; j >= 0; j--)
{
for (int i = 0; i < m; i++)
{
for (int k = -1; k <= 1; k++)
{
dp[i][j] = min(dp[i][j], dp[(i+k+m)%m][j + 1] + edge[i][j]);
}
}
}
for (int i = 0; i < m; i++)
{
for (int j = 0; j < n; j++)
{
printf("%d\t", dp[i][j]);
}
printf("\n");
}
}
}
Hit
在递推公式的时候
要注意的是 j 要在循环的最外面,算完了一组j之后才可以算j-1。
如果j的循环在第二层,那就错了