LeetCode:Combinations

标签：leetcode   位运算   二进制   

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

解题思路:

    首先，我们将该问题转换为求{1,2,3,....n}所包含的所有大小为k的子集.一般关于组合问题

中的n值都不会太大，所以我选择用一个int变量的二进制位用来标记某个元素时候出现，然后通过

从0枚举到(1 << n) - 1，即可知道其所有的子集，由于子集大小为k，故需判断一下当前枚举

值中1的个数时候为k.这种方法显然是可行的，不过时间复杂度有点高O(2^n)，那么，我们能不能枚举

时，就只枚举元素大小为k的子集？通过使用位运算我们可以很容易的做到这点.

解题代码:
class Solution {
public:
    vector<vector<int> > combine(int n, int k) 
    {
        vector<vector<int> > res;
        int comb = (1 << k) - 1;
        while (comb < 1 << n)
        {
            int tmp = comb, cnt = 1;
            vector<int> sub;
            while (tmp)
            {
                if (tmp & 1)
                    sub.push_back(cnt);
                ++cnt;
                tmp >>= 1;
            }
            res.push_back(sub);
            //关键代码
            int x = comb & -comb, y = comb + x;
            comb = ((comb & ~y) / x >> 1) | y;
        }
        return res;
    }
};

LeetCode:Combinations,布布扣,bubuko.com

LeetCode:Combinations

标签：leetcode   位运算   二进制   

原文地址：http://blog.csdn.net/dream_you_to_life/article/details/34481901