码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode 077. 组合 dfs

时间:2021-01-05 10:38:45      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:ems   mamicode   bin   位置   lazy   turn   tco   全选   color   

地址 https://leetcode-cn.com/problems/combinations/

给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。

示例:

输入: n = 4, k = 2
输出:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

 

解答 

观察样例发现
当在某一个位置选中一个数后 后面填充的数字必须比它大
这样可以避免选中 1 5 组合后 又出现了 5 1 组合

技术图片

使用DFS 完成上述操作
那么 每个数字都有两种可能 填充或者 略过
如果操作的值超过n 那么就不再继续
如果当前已经填充K个数字了 那么放入答案不再继续

技术图片

class Solution {
public:
vector<vector<int>> ans;

void dfs(int n, int k, int idx, vector<int> v)
{
    if(v.size()>k) return;
    //剩下的元素 全选也不足以达到选择K个元素 剪枝
    if( (v.size() + n-idx+1) < k  ) return;


    if (v.size() == k) {ans.push_back(v);return;}


    v.push_back(idx);
    dfs(n, k, idx + 1, v);
    v.pop_back();

    dfs(n, k, idx + 1, v);
}

vector<vector<int>> combine(int n, int k) {
    vector<int> v;
    dfs(n, k, 1, v);
    return  ans;
}

};

 

 

LeetCode 077. 组合 dfs

标签:ems   mamicode   bin   位置   lazy   turn   tco   全选   color   

原文地址:https://www.cnblogs.com/itdef/p/14218594.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!