算法题猜数字-Python实现

今天刷的第一道算法题,先拿一道简单点的试试手,这道题目的要求是:

两个人甲乙在猜数字,甲先从1,2,3三个数字中随机抽3次,结果是guess。乙随后也随机抽三次,结果是answer。然后对比甲乙两个人的结果。示例如下:

guess:[1,2,3], answer: [1, 2, 3]
那么结果就是猜对了3次

guess: [1,2,3], answer: [1, 2, 1]
那么结果就是猜对了2次

guess: [1,2,3], answer:[3, 3,1]
那么结果就是猜对了0次
即将guess和answer两个作为参数输入,返回猜对的次数。

我想出来的几个答案如下所示:

答案1:

class Solution:
    def game(self, guess: List[int], answer: List[int]) -> int:
        count = 0
        for i in zip(guess, answer):
            if i[0] == i[1]:
                count += 1
        return count

思路是:使用zip将两个列表进行组合,返回每个列表中单个元素组成的元组,然后循环对比。如果相等就将临时变量值+1,最终返回统计结果。

答案2:

class Solution:
    def game(self, guess: List[int], answer: List[int]) -> int:
        count = 0
        for i in range(3):
            if guess[i] == answer[i]:
                count+=1
        return count

思路:转念一想,发现自己想复杂了,因为不需要拼接,就可以直接取这两个列表对应的值进行对比。就更简单了,直接循环对比,有相等的情况将临时变量+1,最后返回临时变量即可。

答案3:

class Solution:
    def game(self, guess: List[int], answer: List[int]) -> int:
        return sum(map(lambda x,y: x==y, guess, answer))

思路:既然可以直接对比,那么能不能再简单点,我就想到了map函数,map函数第一个参数直接使用匿名函数来进行元素对比,然后收集函数执行结果。如果对比相等,结果是True,对比不相等,结果是False。

最后返回map函数直接结果的总和即可。

答案4:

class Solution:
    def game(self, guess: List[int], answer: List[int]) -> int:
        if guess == answer:
            return 3
        elif guess[0] == answer[0]:
            if guess[1] == answer[1]:
                return 2
            else:
                return 1
        elif guess[1] == answer[1]:
            if guess[2] == answer[2]:
                return 2
            else:
                return 1
        elif guess[2] == answer[2]:
            return 1
        else:
            return 0

思路:将guess和answer进行对比,这是我认为自己最差的一个答案,因为这种情况只适合这种简短的列表对比,再长一点整个代码就没法看了。

以上就是我的4个答案,有没有更好的方式可以来实现呢?有想法的同学可以给我留言。

你可能感兴趣的:(算法题猜数字-Python实现)