给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combinations
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
import java.util.ArrayList; import java.util.LinkedList; import java.util.List; import java.util.Scanner; class Solution { private int n; private int k; private List<List<Integer>> ret = new ArrayList<>(); private LinkedList<Integer> path = new LinkedList<>(); private void solve(int idx) { if (path.size() == k) { ret.add(new ArrayList<>(path)); return; } if (path.size() + (n - idx + 1) < k) { return; } solve(idx + 1); path.offerLast(idx); solve(idx + 1); path.pollLast(); } public List<List<Integer>> combine(int n, int k) { this.n = n; this.k = k; solve(1); return ret; } public static void main(String[] args) { Scanner in = new Scanner(System.in); while (in.hasNext()) { List<List<Integer>> combines = new Solution().combine(in.nextInt(), in.nextInt()); combines.forEach(combine -> { System.out.println("----"); combine.forEach(System.out::println); }); } } }