珠玑妙算

2020-06-19  本文已影响0人  WAI_f

题目:

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

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

示例:

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

解题方法:

简单介绍一下解题思路:

  • 因为只用4中颜色字母,所以用一个数组存储每个字母出现的次数;
  • 统计每个字母出现次数较小的次数,得到猜中和伪猜中的次数;
  • 再次遍历两个数组,比较相同位置相同颜色的次数,得到猜中的次数;
  • 用总次数减去猜中次数,就是伪猜中次数。

代码和结果:

class Solution {
public:
    vector<int> masterMind(string solution, string guess) {
        int a[4]={0};
        int b[4]={0};
        for(int i=0;i<4;i++)
        {
            switch(solution[i])
            {
                case 'R': 
                a[0]++;
                break;
                case 'Y': 
                a[1]++;
                break;
                case 'G': 
                a[2]++;
                break;
                case 'B': 
                a[3]++;
                break;
            }
            switch(guess[i])
            {
                case 'R': 
                b[0]++;
                break;
                case 'Y': 
                b[1]++;
                break;
                case 'G': 
                b[2]++;
                break;
                case 'B': 
                b[3]++;
                break;
            }
        }
        int total=0;
        int right=0;
        for(int i=0;i<4;i++)
        {
            total+=a[i]>b[i]?b[i]:a[i];
            right+=solution[i]==guess[i];
        }

        return vector<int>{right,total-right};
    }
};
运行结果:

这是我刷的第一百题,恭喜自己,虽然刷的都是简单题,但是能够坚持下来,还是小有成就感,要加倍努力,后面的路更加艰难!

原题链接:https://leetcode-cn.com/problems/master-mind-lcci/

上一篇 下一篇

猜你喜欢

热点阅读