动态规划
动态规划是一种以空间换时间的算法思想,其本质是利用数学中递推公式的思想来解决存在大量重复计算的问题。当一个问题可以递归地分解为若干子问题,且这些子问题存在大量重叠时,动态规划通过缓存中间结果来避免重复计算,从而将指数级的时间复杂度优化为多项式级。 动态规划特别适合求解最值问题(最大值、最小值、方案数等),这类问题通常具备三个核心特征:
- 最优子结构:问题的最优解包含子问题的最优解,大问题的答案可以从小问题的答案推导出来。
- 重叠子问题:递归求解过程中,相同的子问题会被反复计算,这是动态规划优于普通递归的关键所在。
- 无后效性:一旦某个状态确定之后,后续的决策不会影响到该状态之前的决策结果,这意味着我们可以放心地用子问题的结果来推导更大的问题,而不需要回头修改。
从复杂度的角度看,动态规划可以将
动态规划解题步骤
解决动态规划问题的核心难点在于两件事:定义状态和推导状态转移方程。其中状态定义是基础,状态定义错了,转移方程就无从谈起。
定义状态:确定
dp[i]或dfs(i)表示什么含义。状态定义需要满足两个条件:能描述问题的解,且能从更小的状态推导而来。状态变量的选择直接决定了问题的维度和复杂度。常见的状态定义模式包括:
- 以位置
为结尾: dp[i]表示前个元素的某个最优值(如最大子数组和) - 以区间
为范围: dp[i][j]表示区间内的最优值(如最长回文子串) - 背包类:
dp[i][w]表示前个物品、容量为 时的最优值 - 状态机:用多个 dp 数组表示不同状态之间的转换(如股票买卖中的持有/不持有)
- 以位置
初始化状态:设置边界条件,这是 DP 的启动点。初始化的值必须与状态定义严格一致,否则后续所有结果都会出错。通常初始化的是最小子问题的解,比如
、 或 等。 推导状态转移方程:分析当前状态可以从哪些更小的状态转移而来,考虑所有可能的转移路径,取最优值(最值问题)或累加(计数问题)。这是 DP 的核心,也是最难的一步。
确定遍历顺序和返回值:遍历顺序必须保证在计算当前状态时,它依赖的所有前置状态已经计算完毕。返回值通常是 dp 数组中的某个特定元素或遍历过程中的极值。
空间优化(可选):如果状态转移只依赖有限个前驱状态,可以用滚动变量代替整个数组,将空间复杂度从
降到 。
function dynamicProgramming(n) {
// 1. 定义状态数组
let dp = new Array(n).fill(0);
// 2. 初始化初始状态和边界条件
dp[0] = initial_value;
// 3. 根据状态转移方程计算每个状态
for (let i = 1; i <= n; i++) {
dp[i] = dpTransition(dp, i);
}
// 4. 返回最终结果
return getResult(dp);
}
// 状态转移函数,根据具体问题定义
function dpTransition(dp, i) {
return some_function_of(dp, i);
// 示例:dp[i] = min(dp[i-1] + cost1, dp[i-2] + cost2);
}
function getResult(dp) {
return dp[dp.length - 1]
}自底向上和自顶向下
动态规划有两种等价的实现方式:dp 数组(自底向上)和记忆化搜索 dfs(自顶向下)。
两者的核心逻辑完全等价——dp 数组的维度 index 与 dfs 函数的参数一一对应,转移方程的形式也类似:
// 斐波那契
dp[i] = dp[i-1] + dp[i-2];
dfs(i) = dfs(i-1) + dfs(i-2);二者区别体现在思考的方向,dp 数组在开展时,从低处往高处逐步填表,每一次递增一次自变量,正向思考;但是 dfs 往往从高处开始搜索,逐步分解题目中的复杂度,在纵深区每一次递减一次自变量,待 dfs 触底时,从回溯区逐步归纳总结各个分支的结果,从而选择出当前节点的最佳状态,并且返回到上一层。
选择策略:
- 选择哪种方式主要取决于问题的自然思考方向。如果问题的递推关系天然是从 base case 向上构建的(如斐波那契、背包问题),dp 数组更直观;如果问题的思考方向是从最终状态向下分解的(如树形 DP、某些博弈问题),dfs 记忆化搜索更容易理解和编写;
- 记忆化搜索通过函数调用栈隐式地处理了遍历顺序的问题,而 dp 数组需要手动确定正确的遍历顺序——这在多维 DP 中是一个常见的出错点;
- 从性能的角度上对比,dp 数组一般比记忆化搜索要好,不过,由于 dfs 可以剪枝,在状态空间稀疏时可能比 dp 数组更高效;
- dp 数组往往是我们知道了推导出最终问题所需要经历的步数,而 dfs 搜索则可以 if 语句灵活判定结束的位置,适合于一开始不知道遍历步数的问题;
如何找到 dp 状态的定义
随着刷题量的增加,我们往往会在接触同类的题目时回忆其类似题目的定义方式,dp 也和其他题目可以通过分析入参是那几个,先初步判定题目中的复杂度规模,从而从复杂度下手,采取降解复杂度的方式,将其定义成 dp 数组的状态,例如入参是一个数组,那么可以预想的一种状态定义方式就是,我们只考虑该数组的前 i 项的时候的子问题作为 dp 数组的状态定义。
dp 数组状态定义对于"结尾"这个限定词,有莫名的情有独钟。例如最长递增子序列这道题中,dp 数组的定义不是 arr[0..i] 的最长递增子序列,而是以 arr[i] 结尾的最长递增子序列,通过限定结尾,加强了 dp[i-1] 和 dp[i] 的关系,从而方便我们进行递推公式的寻找。
如何思考状态转移方程
推导状态转移方程的关键在于:对每个状态,考虑"最后一步"做了什么选择。这个思考方式可以系统化为以下步骤:
- 明确当前状态表示什么(回扣状态定义)
- 列举到达当前状态的所有可能路径(从哪些已知前置状态可以转移到当前状态,往往就是
dp[i-1]迁移到dp[i]) - 对每条路径计算对应的值
- 根据问题类型,取最优值(最值问题)或求和(计数问题)
以经典的爬楼梯问题为例:每次可以爬 1 或 2 个台阶,问到达第 dp[i] 为到达第
这个"看最后一步"的思维方式贯穿了几乎所有的 DP 问题。背包问题中,最后一步是"选或不选当前物品";编辑距离中,最后一步是"插入、删除或替换";股票问题中,最后一步是"买入、卖出或不操作"。只要能穷举最后一步的所有选择,状态转移方程就自然浮现出来了。特别地,如果 dp 的自变量是代表入参的规模,那么最后一步的选项往往就是“选或不选”最后一步所新增的物品/参数。
典型题目类型
理解 DP 问题的高频模式有助于快速识别问题类型和选择合适的状态定义。每种模式有其典型的状态定义方式和推导技巧,下面逐一分析并给出经典例题。
1. 线性 DP
状态沿一个或多个线性序列推进,是最基础的 DP 模式。状态定义通常是"以位置
斐波那契数列是最简单的线性 DP,用来理解状态定义和转移方程的基本模式:
// 递归解法 O(2^n) —— 存在大量重复计算
function fibonacci(n) {
if (n <= 1) return n;
return fibonacci(n-1) + fibonacci(n-2);
}
// 动态规划解法 O(n) —— 自底向上填表
function fibonacciDP(n) {
if (n <= 1) return n;
const dp = [0, 1];
for (let i = 2; i <= n; i++) {
dp[i] = dp[i-1] + dp[i-2];
}
return dp[n];
}
// 空间优化 O(1) —— 只保留前两个状态
function fibonacciOptimized(n) {
if (n <= 1) return n;
let prev = 0, curr = 1;
for (let i = 2; i <= n; i++) {
[prev, curr] = [curr, prev + curr];
}
return curr;
}斐波那契数列揭示了 DP 的核心价值:朴素递归
当线性序列从一维扩展到二维时,状态变成
function longestCommonSubsequence(text1, text2) {
const m = text1.length;
const n = text2.length;
const dp = Array.from({ length: m + 1 }, () => Array(n + 1).fill(0));
for (let i = 1; i <= m; i++) {
for (let j = 1; j <= n; j++) {
if (text1[i-1] === text2[j-1]) {
dp[i][j] = dp[i-1][j-1] + 1;
} else {
dp[i][j] = Math.max(dp[i-1][j], dp[i][j-1]);
}
}
}
return dp[m][n];
}不匹配时取
2. 区间 DP
状态是序列上的一个区间
以石子合并为例,每次只能合并相邻的两堆石子,代价为两堆石子数之和,求将所有石子合并成一堆的最小总代价。定义
function mergeStones(stones) {
const n = stones.length;
const prefix = [0];
for (let i = 0; i < n; i++) prefix.push(prefix[i] + stones[i]);
const dp = Array.from({ length: n }, () => Array(n).fill(0));
for (let len = 2; len <= n; len++) {
for (let i = 0; i + len - 1 < n; i++) {
const j = i + len - 1;
dp[i][j] = Infinity;
for (let k = i; k < j; k++) {
dp[i][j] = Math.min(dp[i][j], dp[i][k] + dp[k+1][j] + prefix[j+1] - prefix[i]);
}
}
}
return dp[0][n-1];
}遍历顺序是关键:外层枚举区间长度、内层枚举起点,确保计算
3. 背包 DP
在容量约束下选择物品的最优方案,是工程中应用最广的 DP 模型之一(资源分配、任务调度等)。状态定义为
// 0/1背包问题 —— 每个物品只能选一次
function knapsack(weights, values, maxWeight) {
const n = weights.length;
const dp = Array.from({ length: n + 1 }, () => Array(maxWeight + 1).fill(0));
for (let i = 1; i <= n; i++) {
const newOne = weights[i-1]
for (let w = 1; w <= maxWeight; w++) {
// 随着 i 的增加,每次对有一个新的物品被纳入我们的考虑范围
// 如果这个新的物品能够被当前的容量所容纳,则我们有完整的两种方案,选或不选
if (newOne <= w) {
dp[i][w] = Math.max(
dp[i-1][w], // 不选,dp[i][w] 继承 i-1
values[i-1] + dp[i-1][w-newOne] // 选,则当前新物品的价格,加上剩余空间所能够容纳的最大价值(已经被计算过了)
);
} else {
dp[i][w] = dp[i-1][w]; // 没得选
}
}
}
return dp[n][maxWeight];
}
// 完全背包问题 —— 物品可以重复选择
function completeKnapsack(weights, values, W) {
const n = weights.length;
const dp = Array(W + 1).fill(0);
for (let i = 0; i < n; i++) {
for (let w = weights[i]; w <= W; w++) {
dp[w] = Math.max(dp[w], dp[w-weights[i]] + values[i]);
}
}
return dp[W];
}0/1 背包的核心技巧是"选或不选":不选则
4. 树形 DP
在树结构上做 DP,通常通过后序遍历自底向上合并子树的结果。状态定义一般以"以节点
以二叉树的最大路径和为例,路径定义为从树中任意节点出发,沿父-子连接到达任意节点的序列,要求找出节点值之和最大的那条路径。dfs 返回以当前节点为端点的最大单向路径和(只取一条分支向上传递),同时在过程中更新全局最大值(考虑两条分支都取的情况)。
function maxPathSum(root) {
let maxSum = -Infinity;
function dfs(node) {
if (!node) return 0;
const left = Math.max(dfs(node.left), 0);
const right = Math.max(dfs(node.right), 0);
maxSum = Math.max(maxSum, left + node.val + right);
return node.val + Math.max(left, right);
}
dfs(root);
return maxSum;
}这里的"取正贡献"是一个实用技巧:如果子树的路径和为负数,不如不选,直接取 0。这种剪枝思维在树形 DP 中很常见,可以简化状态转移的逻辑。后序遍历保证了处理当前节点时左右子树的结果已经就绪,不需要额外的遍历顺序设计。
5. 状态机 DP
通过多个状态变量描述不同的"持有状态",在状态之间进行转移。适用于有明确状态切换规则的场景,比如股票买卖中的持有/不持有、任务调度中的工作/休息。
// 买卖股票的最佳时机(一次交易)
function maxProfit(prices) {
let minPrice = Infinity;
let maxProfit = 0;
for (let price of prices) {
minPrice = Math.min(minPrice, price);
maxProfit = Math.max(maxProfit, price - minPrice);
}
return maxProfit;
}
// 买卖股票的最佳时机(多次交易),搜索方法
function maxProfit(prices) {
const memo = {}
function useMemo(f) {
return (a, b) => {
if (!memo[a])
memo[a] = {}
if (memo[a][b])
return memo[a][b]
return memo[a][b] = f(a, b)
}
}
// 定义为 dfs(idx, holds),在规模为 idx 时,如果那时持有股票,则所获得收益是多少;
// 允许负收益的存在,因为如果只买入,还没有卖,那么此时认为是负收益
const dfs = useMemo(function (idx, holds) {
if (idx <= 0)
return holds ? -prices[0] : 0;
if (holds)
return Math.max(dfs(idx - 1, false) - prices[idx], dfs(idx - 1, true))
else
return Math.max(dfs(idx - 1, true) + prices[idx], dfs(idx - 1, false))
})
return dfs(prices.length - 1, false) // 当数据规模为题目所给定的全部数据时,此时认为我们手上的股票必定要卖出,否则就是倒亏钱
};
// 递推方法
function maxProfit(prices) {
const len = prices.length
const dp = Array.from({ length: len }, () => ({ hold: 0, unhold: 0 }))
dp[0] = { hold: -prices[0], unhold: 0 }
for (let i = 1; i < len; i++) {
const curr = dp[i]
const prev = dp[i - 1]
curr.hold = Math.max(prev.unhold - prices[i], prev.hold)
curr.unhold = Math.max(prev.hold + prices[i], prev.unhold)
}
return dp[len - 1].unhold
};
function maxProfitMultiple(prices) {
let profit = 0;
for (let i = 1; i < prices.length; i++) {
if (prices[i] > prices[i-1]) {
profit += prices[i] - prices[i-1];
}
}
return profit;
}一次交易的做法本质上不是标准的 DP,而是利用了贪心的观察:遍历过程中记录历史最低价,当前价格减去历史最低价就是当天卖出的最大利润。多次交易的贪心策略更巧妙:只要明天涨就今天买明天卖,等价于把所有上涨段都吃满。
股票问题的通用 DP 模型是状态机:定义
6. 状态压缩 DP
用一个整数的二进制位来表示集合状态,适用于状态空间不大但难以用数组直接索引的场景。例如
以旅行商问题(TSP)为例,给定
function tsp(dist) {
const n = dist.length;
const fullMask = (1 << n) - 1;
const dp = Array.from({ length: 1 << n }, () => Array(n).fill(Infinity));
dp[1][0] = 0;
for (let mask = 1; mask <= fullMask; mask++) {
for (let i = 0; i < n; i++) {
if (!(mask & (1 << i))) continue;
for (let j = 0; j < n; j++) {
if (mask & (1 << j)) continue;
const nextMask = mask | (1 << j);
dp[nextMask][j] = Math.min(dp[nextMask][j], dp[mask][i] + dist[i][j]);
}
}
}
let result = Infinity;
for (let i = 1; i < n; i++) {
result = Math.min(result, dp[fullMask][i] + dist[i][0]);
}
return result;
}状态压缩的核心在于用位运算代替集合操作:mask | (1 << j) 表示将城市 mask & (1 << j) 检查城市
7. 数位 DP
处理与数字的各位相关的问题,例如统计
以统计 1 到 tight 记录当前位是否受上界约束:如果前面的所有位都恰好等于上界对应位的值,当前位只能取
function countDigitOne(n) {
const digits = String(n).split('').map(Number);
const memo = new Map();
function dfs(pos, count, tight) {
if (pos === digits.length) return count;
const key = `${pos},${count},${tight}`;
if (memo.has(key)) return memo.get(key);
const limit = tight ? digits[pos] : 9;
let result = 0;
for (let d = 0; d <= limit; d++) {
result += dfs(pos + 1, count + (d === 1 ? 1 : 0), tight && d === limit);
}
memo.set(key, result);
return result;
}
return dfs(0, 0, true);
}tight 参数是数位 DP 的精髓:当 tight 为 true 时,当前位的选择范围被限制在上界以内;一旦某一位选择了小于上界的值,tight 变为 false,后续所有位都可以自由选择
空间优化技巧
空间优化的核心观察是:如果当前状态只依赖有限个前驱状态,就不需要存储整个 DP 表。
// 滚动变量:斐波那契数列,空间 O(n) -> O(1)
function fibonacciCompressed(n) {
if (n <= 1) return n;
let prev = 0, curr = 1;
for (let i = 2; i <= n; i++) {
[prev, curr] = [curr, prev + curr];
}
return curr;
}
// 滚动数组:0/1 背包,空间 O(n*W) -> O(W)
function knapsackRolling(weights, values, W) {
const n = weights.length;
const dp = Array(W + 1).fill(0);
for (let i = 0; i < n; i++) {
for (let w = W; w >= weights[i]; w--) {
dp[w] = Math.max(dp[w], dp[w-weights[i]] + values[i]);
}
}
return dp[W];
}
// 记忆化搜索:自顶向下的 DP
function fibonacciMemo(n) {
const memo = new Map();
function helper(n) {
if (n <= 1) return n;
if (memo.has(n)) return memo.get(n);
const result = helper(n-1) + helper(n-2);
memo.set(n, result);
return result;
}
return helper(n);
}滚动数组的关键在于遍历方向必须与依赖方向一致。以 0/1 背包为例,