【Lintcode】640. One Edit Distance

题目地址：

https://www.lintcode.com/problem/one-edit-distance/description

给定两个单词，允许在一个单词中执行插入一个字符、删除一个字符以及修改一个字符的操作，如果能使得两个字符串变相同，则返回true，否则返回false。

如果两个字符串长度相差$2$或以上，直接返回false；如果长度一样，那必然只能有一个位置上的字符不同，验证之；否则，遍历两个字符串，当有不同字符的时候，长的那个字符串删掉该字符必须与短的相同。代码如下：

public class Solution {
    /**
     * @param s: a string
     * @param t: a string
     * @return: true if they are both one edit distance apart or false
     */
    public boolean isOneEditDistance(String s, String t) {
        // write your code here
        if (Math.abs(s.length() - t.length()) >= 2) {
            return false;
        }
        
        // 长度相等，则不同字符个数必须是1
        if (s.length() == t.length()) {
            int diff = 0;
            for (int i = 0; i < s.length(); i++) {
                if (s.charAt(i) != t.charAt(i)) {
                    diff++;
                }
            }
            
            return diff == 1;
        }
        
        // 长度差1，则找到第一个不同的位置，长的字符串从后面一位截取出来的子串必须等于短的从当前位置截取出来的子串
        for (int i = 0; i < Math.min(s.length(), t.length()); i++) {
            if (s.charAt(i) != t.charAt(i)) {
                if (s.length() < t.length()) {
                    return s.substring(i).equals(t.substring(i + 1));
                } else {
                    return s.substring(i + 1).equals(t.substring(i));
                }
            }
        }
        
        return true;
    }
}

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