每日一题~组合总数III

原题链接: 216. 组合总和 III - 力扣(LeetCode)

题目描述:

思路分析:

这是一个组合的问题,所以我们可以使用深度优先搜索(DFS)的方式将所有的情况都列举出来,然后将其中符合条件的情况添加到列表中,最后返回这个列表就可以了。详细步骤如下:

1、准备一个 tmp 列表保存当前集合的数据,用 list 列表保存所有符合条件的列表,用 sum 来记录当前 tmp 列表中的数据之和,用一个 used 数组来记录该数据是否已经被使用

2、使用 DFS 来找到所有组合,当 tmp 的 size = k 时,说明 tmp 已经满了,如果 sum = n,说明这个 tmp 符合条件,但是需要注意的是【1,2,3】和【1,3,2】是属于同一个集合,所以我们还需要进行去重操作,因为我们是从小到大来搜索的,所以【1,2,3】一定在【1,3,2】的前面,去重的时候我们只需要判断 tmp.get(i+1) 是否小于 tmp.get(i) 就可以了,如果 tmp.get(i+1) < tmp.get(i),则说明 list 中已经保存了该情况,直接返回即可

代码示例:

java 复制代码
class Solution {
    // 标记是否使用
    int[] used = new int[10];
    // 记录链表中的数据和
    int sum = 0;
    List<List<Integer>> list = new ArrayList<>();
    LinkedList<Integer> tmp = new LinkedList<>();
    public List<List<Integer>> combinationSum3(int k, int n) {
        // 递归
        dfs(1,k,n);
        return list;
    }
    public void dfs(int x,int k, int n) {
        if(tmp.size() == k) {
            if(sum == n) {
                // 去重操作
                for(int i = 0; i < k-1; i++) {
                    if(i+1 < k && tmp.get(i) > tmp.get(i+1)) {
                        return;
                    }
                }
                list.add(new ArrayList<>(tmp));
            }
            return;
        }
        for(int i = x; i < 10; i++) {
            if(used[i] == 0) {
                tmp.add(i);
                sum += i;
                used[i] = 1;
                dfs(x+1,k,n);
                tmp.removeLast();
                sum -= i;
                used[i] = 0;
            
            }
            
        }
    }
}

可以看出,上述步骤还是比较麻烦的,我们不仅需要找出来 1~9 所有的组合情况,还需要对这些情况进行一下去重操作,这样一来效率就非常低了。如果我们在寻找组合的时候,就进行剪枝,只保存升序的列表例如,【1,2,3】,【1,2,4】,【1,2,5】,......【7,8,9】,这样的话我们就不需要找到所有的组合情况,而且还节省了去重的步骤,相比第一种暴力的解决方法就显得高效了很多。

java 复制代码
class Solution {
    // 标记是否使用
    int[] used = new int[10];
    // 记录链表中的数据和
    int sum = 0;
    List<List<Integer>> list = new ArrayList<>();
    LinkedList<Integer> tmp = new LinkedList<>();
    public List<List<Integer>> combinationSum3(int k, int n) {
        // 递归
        dfs(1,k,n);
        return list;
    }
    public void dfs(int x,int k, int n) {
        if(tmp.size() == k) {
            if(sum == n) list.add(new ArrayList<>(tmp));            
            return;
        }
        for(int i = x; i < 10; i++) {
            if(used[i] == 0) {
                // 剪枝
                if(tmp.size() > 0 && i < tmp.peekLast()) continue;
                tmp.add(i);
                sum += i;
                used[i] = 1;
                dfs(x+1,k,n);
                tmp.removeLast();
                sum -= i;
                used[i] = 0;
            
            }
            
        }
    }
}
相关推荐
范纹杉想快点毕业几秒前
请创建一个视觉精美、交互流畅的进阶版贪吃蛇游戏
数据库·嵌入式硬件·算法·mongodb·游戏·fpga开发·交互
先知后行。3 分钟前
线程的创建.销毁
开发语言·c++·算法
鱼嘻8 分钟前
西嘎嘎学习 - C++ 继承 - Day 10
开发语言·c++·学习·算法
孤廖14 分钟前
从 “模板” 到 “场景”,用 C++ 磨透拓扑排序的实战逻辑
开发语言·c++·程序人生·算法·贪心算法·动态规划·学习方法
Flash.kkl1 小时前
优先算法——专题十一:字符串
算法
汉克老师7 小时前
第十四届蓝桥杯青少组C++选拔赛[2023.2.12]第二部分编程题(5、机甲战士)
c++·算法·蓝桥杯·01背包·蓝桥杯c++·c++蓝桥杯
Mr_Xuhhh8 小时前
项目需求分析(2)
c++·算法·leetcode·log4j
c++bug9 小时前
六级第一关——下楼梯
算法
Morri39 小时前
[Java恶补day53] 45. 跳跃游戏Ⅱ
java·算法·leetcode
林木辛9 小时前
LeetCode热题 15.三数之和(双指针)
算法·leetcode·双指针