Problem P24. [算法课回溯]组合问题

最新推荐文章于 2024-11-12 12:29:51 发布

Ashbur_xyz

最新推荐文章于 2024-11-12 12:29:51 发布

阅读量298

点赞数 8

分类专栏：某211大学算法课文章标签：算法深度优先

本文链接：https://blog.csdn.net/score_1/article/details/136214788

版权

某211大学算法课专栏收录该内容

34 篇文章 3 订阅

订阅专栏

本文介绍了一个使用C++编写的算法教程，通过递归遍历和剪枝优化，展示了如何利用回溯方法解决一个问题。主要内容包括深度优先搜索(DFS)的实现和示例输出。

摘要由CSDN通过智能技术生成

这将会是一个系统性的算法学习专栏，编程语言为C++，适用于刚开始学算法的学生和博友，建议需要的朋友收藏订阅，是免费的，希望对大家能够有所帮助。

在这里插入图片描述

采用递归遍历所有可能性，再使用剪枝减小运行时间，利用回溯，代码有注释

#include<iostream>
#include<bits/stdc++.h>
#include<cstdio>
#include<string>

using namespace std;

vector<vector<int>> result;
vector<int> path;
void dfs(int n, int k, int cur){
    // 递归结束条件
    if (path.size() == k){
        result.push_back(path);
        return;
    }
    // 剪枝， cur是现在遍历到的数值
    if (n-cur+1+path.size()<k){
        return;
    }
    path.push_back(cur);
    // 选择添加cur进组合
    dfs(n, k, cur+1);
    // 回溯
    path.pop_back();
    // 选择不添加cur进组合
    dfs(n, k, cur+1);
    return;
}

int main()
{
    int n, k;
    cin >> n >> k;
    dfs(n, k, 1);
    printf("[");
    for (int i = 0; i < result.size(); i++){
        printf("[");
        vector<int> x = result[i];
        for (int j = 0; j < x.size(); j++){
            printf("%d", x[j]);
            if (j < x.size()-1){
                printf(", ");
            }
        }
        printf("]");
        if (i < result.size()-1){
            printf(", ");
        }
    }
    printf("]");
    return 0;
}