每日一题-面试题 01.05. 一次编辑

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

输入:
first = “pale”
second = “ple”
输出: True

输入:
first = “pales”
second = “pal”
输出: False

class Solution:
    def oneEditAway(self, first: str, second: str) -> bool:
        if abs(len(first) - len(second)) > 1:
            return False
        if first == second:
            return True
        if len(first) < len(second):
            first, second = second, first
        for i in range(len(second)):
            if first[i] == second[i]:
                continue
            return first[i+1:] == second[i:] or first[i+1:] == second[i+1:] 
        return True