【程序员面试金典】面试题05:一次编辑

面试题05:一次编辑

题目描述

字符串有三种编辑操作:插入一个字符、删除一个字符或者替换一个字符。 给定两个字符串，编写一个函数判定它们是否只需要一次(或者零次)编辑。

示例 1:

输入: 
first = "pale"
second = "ple"
输出: True

示例 2:

输入: 
first = "pales"
second = "pal"
输出: False

code

# 字符串有三种编辑操作：插入一个字符，删除一个字符或者替换一个字符
# 给定两个字符串，编写一个函数判断他们是否只需要一次（或者零次）编辑


class Solution:
    def oneEditAway(self, first: str, second: str) -> bool:
        if abs(len(first) - len(second)) > 1:
            return False

        # 长度将短的放在前面--交换字符串
        if len(first) > len(second):
            first, second = second, first
        i,j = 0, 0
        while i < len(first) and j < len(second):
            if first[i] == second[j]:
                i += 1 # 下标向后移动一个位置
                j += 1
            else:
                # 3种情况，1，first增加一个字母，2，first替换一个字母，（3.删除和增加一样） 
                return first[i:] == second[j+1:] or first[i+1:] == second[j+1:]# 增加和删除效果一样
        return True


#插入，删除，替换
if __name__ == '__main__':
    s  = Solution()
    print(s.oneEditAway("pale", "ple"))