首页 > 代码库 > LeetCode:Combinations
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], ]解题思路: 首先,我们将该问题转换为求{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; } };
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。