zoukankan      html  css  js  c++  java
  • [Leetcode] Combination Sum II

    Given a collection of candidate numbers (C) and a target number (T), find all unique combinations in C where the candidate numbers sums to T.

    Each number in C may only be used once in the combination.

    Note:

    • All numbers (including target) will be positive integers.
    • Elements in a combination (a1, a2, … , ak) must be in non-descending order. (ie, a1 ≤ a2 ≤ … ≤ ak).
    • The solution set must not contain duplicate combinations.

    For example, given candidate set 10,1,2,7,6,1,5 and target 8
    A solution set is: 
    [1, 7] 
    [1, 2, 5] 
    [2, 6] 
    [1, 1, 6] 

    记录当前idx之前的元素在结果中是否被使用,如果之前的与当前元素相等的元素没有被使用的话,那么这个元素也不应该被使用,这样就可以去重了。比如[1, 1, 2],如果第一个1没有被使用,那么第二个1也不能使用。

     1 class Solution {
     2 public:
     3     void findNext(vector<int> &num, int target, vector<vector<int> > &res, vector<int> &v, int idx, int sum, vector<bool> &flag) {
     4         if (sum > target || idx > num.size()) return;
     5         if (sum == target) {
     6             res.push_back(v);
     7             return;
     8         }
     9         bool fflag = false;
    10         for (int i = idx-1; i >= 0; --i) {
    11             if (num[i] == num[idx] && !flag[i]) {
    12                 fflag = true;
    13                 break;
    14             };
    15             if (num[i] != num[idx]) break;
    16         }
    17         if (!fflag) {
    18         v.push_back(num[idx]);
    19             flag[idx] = true;
    20             findNext(num, target, res, v, idx+1, sum+num[idx], flag);
    21             flag[idx] = false;
    22             v.pop_back();
    23         }
    24         findNext(num, target, res, v, idx+1, sum, flag);
    25     }
    26     
    27     vector<vector<int> > combinationSum2(vector<int> &num, int target) {
    28         vector<vector<int> > res;
    29         vector<bool> flag(num.size(), false);
    30         vector<int> v;
    31         sort(num.begin(), num.end());
    32         findNext(num, target, res, v, 0, 0, flag);
    33         return res;
    34     }
    35 };

    直接找的话会有重复的答案,可以先把结果存在一个set里,最然把结果从set转存到vector里。代码如下。

     1 class Solution {
     2 public:
     3     void findNext(vector<int> &num, int target, set<vector<int> > &res, vector<int> &v, int idx, int sum) {
     4         if (sum > target || idx > num.size()) return;
     5         if (sum == target) {
     6             res.insert(v);
     7             return;
     8         }
     9         v.push_back(num[idx]);
    10         findNext(num, target, res, v, idx+1, sum+num[idx]);
    11         v.pop_back();
    12         findNext(num, target, res, v, idx+1, sum);
    13     }
    14     
    15     vector<vector<int> > combinationSum2(vector<int> &num, int target) {
    16         set<vector<int> > res;
    17         vector<int> v;
    18         sort(num.begin(), num.end());
    19         findNext(num, target, res, v, 0, 0);
    20         vector<vector<int> > ret;
    21         for (set<vector<int> >::iterator i = res.begin(); i != res.end(); ++i) {
    22             ret.push_back(*i);
    23         }
    24         return ret;
    25     }
    26 };
  • 相关阅读:
    DS博客作业06--图
    DS博客作业05--树
    DS博客作业03--栈和队列
    DS博客作业02--线性表
    DS博客作业01--日期抽象数据类型设计与实现
    C语言博客作业06--结构体&文件
    c语言博客作业05--指针
    C语言博客作业03--函数
    DS博客作业--课程总结
    DS博客作业07--查找
  • 原文地址:https://www.cnblogs.com/easonliu/p/4343042.html
Copyright © 2011-2022 走看看