Boostable

  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

LeetCode: Combinations

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],
]

地址:https://oj.leetcode.com/problems/combinations/
算法:递归搞定。$\binom{n}{k}=\binom{n-1}{k}+\binom{n-1}{k}$。代码:
 1 class Solution {
 2 public:
 3     vector<vector<int> > combine(int n, int k) {
 4         if(n < 1 || k < 1)  return vector<vector<int> > ();
 5         return combineCore(n, k);
 6     }
 7     vector<vector<int> > combineCore(int n,int k){
 8         vector<vector<int> > result;
 9         if(k <= 0){
10             result.push_back(vector<int>());
11             return result;
12         }
13         if(n <= k){
14             vector<int> temp;
15             for(int i = 1; i <= n; ++i)
16                 temp.push_back(i);
17             result.push_back(temp);
18             return result;
19         }
20         result = combineCore(n-1, k-1);
21         for(int j = 0; j < result.size(); ++j){
22             result[j].push_back(n);
23         }
24         vector<vector<int> > temp = combineCore(n-1,k);
25         result.insert(result.end(),temp.begin(),temp.end());
26         return result;
27     }
28 };

 

posted on 2014-09-04 20:59  Boostable  阅读(214)  评论(0编辑  收藏  举报