leetcode77(组合:回溯法)

给定两个整数 n 和 k,返回 1 … n 中所有可能的 k 个数的组合。
题解:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]

这道题与leetcode17(求电话号码的字符组合)实际上解题思路是一致的,其实也可以这么说,在涉及枚举+组合的问题中,回溯法的普适性是很强的。

题解:维护一个栈,保存组合情况。利用回溯法(递归),穷举所有组合情况,当栈中的元素个数为k时,将栈中的结果插入结果集。

class Solution {
    
    
    private final List<List<Integer>>res=new ArrayList<>();
    public List<List<Integer>> combine(int n, int k) {
    
    
        DFS(new boolean[n],new Stack<>(),n,k,1);
        return res;
    }
    private void DFS(boolean[]search,Stack<Integer>subRes,int n,int k,int min){
    
    
        if(subRes.size()+n-min+1<k)
            return;
        if(subRes.size()==k) {
    
    
            res.add(new ArrayList<>(subRes));
            return;
        }
        for(int i=min;i<=n;i++){
    
    
            if(!search[i-1]) {
    
    
                subRes.add(i);
                search[i-1]=true;
                DFS(search,subRes,n,k,i+1);
                search[i-1]=false;
                subRes.pop();
            }
        }
    }
}

猜你喜欢

转载自blog.csdn.net/CY2333333/article/details/108479179
今日推荐