组合总数二

做~自己de王妃 提交于 2020-01-11 20:12:04

给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用一次。

说明:

所有数字(包括目标数)都是正整数。
解集不能包含重复的组合。 

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]

示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
[1,2,2],
[5]
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution:
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        size=len(candidates)
        if(size==0):
            return []
        candidates.sort()
        result=[]
        self.To_flash_back(candidates,result,target,0,size,[])
        return result

    def To_flash_back(self ,candidates,result,Remainder,start,size,path):
        if(Remainder==0):
            result.append(path[:])
        
        for i in range(start,size):
            if((candidates[i]==candidates[i-1])and (i-1>=start)):
                continue
            Remainder2=Remainder-candidates[i]
            if(Remainder2<0):
                break
            path.append(candidates[i])
            self.To_flash_back(candidates,result,Remainder2,i+1,size,path)
            path.pop()

跟组合总数一相比只要加上一点限制条件就可以了,加一个去重的判断和递归的时候start+1这样就可以使candidates 中的每个数字在每个组合中只能使用一次了。
动态规划
dp的思路由于组合总数的思路
有点相像总的问题在于如何使candidates 中的数字只使用一次,我解决的办法我没有想出来看了题解之后发现他的做法很奇妙,在遍历target的时候他倒着遍历这样就不会出现由 candidates 中的数字出现多次来组成target 的情况。

class Solution:
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        candidates.sort()
        dp=[set() for i in range(target+1)]
        dp[0].add(())
        for j in candidates:
            for k in range(target ,j-1,-1):
                for temp in dp[k-j]:
                    dp[k].add(temp+(j,))
                
        return list(dp[target])
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!