[LeetCode]72. 编辑距离

题目

给你两个单词 word1 和 word2，请你计算出将 word1 转换成 word2 所使用的最少操作数 。

你可以对一个单词进行如下三种操作：

1. 插入一个字符
2. 删除一个字符
3. 替换一个字符

示例 1：

输入：word1 = "horse", word2 = "ros"
输出：3
解释：
horse -> rorse (将 'h' 替换为 'r')
rorse -> rose (删除 'r')
rose -> ros (删除 'e')

示例 2：

输入：word1 = "intention", word2 = "execution"
输出：5
解释：
intention -> inention (删除 't')
inention -> enention (将 'i' 替换为 'e')
enention -> exention (将 'n' 替换为 'x')
exention -> exection (将 'n' 替换为 'c')
exection -> execution (插入 'u')

解题思路

设 dp[i][j] 表示word1的前 i 个字母与word2的前 j 个字母之间的编辑距离，考虑以下几种情况：
1）如果 word1[i] == word2[j]，那么无需进行任何操作，即 dp[i][j] = dp[i-1][j-1]；
2）如果 word1[i] != word2[j]，分三种情况：
2.1）dp[i-1][j]表示word1的前 i-1 个字母与word2的前 j 个字母之间的编辑距离，那么对于word1的第i个字母，我们需要删除它，即 dp[i][j] = dp[i-1][j] + 1。
2.2）dp[i][j-1]表示word1的前 i 个字母与word2的前 j-1 个字母之间的编辑距离，那么对于word2的第j个字母，我们需要在word1的末尾插入一个字母，即 dp[i][j] = dp[i][j-1] + 1。
2.3）dp[i-1][j-1]表示word1的前 i-1 个字母与word2的前 j-1 个字母之间的编辑距离，那么对于word1的第 i 个字母与word2的第 j 个字母，我们需要对word1的第i个字母进行替换操作，即 dp[i][j] = dp[i-1][j-1] + 1。
综上，由于题目要求的是最少的操作数，那么有：
1）若 word1[i] = word2[j]，那么 dp[i][j] = dp[i-1][j-1]；
2）若 word1[i] != word2[j]，那么 dp[i][j] = min(dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + 1
dp数组的初始状态则是一个单词变为空字符串，或者空字符串变为一个单词的过程，此时的编辑距离就是该单词长度。

代码

class Solution:
    def minDistance(self, word1: str, word2: str) -> int:
        m = len(word1)
        n = len(word2)
        # 创建dp数组
        dp = [[0 for i in range(n+1)] for j in range(m+1)] # 之所以加1，是因为字符长度是从0开始算的
        # dp初始化状态
        for i in range(m+1):
            dp[i][0] = i
        for j in range(n+1):
            dp[0][j] = j
        # 遍历dp数组
        for i in range(1, m+1):
            for j in range(1, n+1):
                if word1[i-1] == word2[j-1]: # word中的第一个字母下标是0，所以这里要减1
                    dp[i][j] = dp[i-1][j-1]
                else:
                    dp[i][j] = min(dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + 1
        return dp[-1][-1]