程序员面试指南中的一个题目:
给定一个矩阵m,从左上角开始每次只能向右或向下走,最后达到右下角的位置,路径上所有数字的累加和就是路径和,求所有路径中的最小路径和。
根据动态规划实现,构建矩阵dp,dp[i][j]为从左上角(0,0)到(i,j)的最小路径和。对于第一行,只能向右累加;对于第一列,只能向下累加。其他位置都是从其左侧或上面的点中选择较小的值再加上该节点的值,即dp[i][j] = min(dp[i-1][j], dp[i][j-1]) + m[i][j]。最后右下角的值就是最小路径和。
C++实现如下:
#define MIN(x,y) ((x) <= (y) ? (x) : (y))
int getMinPath(int **Matrix, int row, int column)
{
if (NULL == Matrix)
{
return 0;
}
int **dp = new int*[row];
for (int i = 0; i < row; i++)
{
dp[i] = new int[column];
}
// 注意二维数组参数的处理
//dp[0][0] = Matrix[0][0];
dp[0][0] = *((int*)Matrix);
for (int i = 1; i < column; i++)
{
//dp[0][i] = dp[0][i - 1] + Matrix[0][i];
dp[0][i] = dp[0][i - 1] + *((int*)Matrix + i);
}
for (int i = 1; i < row; i++)
{
//dp[i][0] = dp[i - 1][0] + Matrix[i][0];
dp[i][0] = dp[i - 1][0] + *((int*)Matrix + column*i);
}
for (int i = 1;i < row; i ++)
{
for (int j = 1; j < column; j++)
{
//dp[i][j] = MIN(dp[i][j - 1], dp[i - 1][j]) + Matrix[i][j];
dp[i][j] = MIN(dp[i][j - 1], dp[i - 1][j]) + *((int*)Matrix + column*i + j);
}
}
int result = dp[row - 1][column - 1];
for (int i = 0; i < row; i++)
{
delete[] dp[i];
dp[i] = NULL;
}
delete[] dp;
dp = NULL;
return result;
}
int main(int argc, char* argv[])
{
int Matrix[4][4] = {{1,3,5,9},{8,1,3,4},{5,0,6,1},{8,8,4,0}};
// 注意二维数组作为函数参数的处理方式
printf("Min path is %d\n", getMinPath((int**)Matrix,4,4));
return 0;
}
这里要注意二维数组作为函数参数的处理方式,除了上述方式外,还有其他方式,但都需要传递列数,可以再学习一下。