3 digit code

傷城~ 2024-04-03 12:10 58阅读 0赞

原题如下图示

![alt][]

密码锁是3位的密码，猜5种组合分别有下面的提示

`7 9 3` ：只有1个对的，而且在正确位置；

`7 2 5`：只有1个对的，但位置不对；

`3 1 7`：有两个数是对的，但都不在正确位置；

`8 4 9` ： 没有包含正确的数字

`8 9 1` ： 只有1个对的，但位置不对；

逻辑任务尝试编程解决。好处是有的，例如修改题面的数字，也能迅速获取密码答案。

初始化arrs二维数组表达所有的输入信息！简单将5个声明解析为数组中的第4位和第5位数值，具体如下：

arrs = [
            [7,9,3,1,1], # n=1,p=1, 包含1个密码数字，且位置对
            [7,2,5,1,0], # n=1,p=0, 包含1个密码数字，但位置不对
            [3,1,7,2,0], # n=2,p=0, 包含2个密码数字，但位置都不对
            [8,4,9,0,0], # n=0,p=0, 没有包含
            [8,9,1,1,0]  # n=1,p=0, 包含1个密码数字，但位置不对
            ]

第一步：筛选0-9之间可能出现在密码里的数字

def filter(arrs):
        dt,rem = set(),set()
        for arr in arrs:
            if arr[3] > 0:
                dt.update(arr[:3])
            if arr[3] == 0:
                rem = set(arr[:3])
                # not good for: dt - set(arr[:3])
        return dt - rem

第二步：在组合中筛选符合任务中 5 个声明的

变量n是包含密码数字的个数,p是位置正确的个数;

只有满足全部5个\#声明的组合返回结果 True，否则返回False

def judge(num,arrs):
        # filter(arrs) < digit = list(range(10))
        #变量n是包含密码数字的个数, p是位置正确的个数;
        #只有满足全部5个声明的组合返回结果 True，否则False
        for arr in arrs:
            n,p = arr[3],arr[4]
            temp = list(zip(num,arr[:3]))
            bool_pos = sum([x==y for x,y in temp])
            common = set(arr[:3]) & set(num)
            if bool_pos != p or len(common) != n:
               return False
        else:return True

第三步：排列组合并逐一判断

from itertools import permutations
    def main(arrs,lngth=3):
        ans = []
        s = filter(arrs)
        seq = list(permutations(s,lngth))
        return [num for num in seq if judge(num,arrs)]
        
    print('main:',main(arrs,3))
    main: [(1, 5, 3)]

扩展任务

你可以修改初始化二维数组，例如：arrs第5行的\[8,9,1\],改为\[8,9,5\],你猜猜结果如何？

arrs = [
            [7,9,3,1,1], # n=1,p=1, 包含1个密码数字，且位置对
            [7,2,5,1,0], # n=1,p=0, 包含1个密码数字，但位置不对
            [3,1,7,2,0], # n=2,p=0, 包含2个密码数字，但位置都不对
            [8,4,9,0,0], # n=0,p=0, 没有包含
            [8,9,5,1,0]  # n=1,p=0, 包含1个密码数字，但位置不对
            ]

运行以上程序，发现结果不变！正确密码还是(1, 5, 3) 推理看是否正确？

扩展之二

保持二维数组元素的前3个元素不变，修改声明，例如第3行\[3,1,7,2,0\]，修改位\[3,1,7,1,0\]，返回结果是空，没有满足的组合；

但如果继续修改第4行，\[8,4,9,0,0\]改为\[8,4,9,1,0\]，再看看有什么变化？ 具体如下初始化arrs

arrs = [
            [7,9,3,1,1], # n=1,p=1, 包含1个密码数字，且位置对
            [7,2,5,1,0], # n=1,p=0, 包含1个密码数字，但位置不对
            [3,1,7,1,0], # n=1,p=0, 包含1个密码数字，但位置都不对
            [8,4,9,1,0], # n=0,p=0, 包含1个密码数字，但位置都不对
            [8,9,1,1,0]  # n=1,p=0, 包含1个密码数字，但位置不对
            ]

返回两个结果：(2, 8, 3), (5, 8, 3)

上述可以继续优化，优化的方向是声明变化时，修改初始化arrs，判断是否输出为空。

本文由[ mdnice ][mdnice]多平台发布

[alt]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/03/a8bdea3b6a6b40ff977acdf319ee8522.png
[mdnice]: https://mdnice.com/?platform=4