DFS合集---3.允许重复选择元素的组合（剑指offer-81）

最新推荐文章于 2024-03-01 22:25:17 发布

Ego12138

最新推荐文章于 2024-03-01 22:25:17 发布

阅读量1.2k

点赞数

分类专栏：题解文章标签： leetcode 算法

本文链接：https://blog.csdn.net/ego12138/article/details/120813854

版权

题解专栏收录该内容

39 篇文章 0 订阅

订阅专栏

原题链接

题目描述

给定一个无重复元素的正整数数组 candidates 和一个正整数 target ，找出 candidates 中所有可以使数字和为目标数 target 的唯一组合。

candidates 中的数字可以无限制重复被选取。如果至少一个所选数字数量不同，则两种组合是唯一的。

对于给定的输入，保证和为 target 的唯一组合数少于 150 个。

示例1

输入: candidates = [2,3,6,7], target = 7
输出: [[7],[2,2,3]]

示例2

输入: candidates = [2,3,5], target = 8
输出: [[2,2,2,2],[2,3,3],[3,5]]

示例3

输入: candidates = [2], target = 1
输出: []

示例4

输入: candidates = [1], target = 1
输出: [[1]]

示例5

输入: candidates = [1], target = 2
输出: [[1,1]]

思路

这道题和第二题一样，只需要在上一题的基础上改一下结束条件即可。
如果当前加入集合的数字之和sum和target相等，就加入结果集合。
如果sum已经超过target了，就没有必要再向下进行；同样，下标index不能超过数组元素的个数，不然会产生越界异常。所以在满足这两个任意一个条件时，也要返回。

代码

class Solution {
    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        List<List<Integer>> result = new LinkedList();
        LinkedList<Integer> subset = new LinkedList();
        dfs(candidates, target, 0, 0, subset, result);
        return result;
    }

    private void dfs(int[] candidates, int target, int sum, int index, LinkedList<Integer> subset, List<List<Integer>> result){
        if(sum == target){
            result.add(new LinkedList(subset));
        }else if(sum > target || index == candidates.length){
            return;
        }else{
            dfs(candidates, target, sum, index + 1, subset, result);
            subset.add(candidates[index]);
            dfs(candidates, target, sum + candidates[index], index, subset, result);
            subset.removeLast();
        }
    }
}