面试题 01.05. 一次编辑

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

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

// 编辑距离的方法

class Solution {
public:
    bool oneEditAway(string first, string second) {
		int len1 = first.length();
		int len2 = second.length();
		if(len1 <= 0) {
			if(len2 <= 1)
				return true;
			else
				return false;
		}

		if(len2 <= 0) {
			if(len1 <= 1)
				return true;
			else
				return false;
		}
		
		vector<vector<int>> dp(len1+1, vector<int>(len2+1, 0));
		for(int i = 1; i <= len1; i++) {
			dp[i][0] = i;
		}

		for(int j = 1; j <= len2; j++) {
			dp[0][j] = j;
		}
		
		for(int i = 1; i <= len1; i++) {
			for(int j = 1; j <= len2; j++) {
				if(first[i-1] == second[j-1])
					dp[i][j] = dp[i-1][j-1];
				else { 
					dp[i][j] = min(dp[i-1][j], min(dp[i][j-1], dp[i-1][j-1])) + 1;
				}
			}
		}
		
		// 这个一定要写外面
		if(dp[len1][len2] > 1)
			return false;
		return true;
    }
};

// 观察法
// len1 - len2 > 1 || len2 - len1 > 1
// len1 == len2 最多允许一次元素更改
// len1 - len2 == 1 || len2 - len1 == 1 只需要短的子串+1个字符

class Solution {
public:
    bool oneEditAway(string first, string second) {
		int len1 = first.length();
		int len2 = second.length();
		
		if(len1 - len2 > 1 || len2 - len1 > 1) {
			return false;
		} else if (len1 == len2) {
			// 两个字符串相等或者只需要替换某一位的字符串，才能实现相等
			int count = 0;
			int start = 0;
			while(start < len1) {
				if(first[start] != second[start])
					count++;
				if(count > 1)
					return false;
				start++;
			}
			return true;
		} else {
			// 给len2添加1位
			int start1 = 0;
			int start2 = 0;
			int count = 0;
			while(start1 < len1 && start2 < len2) {
				if(first[start1] != second[start2]) {
					count++;
					if(count == 1) {
						if(len1 - len2 == 1)
							start1++;
						else
							start2++;
					}
					
					if(count > 1)
						return false;
				} else {
					start1++;
					start2++;
				}
			}
			return true;
		}
	}
};