华为OD机试 Python -两个字符串间的最短路径问题

华为OD机试 Python -两个字符串间的最短路径问题

题目描述

给定两个字符串，分别为字符串 A 与字符串 B。例如 A字符串为 “ABCABBA”，B字符串为 “CBABAC” 可以得到下图 m * n 的二维数组，定义原点为(0,0)，终点为(m,n)，水平与垂直的每一条边距离为1，映射成坐标系如下图。从原点 (0,0) 到 (0,A) 为水平边，距离为1，从 (0,A) 到 (A,C) 为垂直边，距离为1；假设两个字符串同一位置的两个字符相同，则可以作一个斜边，如 (A,C) 到 (B,B) 最短距离为斜边，距离同样为1。作出所有的斜边如下图，(0,0) 到 (B,B) 的距离为：1 个水平边 + 1 个垂直边 + 1 个斜边 = 3。

输入描述

空格分割的两个字符串 A 与字符串 B

    字符串不为"空串"
    字符格式满足正则规则：[A-Z]
    字符串长度 < 10000

输出描述

原点到终点的最短距离

样例1

input
ABC ABC
output
3

样例2

input
ABCABBA CBABAC
output
9

代码：python

def minDistance(a, b):
    m, n = len(a), len(b)
    dp = [[float('inf')] * (n + 1) for _ in range(m + 1)]
    dp[0][0] = 0
    for i in range(m + 1):
        for j in range(n + 1):
            if i > 0:
                dp[i][j] = min(dp[i][j], dp[i-1][j] + 1)
            if j > 0:
                dp[i][j] = min(dp[i][j], dp[i][j-1] + 1)
            if i > 0 and j > 0 and a[i-1] == b[j-1]:
                dp[i][j] = min(dp[i][j], dp[i-1][j-1] + 1)
    return dp[m][n]

a, b = input().split()
print(minDistance(a, b))