问题描述
给定一个包含非负整数的 m x n 网格 grid,请找出一条从左上角到右下角的路径,使得路径上的数字总和最小。
移动规则:每次只能向下或者向右移动一步。
例如,给定网格:
grid = [
[1, 3, 1],
[1, 5, 1],
[4, 2, 1]
]
最小路径和为 7,路径为 1 -> 3 -> 1 -> 1 -> 1。
该问题在实际中有广泛应用,如地图导航中的最小代价路径、机器人路径规划、游戏关卡中的最优路线选择等。
核心思路:二维动态规划
状态定义
设 dp[i][j] 表示从左上角 (0, 0) 到达位置 (i, j) 的最小路径和。
状态转移方程
由于每次只能向下或向右移动,到达 (i, j) 只有两个来源:
– 从上方 (i-1, j) 向下移动
– 从左侧 (i, j-1) 向右移动
因此状态转移方程为:
$$dp[i][j] = grid[i][j] + \min(dp[i-1][j], dp[i][j-1])$$
边界条件
dp[0][0] = grid[0][0](起点)- 第一行
dp[0][j] = dp[0][j-1] + grid[0][j](只能从左侧过来) - 第一列
dp[i][0] = dp[i-1][0] + grid[i][0](只能从上方过来)
完整Java实现
/**
* 最小路径和问题求解器
* 使用二维动态规划,支持空间优化到一维数组
*/
public class MinimumPathSum {
/**
* 方法一:标准二维动态规划
* 时间复杂度: O(m*n),空间复杂度: O(m*n)
*
* @param grid 输入的二维网格
* @return 从左上到右下的最小路径和
*/
public int minPathSum2D(int[][] grid) {
if (grid == null || grid.length == 0 || grid[0].length == 0) {
return 0;
}
int m = grid.length; // 行数
int n = grid[0].length; // 列数
// dp[i][j] 表示到达 (i, j) 的最小路径和
int[][] dp = new int[m][n];
// 初始化起点
dp[0][0] = grid[0][0];
// 初始化第一行:只能从左侧移动过来
for (int j = 1; j < n; j++) {
dp[0][j] = dp[0][j - 1] + grid[0][j];
}
// 初始化第一列:只能从上方移动过来
for (int i = 1; i < m; i++) {
dp[i][0] = dp[i - 1][0] + grid[i][0];
}
// 填充其余位置:选择从上方或左侧过来的较小值
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
// 状态转移:当前格子的值 + min(从上方来, 从左侧来)
dp[i][j] = grid[i][j] + Math.min(dp[i - 1][j], dp[i][j - 1]);
}
}
// 右下角即为最终答案
return dp[m - 1][n - 1];
}
/**
* 方法二:空间优化为一维数组
* 时间复杂度: O(m*n),空间复杂度: O(n)
*
* 观察发现:计算dp[i][j]时只需要当前行和上一行的数据
* 因此可以用一维数组滚动更新,大幅降低空间占用
*/
public int minPathSum1D(int[][] grid) {
if (grid == null || grid.length == 0 || grid[0].length == 0) {
return 0;
}
int m = grid.length;
int n = grid[0].length;
// 一维dp数组,dp[j]表示当前行第j列的最小路径和
int[] dp = new int[n];
// 初始化第一行的路径和
dp[0] = grid[0][0];
for (int j = 1; j < n; j++) {
dp[j] = dp[j - 1] + grid[0][j];
}
// 逐行更新
for (int i = 1; i < m; i++) {
// 每行第一个元素只能从上方过来
dp[0] = dp[0] + grid[i][0];
// 更新当前行的其余元素
for (int j = 1; j < n; j++) {
// dp[j] 目前还是上一行第j列的值(从上方来)
// dp[j-1] 已经是当前行第j-1列的值(从左侧来)
dp[j] = grid[i][j] + Math.min(dp[j], dp[j - 1]);
}
}
return dp[n - 1];
}
/**
* 方法三:原地修改(不使用额外空间)
* 时间复杂度: O(m*n),空间复杂度: O(1)
*
* 直接在输入网格上修改,适用于允许修改原数组的场景
*/
public int minPathSumInPlace(int[][] grid) {
if (grid == null || grid.length == 0 || grid[0].length == 0) {
return 0;
}
int m = grid.length;
int n = grid[0].length;
// 初始化第一行
for (int j = 1; j < n; j++) {
grid[0][j] += grid[0][j - 1];
}
// 初始化第一列
for (int i = 1; i < m; i++) {
grid[i][0] += grid[i - 1][0];
}
// 填充其余位置
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
grid[i][j] += Math.min(grid[i - 1][j], grid[i][j - 1]);
}
}
return grid[m - 1][n - 1];
}
/**
* 扩展:不仅返回最小路径和,还返回具体路径
*/
public void printMinPath(int[][] grid) {
int m = grid.length;
int n = grid[0].length;
int[][] dp = new int[m][n];
// path[i][j] 记录到达(i,j)是从上方来(true)还是左侧来(false)
boolean[][] fromAbove = new boolean[m][n];
dp[0][0] = grid[0][0];
for (int j = 1; j < n; j++) {
dp[0][j] = dp[0][j - 1] + grid[0][j];
fromAbove[0][j] = false;
}
for (int i = 1; i < m; i++) {
dp[i][0] = dp[i - 1][0] + grid[i][0];
fromAbove[i][0] = true;
}
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (dp[i - 1][j] < dp[i][j - 1]) {
dp[i][j] = dp[i - 1][j] + grid[i][j];
fromAbove[i][j] = true;
} else {
dp[i][j] = dp[i][j - 1] + grid[i][j];
fromAbove[i][j] = false;
}
}
}
// 回溯打印路径
System.out.println("最小路径和: " + dp[m - 1][n - 1]);
System.out.print("路径: ");
printPath(grid, fromAbove, m - 1, n - 1);
System.out.println();
}
private void printPath(int[][] grid, boolean[][] fromAbove, int i, int j) {
if (i == 0 && j == 0) {
System.out.print(grid[0][0]);
return;
}
if (fromAbove[i][j]) {
printPath(grid, fromAbove, i - 1, j);
} else {
printPath(grid, fromAbove, i, j - 1);
}
System.out.print(" -> " + grid[i][j]);
}
public static void main(String[] args) {
MinimumPathSum solution = new MinimumPathSum();
int[][] grid = {
{1, 3, 1},
{1, 5, 1},
{4, 2, 1}
};
System.out.println("===== 最小路径和 =====");
System.out.println("二维DP结果: " + solution.minPathSum2D(grid)); // 7
System.out.println("一维优化结果: " + solution.minPathSum1D(grid)); // 7
// 复制一份用于原地修改测试
int[][] gridCopy = {
{1, 3, 1},
{1, 5, 1},
{4, 2, 1}
};
System.out.println("原地修改结果: " + solution.minPathSumInPlace(gridCopy)); // 7
// 打印具体路径
solution.printMinPath(grid);
}
}
复杂度分析
| 方法 | 时间复杂度 | 空间复杂度 | 特点 |
|---|---|---|---|
| 二维DP | O(m × n) |
O(m × n) |
思路清晰,易于理解和回溯路径 |
| 一维优化 | O(m × n) |
O(n) |
空间减半,适合大规模数据 |
| 原地修改 | O(m × n) |
O(1) |
不占用额外空间,但会破坏原数据 |
扩展思考
- 带有障碍物的网格:如果某些格子不能通过,可以在状态转移时增加判断条件,跳过硬编码为障碍物的格子。
- 三维扩展:在三维空间中寻找最小代价路径,状态变为
dp[i][j][k],扩展方向增加前后移动。 - 概率路径:如果每个格子有一定概率被阻挡,可以结合期望计算最优策略。
总结
最小路径和是理解二维动态规划的绝佳入门问题。通过定义清晰的状态、分析状态转移来源、处理边界条件,我们能够系统性地解决网格路径问题。更重要的是,通过观察计算依赖关系,我们可以将空间复杂度从二维优化到一维甚至常数级,这是面试和工程实践中非常重要的优化技巧。