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

[LeetCode] Combinations

时间:2018-01-25 23:10:54      阅读:194      评论:0      收藏:0      [点我收藏+]

标签:数字   vector   div   int   避免   ++   数组   ==   第一个   

Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.

For example,
If n = 4 and k = 2, a solution is:

[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

 给定n和k,从1~n中选择k个数作为组合数,并输出。

这是一个回溯问题,重点在于构造回溯辅助函数。
对于n=4,k=2,在1、2、3、4中选择2个数字,
举个例子,选定第一个数1,这时k=1,只需要再选择一个数即可。将选择后的组合放入数组。
当k=0时表面以及选择完毕,当k<0表示返回该数组。引入一个idx变量来避免重复选择。

class Solution {
public:
    vector<vector<int>> combine(int n, int k) {
        vector<vector<int>> res;
        vector<int> tmp;
        int idx = 1;
        helper(n, k, idx, tmp, res);
        return res;
    }
    
    void helper(int n, int k, int idx, vector<int>& tmp, vector<vector<int>>& res) {
        if (k < 0) {
            return;
        }
        else if (k == 0) {
            res.push_back(tmp);
        }
        else {
            for (int i = idx; i <= n; i++) {
                tmp.push_back(i);
                helper(n, k - 1, i + 1, tmp, res);
                tmp.pop_back();
            }
        }
    }
};
// 82 ms

 

[LeetCode] Combinations

标签:数字   vector   div   int   避免   ++   数组   ==   第一个   

原文地址:https://www.cnblogs.com/immjc/p/8353348.html

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