Python算法-子集问题

46. 全排列

给定一个不含重复数字的数组 nums ,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。

输入:nums = [1,2,3]
输出:[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]

class Solution:
    def permute(self, nums: List[int]) -> List[List[int]]:
        res = []
        path = []
        def backtracing(nums):
            # 终止条件:当路径上的元素数量等于最大值
            if len(path) == len(nums):
                res.append(path[:])
                return

            for i in range(len(nums)):
                if nums[i] not in path:
                    path.append(nums[i])
                    # 递归搜索
                    backtracing(nums)
                    # 撤销选择
                    path.pop()
        backtracing(nums)
        return res
47. 全排列 II

给定一个可包含重复数字的序列 nums ,按任意顺序 返回所有不重复的全排列。

输入:nums = [1,1,2]
输出:
[[1,1,2],
[1,2,1],
[2,1,1]]

# 对于数组中包含重复数字,需要对重复的数字提前进行标注,在访问时及时去重
class Solution:
    res = []
    path = []
    # 回溯算法
    def backtracing(self, nums: List[int], visited: List[bool]):
        # 递归的终止条件
        if len(nums) == len(self.path):
            self.res.append(self.path[:])
            return 

        for i in range(len(nums)):
            # 需要及时去重,如果相同元素之前访问过,则跳过
            if i > 0 and nums[i] == nums[i-1] and not visited[i-1]:
                continue
            # 元素没有访问过
            if not visited[i]:
                # 访问之后需要进行标记
                visited[i] = True
                self.path.append(nums[i])
                self.backtracing(nums, visited)
                self.path.pop()
                # 回溯之后对元素进行恢复
                visited[i] = False

    # 主函数
    def permuteUnique(self, nums: List[int]) -> List[List[int]]:
        # 清除缓存
        self.res.clear()
        self.path.clear()
        # 需要对nums进行排序
        nums.sort()
        # 用于记录是否访问过相同元素
        visited = [False]*len(nums)
        self.backtracing(nums, visited)
        return self.res
90. 子集 II

给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。

示例 1:
输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]

class Solution:
    def backtrace(self, nums, index, res, path):
        # 空数组
        res.append(path[:])
        # 从第0个位置开始,进行深度优先搜索
        for i in range(index, len(nums)):
            if i > index and nums[i] == nums[i-1]:
                continue
            path.append(nums[i])
            self.backtrace(nums, i+1, res, path)
            path.pop()


    def subsetsWithDup(self, nums: List[int]) -> List[List[int]]:
        # 首先对数组排序
        nums.sort()
        res, path = [], []
        self.backtrace(nums, 0, res, path)
        return res
784. 字母大小写全排列

给定一个字符串 s ,通过将字符串 s 中的每个字母转变大小写,我们可以获得一个新的字符串。
返回 所有可能得到的字符串集合 。以 任意顺序 返回输出。

输入:s = "a1b2"
输出:["a1b2", "a1B2", "A1b2", "A1B2"]

# 大小写转换
class Solution:
    def dfs(self, s, i, path, ans):
        # 终止条件
        if i == len(s):
            ans.append(path[:])
            return 

        self.dfs(s, i+1, path+s[i], ans)
        # 当字符为小写字母,转换为大写并添加
        if ord('a') <= ord(s[i]) <= ord('z'):
            self.dfs(s, i+1, path + s[i].upper(), ans)
        # 当字符为大写字母是,转换为小写并添加
        elif ord('A') <= ord(s[i]) <= ord('Z'):
            self.dfs(s, i+1, path + s[i].lower(), ans)

    def letterCasePermutation(self, s: str) -> List[str]:
        ans, path = [], ''
        self.dfs(s, 0, path, ans)
        return ans
39. 组合总和

给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。
candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。
对于给定的输入,保证和为 target 的不同组合数少于 150 个。

输入:candidates = [2,3,6,7], target = 7
输出:[[2,2,3],[7]]
解释:
2 和 3 可以形成一组候选,2 + 2 + 3 = 7 。注意 2 可以使用多次。
7 也是一个候选, 7 = 7 。
仅有这两种组合。

class Solution:
    path = []
    res = []
    def backtracing(self, candidates: List[int], target:int, sum_all:int, start_index:int):
        # 递归的终止条件
        if sum_all > target:
            return
        if sum_all == target:
            self.res.append(self.path[:])
            return 
        
        # 开始遍历查找
        for i in range(start_index, len(candidates)):
            # 循环终止条件
            if sum_all + candidates[i] > target:
                break

            sum_all += candidates[i]
            self.path.append(candidates[i])
            # 回溯
            self.backtracing(candidates, target, sum_all, i)
            sum_all -= candidates[i]
            self.path.pop()

    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        self.path.clear()
        self.res.clear()
        candidates.sort()
        self.backtracing(candidates, target, 0, 0)
        # print(self.res)
        return self.res
40. 组合总和 II

给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用 一次 。
注意:解集不能包含重复的组合。

输入: candidates = [10,1,2,7,6,1,5], target = 8,
输出:
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]

# 本题主要考虑去重
class Solution:
    res = []
    path = []
    def backtracing(self, candidates: List[int], target: int, sum_all: int, start_index: int):
        # 递归的终止条件
        if sum_all > target:
            return 
        if sum_all == target:
            self.res.append(self.path[:])
            return
        # 进行回溯
        for i in range(start_index, len(candidates)):
            # 终止条件
            if sum_all + candidates[i] > target:
                break
            # 去重操作
            if i > start_index and candidates[i] == candidates[i-1]:
                continue

            sum_all += candidates[i]
            self.path.append(candidates[i])
            self.backtracing(candidates, target, sum_all, i+1)
            sum_all -= candidates[i]
            self.path.pop()

    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        self.res.clear()
        self.path.clear()
        # 先进行排序
        candidates.sort()
        self.backtracing(candidates, target, 0, 0)
        return self.res
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,133评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,682评论 3 390
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,784评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,508评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,603评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,607评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,604评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,359评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,805评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,121评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,280评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,959评论 5 339
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,588评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,206评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,442评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,193评论 2 367
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,144评论 2 352

推荐阅读更多精彩内容