LeetCode(77):组合 Combinations(Java)

2019.12.1 LeetCode 从零单刷个人笔记整理(持续更新)

github:https://github.com/ChopinXBP/LeetCode-Babel

这题可以很容易想到用剪枝回溯法求解,不过还有一种字典序的思路:

1.将数字序列nums初始化为[1…k, n+1],一共k+1个元素,最后一位n+1用于作为哨兵。

2.循环将nums从最小排列[1…k, n+1]更新至最大排列[n-k…n-1, n+1]。每次添加前k个元素至结果数组(除去末尾哨兵元素)。

3.更新nums:遍历nums找出第一个与下一个数不连续的数字下标idx,将其数字加1(nums[idx]++);idx之前的元素与其下标i一致(nums[i]=i+1)。

4.当num更新至最大排列[n-k…n-1, n+1]时,数组连续,idx=k跳出循环。


传送门:组合

Given two integers n and k, return all possible combinations of k numbers out of 1 … n.

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

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


import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 *
 * Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.
 * 给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。
 *
 */

public class Combinations {

    //回溯法
    public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> result = new ArrayList<>();
        if(k > n){
            return result;
        }
        for(int i = 1; i <= n - k + 1; i++){
            LinkedList<Integer> list = new LinkedList<>();
            list.add(i);
            Solution(n, k - 1, list, result);
            list.pollLast();
        }
        return result;
    }

    private void Solution(int n, int k, LinkedList<Integer> list, List<List<Integer>> result){
        if(k == 0){
            result.add(new LinkedList<>(list));
            return;
        }
        for(int i = list.getLast() + 1; i <= n; i++){
            list.add(i);
            Solution(n, k - 1, list, result);
            list.pollLast();
        }
    }

    //字典序
    public List<List<Integer>> combine2(int n, int k) {
        //将数字序列nums初始化为[1…k, n+1],一共k+1个元素,最后一位n+1用于作为哨兵
        ArrayList<Integer> nums = new ArrayList<Integer>();
        for(int i = 1; i <= k; i++){
            nums.add(i);
        }
        nums.add(n + 1);

        List<List<Integer>> result = new ArrayList<>();
        //循环将nums从最小排列[1…k, n+1]更新至最大排列[n-k…n-1, n+1]
        int idx = 0;
        while(idx < k){
            //每次添加前k个元素至结果数组(除去末尾哨兵元素)
            result.add(new ArrayList<>(nums.subList(0, k)));
            //遍历nums找出第一个与下一个数不连续的数字下标idx,将其数字加1(nums[idx]++);idx之前的元素与其下标i一致(nums[i]=i+1)
            //当num更新至最大排列[n-k…n-1, n+1]时,数组连续,idx=k跳出循环
            idx = 0;
            while(idx < k && nums.get(idx + 1) == nums.get(idx) + 1){
                nums.set(idx, idx + 1);
                idx++;
            }
            nums.set(idx, nums.get(idx) + 1);
        }
        return result;
    }
}




#Coding一小时,Copying一秒钟。留个言点个赞呗,谢谢你#

扫描二维码关注公众号,回复: 8567385 查看本文章
发布了246 篇原创文章 · 获赞 316 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/qq_20304723/article/details/103333845