题目链接
https://leetcode-cn.com/problems/buddy-strings
题目
给定两个由小写字母构成的字符串 A 和 B ,只要我们可以通过交换 A 中的两个字母得到与 B 相等的结果,就返回 true ;否则返回 false 。
交换字母的定义是取两个下标 i 和 j (下标从 0 开始),只要 i!=j 就交换 A[i] 和 A[j] 处的字符。例如,在 "abcd" 中交换下标 0 和下标 2 的元素可以生成 "cbad" 。
示例
示例 1:
输入: A = "ab", B = "ba"
输出: true
解释: 你可以交换 A[0] = 'a' 和 A[1] = 'b' 生成 "ba",此时 A 和 B 相等。示例 2:
输入: A = "ab", B = "ab"
输出: false
解释: 你只能交换 A[0] = 'a' 和 A[1] = 'b' 生成 "ba",此时 A 和 B 不相等。示例 3:
输入: A = "aa", B = "aa"
输出: true
解释: 你可以交换 A[0] = 'a' 和 A[1] = 'a' 生成 "aa",此时 A 和 B 相等。示例 4:
输入: A = "aaaaaaabc", B = "aaaaaaacb"
输出: true示例 5:
输入: A = "", B = "aa"
输出: false
提示
0 <= A.length <= 20000
0 <= B.length <= 20000
A
和B
仅由小写字母构成。
思路
本题难点应该主要在于对各种可能的情况需要考虑周全。
1. 对于字符串A和B,如果两个字符串为空,即长度为0,那首先就可以判断false。
2.如果两个字符串长度不一致,也可以直接排除,无论怎么交换也不可能完全相等。
3.下面我们考虑两个字符串长度一致的情况:
1)同时遍历两个字符串,统计两个字符串中不等的元素的个数(如aa、ab)不等的个数为1.
2)因为只能交换一次,所以不等的字符的个数为1、3、4、5、6....的可以直接判断false,因为无论怎么交换,也不可能交换一次,只交换两个元素就可以把两个字符串变成一样。
3)对于两个字符串字符不等的元素个数为0的情况,由于题目要求必须进行一次交换,我们还需要进一步判断字符串中是否有出现过两次及以上的元素。例如,对于字符串aa和aa来说,答案为true(下标位置0和下标位置1的元素互相交换),而对于字符串ab和ab来说,虽然本身两个字符串相等,但必须进行一次交换,只能ab变成ba,交换后则无法相等。因此,只要有其中一个元素出现过两次及以上,我们就可以让这个元素和另一个位置上相同的这个元素进行交换。这里,我们通过构建一个哈希表来统计字符串中每个字符出现的次数。
4)最后一种情况是两个字符串字符不等的元素个数为2,这里我们也还需要进一步判断这两个不同的元素是否是一样的,例如对于abcdd和abcee两个字符串,abcaa中的dd无法交换成abcee中的ee,因为本身abcdd字符串中和abcee字符串中不相等的两个元素dd和ee就不一样,只有像abcde和abced这样,两个字符串中所有元素的个数一致,相当于只是顺序不同,可以通过交换顺序,de变成ed这样才为true,否则也为false。这里我们直接用sort函数,对两个字符串分别进行排序,进行比较即可。
C++ Code
class Solution {
public:
bool buddyStrings(string s, string goal) {
if( (s.size()!=goal.size()) ||s.size()==0) return false;
//两字符串长度为0或者长度不一肯定错误
int count=0;
for(int i=0;i<s.size();i++) //统计字符串A和B中不等的元素个数
{
if(s[i]!=goal[i]) count++;
}
if(count == 2)//如果有两个字符不同则判断这两个不同的字符是否一致 避免"abcaa""abcbb"
{
sort(s.begin(),s.end());
sort(goal.begin(),goal.end());
return s==goal;
}
else if( count == 0)
//两个字符串全等则还要进一步判断 字符串中是否有元素存在两个以上 排除aa这种情况
{
unordered_map<char, int> map; //构建哈希表统计每个字符出现的次数
for(char ch:s)
{
map[ch]++;
}
int flag=0;
for(auto& ch: map)
{
if (ch.second>1) flag=1;
}
return flag;
}
else return false; //不等的字符个数为1、3、4、5...肯定错误
}
};
/*
int main()
{
Solution S;
string s = "aa";
string goal = "aa";
bool a=S.buddyStrings(s,goal);
cout<<a<<endl;
system("pause");
return 0;
}
*/
结果