【代码随想录】LC 77. 组合

文章目录

  • 前言
  • 一、题目
    • 1、原题链接
    • 2、题目描述
  • 二、解题报告
    • 1、思路分析
    • 2、代码详解

前言

本专栏文章为《代码随想录》书籍的刷题题解以及读书笔记,如有侵权,立即删除。

一、题目

1、原题链接

77. 组合

2、题目描述

【代码随想录】LC 77. 组合_第1张图片

二、解题报告

1、思路分析

(1)利用回溯、搜索算法,每次选取一个数,作为本次结果中的一个数,然后依次向下递归遍历,注意:组合不强调顺序,[1,2]与[2,1]是一种。所以每次需要记录上一次遍历的位置,从而保证下次遍历得到的结果不会与已有结果重复。当遍历到底,本次存储的结果已到达k个数,保存结果,开始回溯。
(2)剪枝优化:当给定数组中待遍历的数已经不足以使本次搜索的结果达到k个数时,可以立即结束此次搜索,节约时间。

2、代码详解

class Solution {
    vector<vector<int>> res;    //存储最终答案
    vector<int> temp;           //存储每次搜索到的路径(即每次的结果)
public:
    //搜索、回溯算法
    //st:每次开始递归时的开始元素
    void dfs(int n, int k, int st) {
        //如果本次路径(结果)中已够k个数,说明已经到底:保存结果,开始回溯
        if (temp.size() == k) {
            res.push_back(temp);
            return;
        }
        for (int i = st; i <= n; i++) {
            //剪枝优化:当剩下的待遍历元素已不足以组成结果的k个数是,直接结束循环
            if (k- temp.size() > n - i + 1) {
                break;
            }
            temp.push_back(i);
            dfs(n, k, i + 1);
            temp.pop_back();
        }
    }
    vector<vector<int>> combine(int n, int k) {
        dfs(n, k, 1);
        return res;
    }

};

你可能感兴趣的:(代码随想录,深度优先,算法,c语言,数据结构,java,c++,python)