JS 笔试题 —— 亲密字符串

539 阅读1分钟

题目:

给你两个字符串 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

分析:

  • 情况一: 两个字符串相等,且字符串内部存在重复字符, 例如 abcbabcb,交换重复字符即可。
  • 情况二: 两个字符串只有两处不同,且 s[i]=goal[j];goal[i]=s[j]s[i] = goal[j]; goal[i] = s[j]。也就是说两个字符串除交换位字符外,其余字符相等。有两处以上不同,则不成立。

代码:

/**
 * @param {string} s
 * @param {string} goal
 * @return {boolean}
 */
var buddyStrings = function(s, goal) {
    if (s.length !== goal.length) {
        return false;
    }
    
    if (s === goal) {
        let map = {}
        for (let i = 0; i < s.length; i++) {
            map[s[i]] = map[s[i]] ? ++map[s[i]] : 1;
            if (map[s[i]] > 1) {
                return true;
            }
        }
        return false;
    } else {
        let pos1 = -1, pos2 = -1;
        for (let i = 0; i < s.length; i++) {
            if (s[i] !== goal[i]) {
                if (pos1 === -1) {
                    pos1 = i;
                } else if (pos2 === -1) {
                    pos2 = i;
                } else {
                    return false;
                }
            }
        }
        return (pos2 !== -1 && s[pos1] === goal[pos2] && s[pos2] === goal[pos1]);
    }
};