【Leetcode】77. 组合

题目描述：

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

示例:

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

思路：

枚举集合 {1, 2, 3, ..., n} 元素个数为2的所有子集，仍然可用二进制枚举法。

代码：

class Solution {
    public List> combine(int n, int k) {
        List> res = new ArrayList<>();
        if (n < 0 || k < 0 || n < k) {
            return res;
        }
        for (int i = 0; i < (1 << n); i++) {
            List list = new ArrayList<>();
            for (int j = 0; j < n; j++) {
                if ((i & (1 << j)) != 0) {
                    list.add(j + 1);   
                }
                if (list.size() > k) {  // 子集元素大于k，显然不满足
                    break;
                }
            }
            if (list.size() == k) {
                res.add(list);
            }
        }
        return res;
    }
}