leetcode 859. 亲密字符串(Buddy Strings)

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

 

提示：

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

 

原题链接

 

刷刷水题，找点感觉，年后状态org

class Solution {
    public boolean buddyStrings(String A, String B) {
    	if(A=="" && B=="" || A.length()!=B.length())
    		return false;
    	char[] ch1 = A.toCharArray();
    	char[] ch2 = B.toCharArray();
    	int[] a1 = new int[26];
    	int[] a2 = new int[26];

    	int t = 0;
    	for(int i=0;i<ch1.length;i++) {
    		a1[ch1[i]-'a']++;
    		a2[ch2[i]-'a']++;
    		if(ch1[i]!=ch2[i]) {
    			t++;
    			if(t>2)
    				return false;
    		}
    	}
    	
    	if(t==0) { //相等的特殊处理 aa是true ab是false 说明只要有2个相同字母的交换也可以
        	for(int i=0;i<26;i++)
        		if(a1[i]>=2)
        			return true;
        	return false;
    	}
    		
    	for(int i=0;i<26;i++)
    		if(a1[i]!=a2[i])
    			return false;
    	
    	return t==2;
    }
}

 

官网题解

方法：情况列举

思路

如果 A[i] == B[i]，我们就说 i 是匹配的，否则称 i 是不匹配的。亲密字符串几乎是完全匹配的，因为一次交换只会影响到两个索引。

如果交换 A[i] 和 A[j] 可以证明 A 和 B 是亲密字符串，那么就有 A[i] == B[j] 以及 A[j] == B[i]。 这意味着在 A[i], A[j], B[i], B[j] 这四个自由变量中，只存在两种情况：A[i] == A[j] 或 A[i] != A[j]

算法

让我们来看看这些情况。

在 A[i] == B[i] 的情况下，字符串 A 与 B 相等。因此，如果 A == B，我们应当检查每个索引 i 以寻找具有相同值的两个匹配。

在 A[i] == B[j], (A[i] != A[j]) 的情况下，其余索引是相匹配的。所以如果 A 和 B 只有两个不匹配的索引（记作 i 和 j），我们应该检查并确保等式 A[i] == B[j] 和 A[j] == B[i] 成立。

class Solution {
    public boolean buddyStrings(String A, String B) {
        if (A.length() != B.length()) return false;
        if (A.equals(B)) {
            int[] count = new int[26];
            for (int i = 0; i < A.length(); ++i)
                count[A.charAt(i) - 'a']++;

            for (int c: count)
                if (c > 1) return true;
            return false;
        } else {
            int first = -1, second = -1;
            for (int i = 0; i < A.length(); ++i) {
                if (A.charAt(i) != B.charAt(i)) {
                    if (first == -1)
                        first = i;
                    else if (second == -1)
                        second = i;
                    else
                        return false;
                }
            }

            return (second != -1 && A.charAt(first) == B.charAt(second) &&
                    A.charAt(second) == B.charAt(first));
        }
    }
}

复杂度分析

• 时间复杂度：O(N)，其中 N是 A 和 B 的长度。

• 空间复杂度：O(1)。