地址:
力扣https://leetcode-cn.com/problems/buddy-strings/
题目:
给你两个字符串 s 和 goal ,只要我们可以通过交换 s 中的两个字母得到与 goal 相等的结果,就返回 true ;否则返回 false 。
交换字母的定义是:取两个下标 i 和 j (下标从 0 开始)且满足 i != j ,接着交换 s[i] 和 s[j] 处的字符。
例如,在 "abcd" 中交换下标 0 和下标 2 的元素可以生成 "cbad" 。
示例 1:
输入:s = "ab", goal = "ba"
输出:true
解释:你可以交换 s[0] = 'a' 和 s[1] = 'b' 生成 "ba",此时 s 和 goal 相等。
示例 2:
输入:s = "ab", goal = "ab"
输出:false
解释:你只能交换 s[0] = 'a' 和 s[1] = 'b' 生成 "ba",此时 s 和 goal 不相等。
示例 3:
输入:s = "aa", goal = "aa"
输出:true
解释:你可以交换 s[0] = 'a' 和 s[1] = 'a' 生成 "aa",此时 s 和 goal 相等。
示例 4:
输入:s = "aaaaaaabc", goal = "aaaaaaacb"
输出:true
提示:
1 <= s.length, goal.length <= 2 * 104
s 和 goal 由小写英文字母组成
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/buddy-strings
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
需要考虑的条件比较隐晦:
1. 找到并记录两个不相等字符的 idx,最后通过交换 idx 看字符是否相等进行判断
2. 不过还不够,比如 s="aa", gloa="aa" ,没有不相等的字符,但是我们还是可以交换
3. 还要注意如果两个字符串长度都不相等,自然不用交换,直接 false
那么我们需要分两个部分来处理:
1. 字符串相等,找找看有没有同一个字母出现 >= 2 次,如果有,那么可以在这里面交换
计数数组就能搞定
2. 字符串不相等,需要记录交换的 idx
两个变量分别记录第一次,第二次不相等的 idx;
另外一个变量记录不相等的次数,> 2 直接 false
方法一、
bool buddyStrings(char * s, char * goal){
int p1 = -1, p2 = -1;
int cnt = 0;
int slen = strlen(s);
int glen = strlen(goal);
if(slen != glen)
return false;
// same
if(strcmp(s, goal) == 0)
{
int numArr[26] = {0};
for(int i=0; i<slen; i++)
{
int sidx = s[i] - 'a';
numArr[sidx]++;
if(numArr[sidx] == 2) // swap the two
return true;
}
}
else // different
{
for(int i=0; i<slen; i++)
{
if(s[i] != goal[i])
{
//printf("s[%d]=%c, goal[%d]=%c\n", i, s[i], i, goal[i]);
cnt++;
if(cnt > 2)
return false;
if(p1 == -1)
p1 = i;
else if(p2 == -1)
p2 = i;
}
}
if(cnt == 2 && s[p1] == goal[p2] && s[p2] == goal[p1])
return true;
}
return false;
}