LeetCode 算法:珠玑妙算

426 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第 8 天,点击查看活动详情

珠玑妙算

原题地址

珠玑妙算游戏(the game of master mind)的玩法如下。

计算机有4个槽,每个槽放一个球,颜色可能是红色(R)、黄色(Y)、绿色(G)或蓝色(B)。例如,计算机可能有RGGB 4种(槽1为红色,槽2、3为绿色,槽4为蓝色)。作为用户,你试图猜出颜色组合。打个比方,你可能会猜YRGB。要是猜对某个槽的颜色,则算一次“猜中”;要是只猜对颜色但槽位猜错了,则算一次“伪猜中”。注意,“猜中”不能算入“伪猜中”。

给定一种颜色组合 solution 和一个猜测 guess,编写一个方法,返回猜中和伪猜中的次数 answer,其中answer[0] 为猜中的次数,answer[1] 为伪猜中的次数。

示例1:

输入: solution="RGBY",guess="GGRR"
输出: [1,1]
解释: 猜中1次,伪猜中1次。

示例2:

输入: solution="RYGB", guess="BYGR"
输出: [2,2]
解释: 猜中2次,伪猜中2次。

提示:

  • len(solution) = len(guess) = 4
  • solutionguess 仅包含"R","G","B","Y"这4种字符

思路分析

  1. 按照题目描述中,需要先找出猜中的,然后将其排除,再去寻找伪猜中的,因为题目中说猜中不能算入伪猜中;
  2. 定义两个变量,right 表示猜中的数目,fakeRight 表示伪猜中的数量;
  3. 再定义两个变量,s1 用来存储 solution 中除去猜中元素的其他元素,g1 用来存储 guess 中除去猜中元素的其他元素;
  4. 第一次循环 solution,判断 solution[i]guess[i] 是否相同,若相同,则 right + 1;若不同,则将 solution[i] 存储到 s1 中,把 guess[i] 存储到 g1 中;
  5. 再次循环 s1,若 g1 中含有 s1[i] 那么 fakeRight + 1,并且将 g1 中对应位置的元素删掉,寻找剩余的元素位置;
  6. 上述两步由于 solutionguesslength 相同,因此不需要考虑下标越界的问题;
  7. 经过上述两次循环,即可得到 rightfakeRight,返回即可。

AC 代码

/**
 * @param {string} solution
 * @param {string} guess
 * @return {number[]}
 */
var masterMind = function(solution, guess) {
    let right = 0
    let fakeRight = 0
    let s1 = []
    let g1 = []
    for(let i = 0; i < solution.length; i++) {
        if(solution[i] === guess[i]) {
            right += 1
        } else {
            s1.push(solution[i])
            g1.push(guess[i])
        }
    }
    for(let i = 0; i < s1.length; i++) {
        const index = g1.indexOf(s1[i])
        if(index > -1) {
            fakeRight += 1
            g1.splice(index, 1)
        }
    }
    return [right, fakeRight]
};

结果:

  • 执行结果: 通过
  • 执行用时:48 ms, 在所有 JavaScript 提交中击败了97.37%的用户
  • 内存消耗:41.2 MB, 在所有 JavaScript 提交中击败了34.21%的用户
  • 通过测试用例:60 / 60

END