LeetCode 859. 亲密字符串(Buddy Strings) 23

859. 亲密字符串
859. Buddy Strings

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

每日一算法2019/5/26Day 23LeetCode859. Buddy Strings

示例 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 仅由小写字母构成。
   

解题思路

如果 A.length() != B.length()，不满足交换条件，直接返回 false；

如果 A.equals(B)，我们需要交换两个相同的字符，所以需要检查 A 中是否存在两个相同的字符；

在其他情况下，我们要找到索引 A.charAt(i) != B.charAt(i)。应该只有 2 个差异，这是我们的一次交换。

Java 实现

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

class Solution {
    public boolean buddyStrings(String A, String B) {
        if (A == null || B == null || A.length() != B.length()) {
            return false;
        }
        if (A.equals(B)) {
            Set<Character> set = new HashSet<>();
            for (char c : A.toCharArray()) {
                set.add(c);
            }
            return set.size() < A.length();
        }
        List<Integer> list = new ArrayList<>();
        for (int i = 0; i < A.length(); i++) {
            if (A.charAt(i) != B.charAt(i)) {
                list.add(i);
            }
        }
        return list.size() == 2 && A.charAt(list.get(0)) == B.charAt(list.get(1)) && A.charAt(list.get(1)) == B.charAt(list.get(0));
    }
}

参考资料

• https://leetcode.com/problems/buddy-strings/
• https://leetcode-cn.com/problems/buddy-strings/