Leetcode859-亲密字符串

题目描述

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

示例一

输入： A = “ab”, B = “ba”
输出： true

示例二

输入： A = “ab”, B = “ab”
输出： false

示例三

输入： A = “aa”, B = “aa”
输出： true

示例四

输入： A = “aaaaaaabc”, B = “aaaaaaacb”
输出： true

示例五

输入： A = “”, B = “aa”
输出： false

思路

这道题并不难，只是需要考虑的情况特别多，只要都考虑到就行。

1. 如果两个字符串长度不相等，则返回 false。
2. 如果两个字符串有超过两处位置，其对应字符不一致，返回 false。
3. 如果仅有两处位置字符不相同，则分别判断这两处位置交换后是否相同。
4. 如果仅有一处位置字符不相同，则返回 false。
5. 如果没有位置字符不相同，则根据题意，我们不得不找到两处位置交换。如果字符串中有两处位置字符相同，则返回 true；否则返回 false。

代码

class Solution {
public:
    bool buddyStrings(string A, string B) {
        int n=A.length(),m=B.length();
        if(n!=m) //判断长度是否相同
            return false;
        int flag1=-1,flag2=-1; //用两个标记，因为超过两处不一样，那么A是不可能交换得到B的。
        for(int i=0;i<n;i++)
        {
            if(A[i]!=B[i])
            {
                if(flag1!=-1)
                {
                    if(flag2!=-1)
                       return false;
                    flag2=i;   
                }
                else
                    flag1=i;  
            }
        }
        if(A[flag1]!=B[flag2]||A[flag2]!=B[flag1])  //判断两处交换后，A，B是否相同
            return false;
        if(flag1!=-1&&flag2==-1) //如果只有一处不同，A不可能交换得到B
            return false;
        if(flag1==-1&&flag2==-1)  //如果两个字符串相同。就看看字符串中有没有相同的字母，有就有可能通过交换得到。例如：acdfa于acdfa字符串中有两个a故可以交换得到
        {
            vector<int>a(26,0);
            for(int i=0;i<n;i++)
            {
                a[A[i]-'a']++;
                if(a[A[i]-'a']>=2)
                    return true;
            }
            return false;
        }
        return true; 
    }
};