Question

161. 相隔为 1 的编辑距离

English Version

题目描述

给定两个字符串 s 和 t ，如果它们的编辑距离为 1 ，则返回 true ，否则返回 false 。

字符串 s 和字符串 t 之间满足编辑距离等于 1 有三种可能的情形：

• 往 s 中插入 恰好一个 字符得到 t
• 从 s 中删除 恰好一个 字符得到 t
• 在 s 中用 一个不同的字符 替换 恰好一个 字符得到 t

 

示例 1：

输入: _s_ = "ab", _t_ = "acb"
输出: true
解释: 可以将 'c' 插入字符串 _s_ 来得到 _t_。

示例 2:

输入: _s_ = "cab", _t_ = "ad"
输出: false
解释: 无法通过 1 步操作使 _s_ 变为 _t_。

 

提示:

• 0 <= s.length, t.length <= 104
• s 和 t 由小写字母，大写字母和数字组成

解法

方法一：分情况讨论

记 $m$ 表示字符串 $s$ 的长度，$n$ 表示字符串 $t$ 的长度。我们可以假定 $m$ 恒大于等于 $n$。

若 $m-n\gt1$，直接返回 false；

否则，遍历 $s$ 和 $t$，若遇到 $s[i]$ 不等于 $t[i]$：

• 若 $m \neq n$，比较 $s[i+1:]$ 与 $t[i:]$，相等返回 true，否则返回 false；
• 若 $m = n$，比较 $s[i:]$ 与 $t[i:]$，相等返回 true，否则返回 false。

遍历结束，说明遍历过的 $s$ 跟 $t$ 所有字符相等，此时需要满足 $m=n+1$。

时间复杂度 $O(m)$，空间复杂度 $O(1)$。

class Solution:
  def isOneEditDistance(self, s: str, t: str) -> bool:
    if len(s) < len(t):
      return self.isOneEditDistance(t, s)
    m, n = len(s), len(t)
    if m - n > 1:
      return False
    for i, c in enumerate(t):
      if c != s[i]:
        return s[i + 1 :] == t[i + 1 :] if m == n else s[i + 1 :] == t[i:]
    return m == n + 1

class Solution {
  public boolean isOneEditDistance(String s, String t) {
    int m = s.length(), n = t.length();
    if (m < n) {
      return isOneEditDistance(t, s);
    }
    if (m - n > 1) {
      return false;
    }
    for (int i = 0; i < n; ++i) {
      if (s.charAt(i) != t.charAt(i)) {
        if (m == n) {
          return s.substring(i + 1).equals(t.substring(i + 1));
        }
        return s.substring(i + 1).equals(t.substring(i));
      }
    }
    return m == n + 1;
  }
}

class Solution {
public:
  bool isOneEditDistance(string s, string t) {
    int m = s.size(), n = t.size();
    if (m < n) return isOneEditDistance(t, s);
    if (m - n > 1) return false;
    for (int i = 0; i < n; ++i) {
      if (s[i] != t[i]) {
        if (m == n) return s.substr(i + 1) == t.substr(i + 1);
        return s.substr(i + 1) == t.substr(i);
      }
    }
    return m == n + 1;
  }
};

func isOneEditDistance(s string, t string) bool {
  m, n := len(s), len(t)
  if m < n {
    return isOneEditDistance(t, s)
  }
  if m-n > 1 {
    return false
  }
  for i := range t {
    if s[i] != t[i] {
      if m == n {
        return s[i+1:] == t[i+1:]
      }
      return s[i+1:] == t[i:]
    }
  }
  return m == n+1
}