亲密字符串（每日一题）

给定两个由小写字母构成的字符串 A 和 B ，只要我们可以通过交换 A 中的两个字母得到与 B 相等的结果，就返回 true ；否则返回 false 。

 

示例 1：

输入： A = "ab", B = "ba"
输出： true

示例 2：

输入： A = "ab", B = "ab"
输出： false

示例 3:

输入： A = "aa", B = "aa"
输出： true

示例 4：

输入： A = "aaaaaaabc", B = "aaaaaaacb"
输出： true

示例 5：

输入： A = "", B = "aa"
输出： false

 

提示：

    0 <= A.length <= 20000
    0 <= B.length <= 20000
    A 和 B 仅由小写字母构成。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/buddy-strings
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

阅读理解题，说什么做什么

class Solution {
public:
    bool buddyStrings(string A, string B) {
    if (A.size() == 0 || B.size() == 0 || A.size() != B.size())
	{
		return false;
	}
	int i, j;
	for (i = 0; i < A.size(); i++)
	{
		if (A[i] != B[i])
			break;
	}
	if (i == A.size())
	{
		for (i = 0; i < A.size(); i++)
		{
			for (j = i + 1; j < A.size(); j++)
			{
				if (A[j] == A[i])
				{
					return true;
				}
			}
		}
		return false;
	}
	for (j = i+1; j < A.size(); j++)
	{
		if (A[j] != B[j])
			break;
	}
	swap(A[i], A[j]);
	for (i = 0; i < A.size(); i++)
	{
		if (A[i] != B[i])
			break;
	}
	if (i == A.size() )
	{
		return true;
	}
	else
	{
		return false;
	}
    }
};

上面暴力的方法，感觉时间效率太低了，需要优化，给出下面一个方法，用空间换时间的方法。

class Solution {
public:
    bool buddyStrings(string A, string B) {
     if (A.size() == 0 || B.size() == 0 || A.size() != B.size())
	{
		return false;
	}
	bool k = false;
	int a[26] = {0};
	int q = -1, j=-1;
	int c = 0;
	for (int i = 0; i < A.size(); i++)
	{
		if (A[i] != B[i])
		{
			if (c == 0)
			{
				q = i;
				c++;
			}
			else if (c == 1)
			{
				j = i;
				c++;
			}
			else
				return false;
		}
		a[A[i] - 'a']++;
		if (a[A[i] - 'a'] >= 2)
		{
			k = true;
		}
	}
	if (q == -1 && k)
	{
		return true;
	}
	else if(q==-1)
	{
		return false;
	}
	if (c==2)
	{
		return A[q] == B[j] && A[j] == B[q];
	}
	else
	{
		return false;
	}
    }
};