• 【LeetCode-回溯】组合总和 II


    题目描述

    给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
    candidates 中的每个数字在每个组合中只能使用一次。
    说明:

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

    示例:

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

    题目链接: https://leetcode-cn.com/problems/combination-sum-ii/
    做这题之前可以先做一下组合总和

    思路

    使用回溯。这个题的思路总体和组合总和的思路是一样的,这道题多了限制条件,第一个是每个数字只能用一次,所以递归的其实位置要是当前位置的下一个位置;另一个问题是数组中有重复的数字,这样最终的结果中也会有重复的答案,所以要去重。去重的方法是对数组进行排序,然后找和起始位置不同的数字开始递归,类似于全排列II。具体代码如下:

    class Solution {
    public:
        vector<vector<int>> ans;
        vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
            if(candidates.empty() || target<=0) return {{}}; 
    
            vector<int> track;
            int curSum = 0;
            int start = 0;
            sort(candidates.begin(), candidates.end()); // 对数组排序
            dfs(candidates, start, curSum, target, track);
            return ans;
        }
    
        void dfs(vector<int> candidates, int start, int curSum, int target, vector<int> track){
            if(curSum==target){
                ans.push_back(track);
                return;
            }
            if(curSum>target) return;
    
            for(int i=start; i<candidates.size(); i++){
                if(candidates[i]>target) continue;  // 当前数字大于target,因为数组升序,所以不可能组合得到target
                if(i>start && candidates[i]==candidates[i-1]) continue; // 去重
                curSum += candidates[i];
                track.push_back(candidates[i]);
                dfs(candidates, i+1, curSum, target, track);
                curSum -= candidates[i];
                track.pop_back();
            }
        }
    };
    

    总结

    一般题目中要求“答案中不能包含重复的数字”,通常的做法是对数组进行排序,然后在添加元素到答案中时添加一些条件。

  • 相关阅读:
    怎样理解 C++ 11中的move语义
    面试---计算机网络
    TCP报头中的选项字段
    协程的实现之调度器
    CPU缓存
    TLS---线程局部存储
    微信libco协程设计及实现---总结
    libco源码解析---协程运行与基本结构
    2014年第五届蓝桥杯B组(C/C++)预赛题目及个人答案(欢迎指正)
    并查集(模板&典型例题整理)
  • 原文地址:https://www.cnblogs.com/flix/p/12773853.html
Copyright © 2020-2023  润新知