面试题 01.05. 一次编辑

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

示例 1:

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

示例 2:

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

class Solution {
public:
    bool oneEditAway(string first, string second) {
        int len1 = first.size(), len2 = second.size();
        if(abs(len1 - len2) > 1){
            return false;
        }
        if(first == second){
            return true;
        }
        int i = 0, j = 0;
        int count = 0;
        while(i < len1 && j < len2){
            if(first[i++] == second[j++]){
               
                continue;

            }
            ++count;
            if(count > 1){
                return false;
            }

            
            if(len1 > len2){
                --j;
            } else if(len1 < len2){
                --i;
            }
            
            
        }
        return true;

    }

    
};