题目描述(困难)
给定两个单词 word1 和 word2,计算出将 word1 转换成 word2 所使用的最少操作数 。
你可以对一个单词进行如下三种操作:
- 插入一个字符
- 删除一个字符
- 替换一个字符
示例 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二维数组存储word1的前 i 个字符和Word2的前 j 个字符达成匹配的步数。
边界条件:dp[i][0]=0:m,dp[0][j]=0:n,其中,m和n分别表示word1和word2的字符数。
状态转移方程是:
如果word[i]==word[j],则当前达到匹配的步数dp[i+1][j+1]=dp[i][j]
如果word[i] word[j],则需要在word1进行添加,删除,替换的一种操作。选择当前位置之前已经累计最少的次数,然后加上当前的操作数 1 即可。
- 如果是添加操作,则先统计word1的前 i 个字符匹配word2的前 j+1 个字符(因为word1添了一个字符)的方法数。即add=dp[i][j+1];
- 如果是删除操作,则先统计word1的前 i +1个字符匹配word2的前 j 个字符(因为word1删了一个字符)的方法数。即delete=dp[i+1][j];
- 如果是替换操作,则统计word1的前 i 个字符匹配word2的前 j 个字符的方法数。即replace=dp[i][j];
最后,dp[i+1][j+1]=min(add,delete,replace)+1
代码
class Solution:
def minDistance(self, word1, word2):
"""
:type word1: str
:type word2: str
:rtype: int
"""
m = len(word1)
n = len(word2)
dp = [[0] * (n + 1) for _ in range(m + 1)]
for i in range(m + 1):
dp[i][0] = i
for i in range(n + 1):
dp[0][i] = i
for i in range(m):
for j in range(n):
if word1[i] == word2[j]:
dp[i + 1][j + 1] = dp[i][j]
else:
add = dp[i][j + 1]
delete = dp[i + 1][j]
replace = dp[i][j]
dp[i + 1][j + 1] = min(add, delete, replace) + 1
return dp[m][n]