C++编辑距离问题
编辑距离,也被称为 Levenshtein 距离,指两个字符串之间互相转换的最小修改次数,通常用于在信息检索和自然语言处理中度量两个序列的相似度。
Question
输入两个字符串
你可以在一个字符串中进行三种编辑操作:插入一个字符、删除一个字符、替换字符为任意一个字符。
如图 14-27 所示,将 kitten
转换为 sitting
需要编辑 3 步,包括 2 次替换操作与 1 次添加操作;将 hello
转换为 algo
需要 3 步,包括 2 次替换操作和 1 次删除操作。
图 14-27 编辑距离的示例数据
编辑距离问题可以很自然地用决策树模型来解释。字符串对应树节点,一轮决策(一次编辑操作)对应树的一条边。
如图 14-28 所示,在不限制操作的情况下,每个节点都可以派生出许多条边,每条边对应一种操作,这意味着从 hello
转换到 algo
有许多种可能的路径。
从决策树的角度看,本题的目标是求解节点 hello
和节点 algo
之间的最短路径。
图 14-28 基于决策树模型表示编辑距离问题
1. 动态规划思路
第一步:思考每轮的决策,定义状态,从而得到
每一轮的决策是对字符串
我们希望在编辑操作的过程中,问题的规模逐渐缩小,这样才能构建子问题。设字符串
- 若
和 相同,我们可以跳过它们,直接考虑 和 。 - 若
和 不同,我们需要对 进行一次编辑(插入、删除、替换),使得两字符串尾部的字符相同,从而可以跳过它们,考虑规模更小的问题。
也就是说,我们在字符串
状态
至此,得到一个尺寸为
第二步:找出最优子结构,进而推导出状态转移方程
考虑子问题
- 在
之后添加 ,则剩余子问题 。 - 删除
,则剩余子问题 。 - 将
替换为 ,则剩余子问题 。
图 14-29 编辑距离的状态转移
根据以上分析,可得最优子结构:
请注意,当
第三步:确定边界条件和状态转移顺序
当两字符串都为空时,编辑步数为
观察状态转移方程,解
2. 代码实现
edit_distance.cpp
/* 编辑距离:动态规划 */
int editDistanceDP(string s, string t) {
int n = s.length(), m = t.length();
vector<vector<int>> dp(n + 1, vector<int>(m + 1, 0));
// 状态转移:首行首列
for (int i = 1; i <= n; i++) {
dp[i][0] = i;
}
for (int j = 1; j <= m; j++) {
dp[0][j] = j;
}
// 状态转移:其余行列
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= m; j++) {
if (s[i - 1] == t[j - 1]) {
// 若两字符相等,则直接跳过此两字符
dp[i][j] = dp[i - 1][j - 1];
} else {
// 最少编辑步数 = 插入、删除、替换这三种操作的最少编辑步数 + 1
dp[i][j] = min(min(dp[i][j - 1], dp[i - 1][j]), dp[i - 1][j - 1]) + 1;
}
}
}
return dp[n][m];
}
如图 14-30 所示,编辑距离问题的状态转移过程与背包问题非常类似,都可以看作是填写一个二维网格的过程。
图 14-30 编辑距离的动态规划过程
3. 空间优化
由于
为此,我们可以使用一个变量 leftup
来暂存左上方的解
edit_distance.cpp
/* 编辑距离:空间优化后的动态规划 */
int editDistanceDPComp(string s, string t) {
int n = s.length(), m = t.length();
vector<int> dp(m + 1, 0);
// 状态转移:首行
for (int j = 1; j <= m; j++) {
dp[j] = j;
}
// 状态转移:其余行
for (int i = 1; i <= n; i++) {
// 状态转移:首列
int leftup = dp[0]; // 暂存 dp[i-1, j-1]
dp[0] = i;
// 状态转移:其余列
for (int j = 1; j <= m; j++) {
int temp = dp[j];
if (s[i - 1] == t[j - 1]) {
// 若两字符相等,则直接跳过此两字符
dp[j] = leftup;
} else {
// 最少编辑步数 = 插入、删除、替换这三种操作的最少编辑步数 + 1
dp[j] = min(min(dp[j - 1], dp[j]), leftup) + 1;
}
leftup = temp; // 更新为下一轮的 dp[i-1, j-1]
}
}
return dp[m];
}