leetcode:77. 组合(java 深搜(dfs)+剪枝)

package LeetCode;

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

/*
给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]
 */
public class Combine {
    public List> combine(int n, int k) {
        List> res=new ArrayList<>();
        List r=new ArrayList<>();
        if (n<=0||k<=0||k>n)
            return res;
            combimedfs(n,k,1,r,res);
            return res;
    }
    public void combimedfs(int n,int k,int i,List r,List> res){
            if (r.size()==k){
                res.add(new ArrayList(r));
                return;
            }
        for (int j = i; j <=n ; j++) {
                //当剩下的循环不能拼够r的list时直接剪枝
            if ((n-j)<(k-(r.size()+1)))return;
            r.add(j);
            combimedfs(n,k,j+1,r,res);
            r.remove(r.size()-1);
        }
    }

    public static void main(String[] args) {
        Combine c=new Combine();
        System.out.println(c.combine(1,1));
    }
}

 

你可能感兴趣的:(算法试题(java))