Java教程

77. 组合

本文主要是介绍77. 组合,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

给定两个整数 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);
            });
        }
    }
}
这篇关于77. 组合的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!