欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

oi笔记——抽象的深度优先搜索

程序员文章站 2022-10-04 12:01:22
oi笔记——抽象的深度优先搜索 例题: $N个数中选K个数,选出的和要为sum$ 例题分析: 对于每个点,我们可以按“选”和“不选”进行搜索,如图: 或者01背包求解 求解示例(抽象深搜版代码) 定义: 前面说过,dfs 看起来是运行在图上的搜索算法,而前一节给大家展示的 dfs 过程,我们没有看到 ......

oi笔记——抽象的深度优先搜索

例题:

\(n个数中选k个数,选出的和要为sum\)

例题分析:

对于每个点,我们可以按“选”和“不选”进行搜索,如图:
oi笔记——抽象的深度优先搜索

或者01背包求解

求解示例(抽象深搜版代码)

#include <iostream>
using namespace std;
int n, k, sum, ans;
int a[40];
void dfs(int i,int cnt,int s) {
    if(i==n) {
        if(cnt==k&&s==sum) {
            ans++;
        }
        return;
    }
    dfs(i+1,cnt,s);
    dfs(i+1,cnt+1,s+a[i]);
}
int main() {
    cin >> n >> k >> sum;
    for (int i = 0; i < n; i++) {
        cin >> a[i];
    }
    ans = 0;
    dfs(0,0,0);
    cout<<ans<<endl;
    return 0;
}

定义:
前面说过,dfs 看起来是运行在图上的搜索算法,而前一节给大家展示的 dfs 过程,我们没有看到图的存在,这就是抽象形式的 dfs 的特点。我们可以根据搜索状态构建一张抽象的图,图上的一个顶点就是一个状态,而图上的边就是状态之间的转移关系(进一步搜索或回溯)。虽然 dfs 是在这张抽象的图上进行的,但我们不必把这张图真正地建立出来。