码迷,mamicode.com
首页 > 其他好文 > 详细

Leetcode篇:全排列II

时间:2018-11-16 17:23:17      阅读:158      评论:0      收藏:0      [点我收藏+]

标签:pycha   输入   har   elf   pen   index   跳过   continue   ini   


@author: ZZQ
@software: PyCharm
@file: permuteUnique.py
@time: 2018/11/16 13:34
要求:给定一个可包含重复数字的序列,返回所有不重复的全排列。
示例:
输入: [1,1,2]
输出:
[
[1,1,2],
[1,2,1],
[2,1,1]
]
思路:深搜,然后去掉不满足条件的
去重过程:
1. 每次向下搜索时,都去除掉父节点这个元素
2. 给每个元素做一个标记,来表示当前元素是否被用过,如果被用过,则结束向下搜索;
3. 判断相邻的相同元素知否被用过,如果用过则跳过该元素。

import copy
class Solution():
    def __init__(self):
        tt_ans = []
        pass

    def permuteUnique(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        ans = []
        temp_ans = []
        nums.sort()
        length = len(nums)
        if length == 0 or nums == []:
            return []
        flag = []
        for i in range(length):
            flag.append(0)
        cur_length = 0
        start_index = -1
        self.dfs(temp_ans, start_index, nums, flag, cur_length, length, ans)
        return ans

    def dfs(self, temp_ans,start_index, nums, flag, cur_length, length, ans):
        if cur_length == length:
            tt_ans = copy.deepcopy(temp_ans)
            ans.append(tt_ans)

        else:
            for i in range(length):
                if i > 0 and nums[i] == nums[i - 1] and flag[i - 1] == 1:
                    continue
                if i != start_index and flag[i] == 0:
                    temp_ans.append(nums[i])
                    flag[i] = 1
                    self.dfs(temp_ans, i, nums, flag, cur_length+1, length, ans)
                    temp_ans.pop()
                    flag[i] = 0

Leetcode篇:全排列II

标签:pycha   输入   har   elf   pen   index   跳过   continue   ini   

原文地址:https://www.cnblogs.com/zzq-123456/p/9969854.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!