python回溯法解决排列与组合问题

     回溯法解决排列与组合问题

组合问题：

    给定两个整数 n 和 k，返回 1 … n 中所有可能的 k 个数的组合，例如：
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]

# 对应leetcode 77
class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:

        nums = [i for i in range(1,n+1)]

        res = []
        def backTrace(start, tmpNums ):
            if len(tmpNums) == k:
                res.append(tmpNums[:])
                return 
            
            for i in range(start, n):
                tmpNums.append(nums[i])
                backTrace(i+1, tmpNums)
                tmpNums.pop()

        if n== 0 or k==0 or k>n:
            return 
        
        if k==n:
            return [nums]

        backTrace(0, [])

        return res

排列问题：

    给定一个 没有重复 数字的序列，返回其所有可能的全排列，例如：
输入: [1,2,3]
输出:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]

# 对应leetcode 46
class Solution:
    def permute(self, nums: List[int]) -> List[List[int]]:
        
        res = []
        length = len(nums)
        if length == 0:
            return 

        def bactTrace( state, tmpRes, depth):
            if len(tmpRes) == length:
                res.append(tmpRes[:])
                return 

            for i in range(length):
                if state[i] == 1:
                    tmpRes.append(nums[i])
                    state[i] = 0
                    bactTrace(state, tmpRes, depth+1)
                    state[i] = 1
                    tmpRes.pop()
        
        bactTrace([1]*length, [], 0)

        return res