leetcode 77 组合(c++)

### 题目

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

示例:

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

就是一个回溯法,然后返回条件改成i<=n-(k-temp.size()),这样可以减少很多计算,因为temp必要要满足长度为k,如果i的长度过大这样i后面的元素并不足以构成一个长度为k的数组,就不用遍历了。

### code

    

class Solution {
public:
void huishu(int k,int n,int begin,vector>&res,vectortemp)
{
    if(temp.size()==k)
    {
            res.push_back(temp);
            return ;
    }
    for(int i=begin;i<=n-(k-temp.size());++i)
    {
        temp.push_back(i+1);
        huishu(k,n,i+1,res,temp);
        temp.pop_back(); 
    }
    return;
}

vector> combine(int n, int k) {
    vector> res;
    vector temp;
    huishu(k,n,0,res,temp);
    return res;
}
};

分享一个别人的解法,这个写法挺有意思的。

首先是找到第一位可以的最小的数字,然后找第二位可以的最小数字,一直到第k位可以的最小数字,然后第k位开始不断增加,直到第k位已经到达最大,然后将第k-1位更改成第二小的数字,然后再依次增加第k位,如果第k-1位也不能增加了,在增加第k-2位,一直到第一位也不能再增加,那么循环结束。

class Solution {
public:
vector> combine(int n, int k) {
    vector> result;
    int i = 0;
    vector p(k, 0);
    while (i >= 0) {
        p[i]++;
        if (p[i] > n) --i;
        else if (i == k - 1) result.push_back(p);
        else {
            ++i;
            p[i] = p[i - 1];
        }
    }
    return result;
}
};

 

你可能感兴趣的:(leetcode 77 组合(c++))