1. 从“最短路径”到“多阶段决策”多段图问题的本质在算法和运筹学的世界里动态规划Dynamic Programming, DP常常被初学者视为一座难以逾越的高山。大家一提到DP脑海里可能立刻蹦出“01背包”、“最长公共子序列”或者“接雨水”这些经典例题。确实这些题目是理解DP思想的绝佳入口它们教会我们如何将大问题分解为重叠子问题并通过记忆化Memoization或制表法Tabulation来避免重复计算。但今天我想聊一个同样经典却在教学和面试中相对“低调”的模型——多段图问题。多段图问题乍一听名字可能有点抽象但它解决的问题场景却非常直观在一个有向图中如何找到从起点到终点的最短或最长路径并且这个图被明确地划分成了若干个连续的“阶段”。这里的“阶段”是关键。想象一下你要规划一个从北京到广州的行程但行程被强制要求分为几个阶段第一阶段必须从北京出发到达华北的某个城市如石家庄、天津第二阶段必须从华北的某个城市到达华中的某个城市如郑州、武汉以此类推直到最后阶段到达广州。你不能“跳级”比如从北京直接飞到长沙。这就是一个典型的多段图决策过程。为什么这个问题值得单独拿出来说因为它完美地体现了动态规划中“阶段”和“状态”这两个核心概念。在“接雨水”问题中状态可能是每个柱子的左边最高和右边最高值在“01背包”中状态是“考虑前i个物品背包容量为j时的最大价值”。而在多段图问题中状态的定义异常清晰dp[stage][node]表示到达第stage阶段、位于节点node时的最优累积代价如最短距离、最小花费。这种结构化的状态定义使得状态转移方程也格外规整几乎就是“看图说话”。很多朋友在刷了上百道DP题目后反而对最基础的模型感到模糊。多段图问题就像一块试金石它能帮你厘清DP中“阶段决策”的本质。如果你能清晰地解决一个多段图问题那么你对DP状态设计的理解会上一个台阶。接下来我们就从零开始拆解这个问题的每一个环节。2. 问题定义与形式化建模把现实问题装进“图”里在动手写代码之前我们必须把问题用数学和计算机能理解的语言描述清楚。一个标准的多段图问题通常包含以下几个要素图结构一个带权有向图G (V, E)其中V是顶点集合E是边集合每条边(u, v)都有一个权值w(u, v)代表从u到v的代价如距离、时间、成本。阶段划分顶点集合V被划分成k个互不相交的子集V1, V2, ..., Vk。这意味着每个顶点只属于一个阶段。源点与汇点V1中只包含一个顶点称为源点sVk中也只包含一个顶点称为汇点t。边约束所有的边(u, v)都满足一个关键性质如果u属于阶段Vi那么v必须属于下一个阶段Vi1。也就是说边只能从前一个阶段指向后一个相邻的阶段不能跨阶段更不能反向。这保证了整个决策过程是单向、分步推进的。我们的目标是找到一条从源点s到汇点t的路径使得路径上所有边的权值之和最小或最大。如何将现实问题建模成多段图举个例子假设我们要为一个项目做资源分配规划项目分为需求分析阶段1、设计阶段2、开发阶段3、测试阶段4四个阶段。每个阶段都有多种实施方案对应每个阶段的多个节点不同方案的成本不同并且从一个阶段的某个方案切换到下一个阶段的某个方案会产生额外的衔接成本对应边的权值。我们的目标就是以最小的总成本完成项目。这就可以用一个4段图来建模。再比如网络数据传输中选择不同层级的中继节点生产线上不同工序的机器选择都可以抽象成多段图。关键在于识别出决策的“阶段”和每个阶段的“可选状态节点”。形式化定义 设dp[i][j]表示从源点s出发到达第i阶段中第j个节点假设我们对每个阶段的节点有内部编号所需的最小累积代价。 我们的状态转移方程可以非常直观地写出来dp[i][j] min_{p in Pre(j)} { dp[i-1][p] w(p, j) }其中Pre(j)表示在第i-1阶段中所有有边指向当前节点j的节点集合。w(p, j)是从节点p到节点j的边权值。这个方程的含义是要到达当前节点j我们必须从上一个阶段的某个节点p过来那么总代价就是“到达p的最小代价”加上“从p到j的代价”。我们遍历所有可能的p选择总代价最小的那条路径。基础的定义看似简单但这里面有几个关键的建模细节直接影响到后续算法的实现复杂度和效率。2.1 图的存储结构选择邻接矩阵 vs 邻接表这是一个经典的权衡。对于多段图由于其特殊的结构我们通常知道边只存在于相邻阶段之间。邻接矩阵如果每个阶段的节点数不多且图比较“稠密”即很多节点之间都有边使用二维数组matrix[u][v] w会很方便。查询任意两点间是否有边及权值的时间复杂度是O(1)。但是当图“稀疏”时会浪费大量空间存储不存在的边值为无穷大或0。邻接表这是更通用和节省空间的选择。我们可以用一个列表的列表或字典来存储。例如adj_list[u]存储一个列表里面是所有从u出发的边(v, w)。在状态转移时要计算dp[i][j]我们需要遍历所有可能的前驱节点p。如果使用邻接表我们需要知道哪些节点指向j这需要构建一个“逆邻接表”来高效查询Pre(j)。或者在动态规划递推时我们换一种思路正向递推。即用dp[i-1][p]去更新所有p的后继节点j的dp[i][j]值。这样只需要原邻接表即可无需逆邻接表。实操心得在竞赛或面试中如果节点总数N不大比如几百用邻接矩阵写起来快不易出错。但在工程或节点数较多时邻接表是首选。对于多段图我个人的习惯是存储阶段信息。用一个二维列表stages[k]其中stages[i]存储第i1阶段的所有节点编号。再配合一个邻接表adj只存储从stages[i]中的节点到stages[i1]中节点的边。这样逻辑非常清晰。2.2 “虚拟节点”处理边界情况源点s和汇点t的处理有时需要一点技巧。严格来说s没有前驱节点t没有后继节点。在初始化dp数组时dp[0][s] 0假设阶段从0开始编号s在第0阶段对于第0阶段的其他节点如果存在dp[0][other] INF无穷大表示不可达。最终答案就是dp[k-1][t]。有时候为了方便我们可能会引入“虚拟源点”和“虚拟汇点”特别是当问题描述中的起点和终点不属于任何阶段或者有多个可能的起点/终点时。但在标准多段图定义下我们按上述方式处理即可。3. 算法核心动态规划递推与路径还原理解了模型我们就可以动手实现算法了。整个过程分为三步初始化、递推计算、路径还原。3.1 动态规划递推过程详解我们以一个具体的例子来贯穿整个讲解。假设有一个4段图节点划分如下阶段0 (V0): {0} (源点 s0)阶段1 (V1): {1, 2, 3}阶段2 (V2): {4, 5, 6}阶段3 (V3): {7} (汇点 t7)边及其权值距离如下(0-1):9, (0-2):7, (0-3):3(1-4):2, (1-5):4, (2-4):2, (2-5):2, (2-6):7, (3-5):11, (3-6):5(4-7):11, (5-7):8, (6-7):6我们的目标是求从节点0到节点7的最短路径。第一步初始化DP表我们创建一个二维数组dp[阶段数][节点数]或者更高效地用一个字典dp[node]来存储到达每个节点的最小代价同时用另一个数组stage[node]记录每个节点所属的阶段。 初始化所有dp[node] INF一个很大的数如float(inf)。 设置dp[s] 0。 同时为了最后还原路径我们需要一个pre[node]数组记录到达node的最优路径中它的前驱节点是哪个。第二步按阶段递推这是动态规划的主循环。我们按阶段顺序从第1阶段i1开始遍历到最后一个阶段ik-1。 对于当前阶段i的每一个节点v遍历所有可能的前驱节点u即所有存在边(u, v)且u属于阶段i-1的节点。计算候选值candidate dp[u] w(u, v)。如果candidate dp[v]则更新dp[v] candidate并记录pre[v] u。这个过程本质上是在做松弛操作和Bellman-Ford算法的思想有相通之处但因为我们有严格的阶段限制所以不需要像Bellman-Ford那样进行V-1轮对所有边的松弛只需要按照阶段数k-1轮每轮只处理从前一阶段指向当前阶段的边即可效率更高。用例子手动演算一下初始化dp[0]0, 其他为INF。阶段1节点1,2,3对于节点1前驱只有0dp[1] dp[0]99,pre[1]0对于节点2dp[2] dp[0]77,pre[2]0对于节点3dp[3] dp[0]33,pre[3]0阶段2节点4,5,6节点4可能前驱是1(9211)和2(729)。取最小9dp[4]9,pre[4]2节点5可能前驱是1(9413)、2(729)、3(31114)。取最小9dp[5]9,pre[5]2节点6可能前驱是2(7714)、3(358)。取最小8dp[6]8,pre[6]3阶段3节点7节点7可能前驱是4(91120)、5(9817)、6(8614)。取最小14dp[7]14,pre[7]6最终最短路径代价为14。3.2 路径还原从结果回溯决策链得到最优值很重要但知道具体怎么走的往往更重要。这就是pre数组的作用。它是一个最简单的“决策链”记录。 从终点t开始根据pre[t]找到它的前驱节点再找前驱的前驱一直回溯到源点s就得到了逆序的路径。对于我们的例子pre[7]6-pre[6]3-pre[3]0所以逆序路径是7 - 6 - 3 - 0反转后得到最短路径0 - 3 - 6 - 7总代价为35614。代码实现要点def reconstruct_path(pre, t): path [] while t is not None: # 假设pre[s] None path.append(t) t pre[t] return path[::-1] # 反转列表避坑提示在初始化pre数组时记得将源点s的前驱设为None或一个特殊值如-1作为回溯的终止条件。否则回溯时会陷入死循环或索引错误。4. 时空复杂度分析与优化思路对于一个有k个阶段的多段图假设第i个阶段有n_i个节点总节点数N sum(n_i)。边通常只存在于相邻阶段之间。时间复杂度动态规划的过程需要遍历每个阶段k-1次循环。对于每个阶段i的每个节点v我们需要检查所有可能的前驱u。最坏情况下如果每个阶段节点都全连接那么检查所有边E。因此时间复杂度为O(N E)其中E是边的总数。这比通用的最短路径算法Dijkstra的O((NE)logN)在某些情况下更优因为它利用了图的特殊结构无需优先队列。空间复杂度主要是存储dp数组和pre数组都是O(N)。如果使用邻接表存储图空间复杂度为O(NE)。优化思路滚动数组这是动态规划常见的空间优化技巧。注意到在计算dp[i][v]时我们只依赖于dp[i-1][...]。因此我们不需要保存所有阶段的dp值只需要保存当前阶段和上一个阶段的两个一维数组即可。空间复杂度可以从O(k*M)M为最大阶段节点数降为O(M)。# 伪代码示例 prev_dp [INF] * N curr_dp [INF] * N prev_dp[s] 0 for stage in range(1, k): curr_dp.fill(INF) # 初始化当前阶段 for v in stages[stage]: for u, w in inverse_adj[v]: # v的前驱节点u和边权w if prev_dp[u] w curr_dp[v]: curr_dp[v] prev_dp[u] w pre[v] u # 路径记录仍需全局或特殊处理 # 交换准备下一轮 prev_dp, curr_dp curr_dp, prev_dp但要注意使用滚动数组时路径还原数组pre的记录会变得稍微麻烦因为pre是在不断被覆盖的。一种方法是仍然使用全局的pre数组在更新curr_dp[v]时同步更新pre[v]。由于动态规划的无后效性当前决策只依赖前一阶段这样记录是可行的。并行计算潜力由于每个阶段内部不同节点v的dp[i][v]计算是相互独立的它们都只依赖于上一阶段的dp值因此理论上可以对一个阶段内的所有节点进行并行计算。这在阶段内节点数很多时可以利用多核或GPU加速。剪枝如果某些边的权值非常大或者某些节点明显不可能成为最优路径的一部分可以在构建图或计算过程中提前剔除减少计算量。但这通常需要根据具体问题设计启发式规则。5. 从理论到实践解决“接雨水”与“编辑距离”的另一种视角你可能会问多段图模型和“接雨水”、“编辑距离”这些经典DP问题有什么关系其实很多动态规划问题都可以隐式地建模成一个多段图决策过程。理解这一点能让你对DP有更统一的认识。以“接雨水”问题为例 问题描述给定一个非负整数数组height表示每个柱子的高度计算按此排列的柱子下雨之后能接多少雨水。 传统的DP解法是计算每个位置i的左边最高柱子left_max[i]和右边最高柱子right_max[i]然后每个位置能接的水量是min(left_max[i], right_max[i]) - height[i]。多段图视角 我们可以把计算每个位置i的接水量看作一个“阶段”。但更贴切的类比是将“确定每个柱子的最终水面高度”看作一个决策过程。不过这个问题更经典的图论模型是“单调栈”其阶段特性不如编辑距离明显。再以“编辑距离”问题为例 问题描述给定两个单词word1和word2计算将word1转换成word2所需的最少操作数插入、删除、替换一个字符。 这是一个非常典型的多段图问题阶段我们可以把处理word1的前i个字符看作第i个阶段。总共有len(word1)1个阶段包括处理0个字符的阶段。状态每个阶段的状态是word2已经匹配了j个字符。所以状态可以定义为dp[i][j]将word1的前i个字符转换为word2的前j个字符所需的最少操作数。决策在阶段i即考虑word1的前i个字符为了达到状态jword2的前j个字符我们有三种“边”可以走删除从状态(i-1, j)过来代价为1。相当于删掉word1的第i个字符。插入从状态(i, j-1)过来代价为1。相当于在word1中插入一个字符匹配word2的第j个字符。替换或匹配从状态(i-1, j-1)过来。如果word1[i-1] word2[j-1]代价为0匹配否则代价为1替换。源点与汇点源点是dp[0][0]两个空字符串汇点是dp[m][n]完整的word1转成完整的word2。这样一看编辑距离的DP表格其计算顺序通常从左到右、从上到下正好对应了在多段图中按阶段推进。每个dp[i][j]的值都只依赖于“左上”、“左”、“上”这三个方向的状态这正是因为“边”只存在于相邻“阶段”的状态之间这里的阶段是i和j的组合推进。深度思考将编辑距离理解为多段图最大的好处是强化了“决策阶段”的概念。每一次操作保持、替换、插入、删除都是一次向下一阶段的转移。这种视角下状态转移方程不再是死记硬背的公式而是对图中三条可能入边的代价比较非常直观。6. 常见误区与实战调试技巧即使理解了原理在实现多段图DP时依然有几个坑容易踩进去。误区一阶段划分错误这是最根本的错误。必须确保图中的所有边都严格从阶段i指向阶段i1。如果在建模时有一条边从阶段i指向了阶段i2或更后或者指回了阶段i-1那么标准的按阶段递推算法将失效。你需要重新审视问题看是否能通过增加虚拟节点或调整阶段定义来满足条件。如果不行那么这个问题可能不是标准的多段图问题可能需要更一般的动态规划或图算法如Dijkstra。误区二DP数组初始化不当源点初始化务必正确初始化dp[s] 0。我见过有人忘记初始化或者错误地将其初始化为1或其他值。不可达状态对于其他节点必须初始化为一个“无穷大”值表示初始时不可达。在Python中可以用float(inf)在Java/C中可以用一个很大的整数如Integer.MAX_VALUE / 2防止后续加法溢出。关键点在更新dp[v] min(dp[v], dp[u] w)时如果dp[u]是无穷大加上w可能溢出在浮点数里是inf在整数里可能变成负数。所以选择INF时要足够大但INF w不能溢出成负数。误区三路径还原时pre数组记录错误未记录源点前驱导致回溯时无法终止。确保pre[s] -1或None。在滚动数组优化中丢失路径如前所述如果使用滚动数组pre数组的更新必须与dp值的更新严格同步。当dp[v]被更新时pre[v]必须同时更新为当前的前驱u。因为dp数组被复用了但pre记录的是全局最优路径的前驱不能被覆盖后再用于错误的前驱。调试技巧小数据手工模拟像我们上面做的那样画一个简单的多段图手动演算DP表和pre数组。这是验证算法逻辑最有效的方法。打印DP表在代码中每完成一个阶段的递推就打印出当前阶段的dp值。与你的手工计算结果对比可以快速定位是哪个阶段、哪个节点的计算出了错。检查边界特别注意第一个阶段i1和最后一个阶段的计算。第一个阶段的前驱只有源点最后一个阶段的目标是汇点。验证路径算出最短路径代价后一定要用pre数组还原出路径并手动计算这条路径的总权值看是否与dp[t]相等。如果不相等说明pre记录的逻辑有误。7. 变种与扩展当问题不那么“标准”现实世界的问题不会总是乖乖地套用标准模型。多段图问题有几个常见的变种1. 最长路径问题如果边的权值代表收益我们想找最大收益的路径。算法完全一样只需把状态转移方程中的min改为max并把不可达状态的初始化从INF改为-INF或一个很小的数即可。需要注意的是如果图中存在正权环但在多段图中由于边只能向前所以不可能有环最长路径问题可能无界但多段图是无环的所以没问题。2. 顶点带权标准问题是边带权。如果顶点也有权值比如到达某个城市需要交入城费如何处理很简单可以将顶点权值合并到边上。有两种方法出边合并将节点u的权值cost_u加到所有从u出发的边(u, v)的权值上。入边合并将节点v的权值cost_v加到所有进入v的边(u, v)的权值上。 通常选择一种统一的方式即可。注意源点s的权值可能需要特殊处理在初始化dp[s]时加上或者单独加在最终结果上。3. 多源点多汇点标准定义是单源单汇。如果问题有多个可能的起点和终点一个常用的技巧是超级源点创建一个虚拟的超级源点S从S到所有真实起点的边权值为0或起点本身的代价。然后以S作为算法起点。超级汇点创建一个虚拟的超级汇点T从所有真实终点到T的边权值为0。最终答案就是dp[T]。 这样就把问题转化回了单源单汇的标准形式。4. 资源约束的多段图如带容量限制这是更复杂的变种例如在每一段选择路径不仅考虑距离还考虑资源消耗如资金、时间并且总资源有限。这就变成了一个“多段图上的资源约束最短路径问题”通常需要引入多维动态规划。状态变量需要增加一维或多维来表示剩余资源量例如dp[i][j][r]表示到达阶段i节点j时剩余资源为r的最优值。状态转移时除了考虑前驱节点还要确保资源消耗不超过限制。这类问题的复杂度会显著增加。理解这些变种能让你在面对新问题时快速判断其是否属于多段图家族并选择合适的解决方案或进行适当的变形。多段图模型的价值不仅在于解决那一类特定问题更在于它提供了一种清晰的“分阶段决策”的思维框架这是动态规划乃至更广泛的优化问题求解中非常宝贵的思维方式。