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

文章目录

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

前言

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

一、题目

1、原题链接

77. 组合

2、题目描述

二、解题报告

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;
    }

};