现在有一个 m * n 的整数矩阵,请你编写一个程序计算出一条从左到右穿过矩阵的路径,并使此路径的费用最小。路径从矩阵的左侧的第一列的任意单元格开始,逐步穿过矩阵到达最右侧的一列的任意单元格。每一步是指从某单元格进入它一列的相邻单元格(如下图,可以是横向或斜向)。矩阵的第一行和最后一行实际是相邻的,你可以想象矩阵是包裹在一个横放的圆柱体外面。
路径的花费是指这条路径所穿越的所有单元格中的数字之和。
穿越两个略有不同的 5 * 6 的矩阵的路径如下图所示,这两个矩阵只有最后一行的数字不同。右侧的路径显示了第一行和最后一行相邻的效果。
输入
输入包括一系列矩阵描述。每个矩阵描述的第一行是 m 和 n,即矩阵的行数和列数;之后的 m 行,每行包括 n 个以空格分开的整数,则是当前矩阵的值,注意矩阵的值未必是正数。
矩阵的行数 m 和列数 n 的范围是:1 <= m <= 10、 1 <= n <= 100;所有路径的费用值都可以用 30bit 的整数表示。
输出
针对每一个矩阵,找出费用最小的路径,并将其输出。每个矩阵的输出包括两行,第一行是路径本身,即输出每一步所在的行,第二行则是该路径的费用。
如果对于同一个矩阵有多条不同的费用最小路径,则输出左端行号较小的一条。
来源
UVa: 116
测试输入 | 期待的输出 | 时间限制 | 内存限制 | 额外进程 | |
---|---|---|---|---|---|
测试用例 1 | 以文本方式显示
| 以文本方式显示
| 1秒 | 1024KB | 0 |
乍一看像是搜索,但是其实是个dp的题。其实有了思路之后还是很好做的。
首先说顺序问题。如果是正序的话,很难确定应该是从哪一行开始找起,即找到最后一列之后如果结果有相同的话还得再返回来,看看哪个才是左端行数最短的哪条,这样就有些冗杂了。所以采用倒序,从后往前找,找到结果之后如果有相同的话直接看哪个左端行数最短即可。
然后就是循环,dp数组存当前点时候的ans,然后把所有的点都过一遍,对于每个点而言是找右,右上,右下的值来确定这个点的结果。记得要考虑边界点时候的特殊情况。
遍历到最左端之后就有了min值,再一个一个返回去找,就找到了路径,然后记录一下每次的行数即可。
一开始错了一个地方,就是初始化的时候应该给dp数组初始化为0还是无穷大。初始化是0的话,当只有一行的时候它更倾向于去加0而不是那一行的数,所以初始化的时候用INF。
AC代码:
#include<stdio.h>
#include<string.h>
int dp[11][101], a[11][101], num[101] = { 0 };
#define INF 0x3f3f3f3f
int min(int a, int b)
{
return a < b ? a : b;
}
int main()
{
int n, m;
while (scanf("%d%d", &n, &m) != EOF)
{
memset(dp, INF, sizeof(dp));
memset(num, 0, sizeof(num));
for (int i = 0; i < n; i++)
for (int j = 0; j < m; j++)
scanf("%d", &a[i][j]);
for (int i = 0; i < n; i++)
dp[i][m - 1] = a[i][m - 1];
for (int i = m - 2; i >= 0; i--)
for (int j = 0; j < n; j++)
{
dp[j][i] = a[j][i];
if (j>0 && j < n - 1)
dp[j][i] += min(min(dp[j][i + 1], dp[j + 1][i + 1]), dp[j - 1][i + 1]);
else if (j == 0)
dp[j][i] += min(min(dp[j][i + 1], dp[j + 1][i + 1]), dp[n - 1][i + 1]);
else if (j == n - 1)
dp[j][i] += min(min(dp[j][i + 1], dp[0][i + 1]), dp[j - 1][i + 1]);
}
int mini, mine = dp[n - 1][0];
for (int i = n - 1; i >= 0; i--)
if (mine >= dp[i][0])
{
mine = dp[i][0];
mini = i;
}
num[0] = mini;
for (int i = 1; i < m; i++)
{
int flag = 0;
for (int j = 0; j < n; j++)
{
if (mini == 0 && j != 0 && j != 1 && j != n - 1)
continue;
if (mini == n - 1 && j != n - 1 && j != n - 2 && j != 0)
continue;
if (mini>0 && mini < n - 1 && j != mini&&j != mini + 1 && j != mini - 1)
continue;
if (dp[mini][i - 1] - a[mini][i - 1] == dp[j][i])
{
mini = j;
num[i] = mini;
flag = 1;
}
if (flag)
break;
}
}
printf("%d", num[0]+1);
for (int i = 1; i < m; i++)
printf(" %d", num[i]+1);
printf("\n%d\n", mine);
}
return 0;
}