给定两个由小写字母构成的字符串 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;
}
}
};