Leetcode 1024马尔可夫链小游戏暴力运行代码（python）

爱白衣苍狗啊

已于 2022-10-21 10:25:22 修改

阅读量852

点赞数 1

分类专栏： python编程文章标签： leetcode python 算法

于 2022-10-20 10:18:14 首次发布

本文链接：https://blog.csdn.net/qq_53720761/article/details/127421339

版权

python编程专栏收录该内容

1 篇文章 0 订阅

订阅专栏

必要的库itertools

主要思路：利用其内部提供的库函数itertools.permutations()来求所有的组合，依次进行运算，利用set集合自动去重。

代码如下：

import itertools


# 符号运算
def function(a: int, b: int, f: str) -> int:
    if f == "+":
        return a + b
    if f == "-":
        return a - b
    elif f == "*":
        return a * b
    elif f == "//":
        return a // b
    elif f == "**":
        return a ** b
    elif f == "^":
        return a ^ b
    elif f == "|":
        return a | b
    elif f == "&":
        return a & b
    elif f == ">>":
        return a >> b
    elif f == "<<":
        return a << b


# 这里是你有的数字
nums = [2, 2, 2, 2, 6, 6, 9, 13, 13, 14, 34, 1024]
# 这里是你有的符号
f = ['&', '&', '|', '|', '+', '*', '*', '^', '<<']

list_r = list(itertools.permutations(f, 3))
num_r = list(itertools.permutations(nums, 4))
set_res = set()

for element in list_r:
    f1 = element[0]
    f2 = element[1]
    f3 = element[2]
    for num in num_r:
        res1 = function(num[0], num[1], f1)
        res2 = function(res1, num[2], f2)
        res3 = function(res2, num[3], f3)
        if res3 == 1024:
            # print(str(num[0]) + f1 + str(num[1]) + f2 + str(num[2]) + f3 + str(num[3]) + "={}".format(res3))
            string = "(((" + str(num[0]) + f1 + str(num[1]) + ")" + f2 + str(num[2]) + ")" + f3 + str(
                num[3]) + ") = 1024"
            set_res.add(string)

if len(set_res) == 0:
    print("没有合适的组合")
    sys.exit(0)
# 输出，因为是集合自动去重,1024为什么是SSR卡?
for i in set_res:
    print(i)
sys.exit(0)