2017虹软校招算法题

前段时间参加了虹软的校招笔试，遇到了一道算法题，现摘录下来做个备份。

题目： 给定一个mXn的矩阵，如下图所示，里面所有元素都非负，试找出一条从左上角到右下角的一条路径，要求路径上的数字之和最小。注：在某一个位置只有两种选择，向下或者向右。

1	4	8	2	9
1	4	6	7	8
1	1	1	1	1

在以上矩阵中，其中数字之和最小的路径为1-1-1-1-1-1-1，其最小和为7.

请给出你的解题思路，并完成int min_sum_path(int *mat, int row, int col)。其中函数返回值为路径和的最小值，最后给出你算法的时间复杂度。

思路一： 可直接用回溯算法，每次从当前位置有2种选择，要么向右，要么向下，示例代码如下：

//===============================================================
//Author : 
//                    Andrew
//Date   :
//                2016/09/20 10:32:32
//Mail   :
//               andrewzhao1020@gmail.com
//===============================================================

#include <iostream>
#include<string>
#include<vector>
using namespace std;

//存放位置
int POSX[10];
int POSY[10];

void DFS(int depth, int i, int j, int *mat, int m, int n, int &cursum, int &minsum)
{
	//记录位置
	POSX[depth] = i;
	POSY[depth] = j;
	//迭代结束
	if (depth == n + m - 1)
	{
		return;
	}
	//更新最小值
	if (i == m - 1 && j == n - 1 && cursum < minsum)
		minsum = cursum;
	//右
	if (i<m && j + 1<n)
	{
		cursum += mat[i*n + (j + 1)];
		DFS(depth + 1, i, j + 1, mat, m, n, cursum, minsum);
		cursum -= mat[i*n + (j + 1)];
	}
	//下
	if (i + 1<m && j<n)
	{
		cursum += mat[(i + 1)*n + j];
		DFS(depth + 1, i + 1, j, mat, m, n, cursum, minsum);
		cursum -= mat[(i + 1)*n + j];
	}

}

int min_sum_path(int *mat, int row, int col)
{
	int cursum = mat[0];
	int minsum = 10000;
	//求最值
	DFS(0, 0, 0, mat, row, col, cursum, minsum);

	return minsum;
}

int main()
{
	int mat[] = { 1, 4, 8, 2, 9, 1, 4, 6, 7, 8, 1, 1, 1, 1, 1 };
	int row = 3;
	int col = 5;
	
	int answer = min_sum_path(mat,  row,  col);
	cout << "answer = " << answer << endl;
	
	//输出位置
	for (int i = 0; i<row + col - 1; i++)
	{
		cout << mat[POSX[i]* col + POSY[i]] << " ";
	}
	cout << endl;

	return 0;
}

思路二： 动态规划，将该一维数组视为二维数组，由于每个位置只能往右或者往下走，不妨记F(i,j)表示从起点到位置（i，j）处的路径和的最小值，则状态转移方程为:

F(i+1,j+1) = min(F(i,j+1), F(i+1,j)) + mat[i+1][j+1]   其中 0< i + 1<row   0< j+1<col

初始条件为，对于第一行的位置，因为只能向右到达 ，所以 F(0,j) = sum(0,j)    j=0...col; 对于第一列的位置，只能从起点向下到达，所以 F(i,0) = sum(i,0)  i = 0...row

据此，可得代码如下：

//===============================================================
//Author : 
//                    Andrew
//Date   :
//                2016/09/20 10:32:32
//Mail   :
//               andrewzhao1020@gmail.com
//===============================================================

#include <iostream>
#include<string>
#include<vector>
#include<algorithm>
using namespace std;

int min_sum_path(int *mat, int row, int col)
{
	int i, j;
	int dp[100][100] = { 0 };
	dp[0][0] = mat[0];
	//第一行
	for ( j = 1; j < col; j++)
		dp[0][j] = dp[0][j - 1] + mat[j];
	//第一列
	for ( i = 1; i < row; i++)
		dp[i][0] = dp[i - 1][0] + mat[i*col];
	//其他位置
	for ( i = 1; i < row;i++)
		for (j = 1; j < col; j++)
			dp[i][j] = min(dp[i - 1][j], dp[i][j - 1]) + mat[i*col + j];
	return dp[row-1][col-1];
}
int main()
{
	int mat[] = { 1, 4, 8, 2, 9, 1, 4, 6, 7, 8, 1, 1, 1, 1, 1 };
	int row = 3;
	int col = 5;
	
	int answer = min_sum_path(mat,  row,  col);
	cout << "answer = " << answer << endl;
	
	return 0;
}