DFS专题训练1

1_1:AcWing 93. 递归实现组合型枚举(dfs)

题目链接
从 1∼n 这 n 个整数中随机选出 m 个,输出所有可能的选择方案。

输入格式
两个整数 n,m ,在同一行用空格隔开。

输出格式
按照从小到大的顺序输出所有方案,每行 1 个。

首先,同一行内的数升序排列,相邻两个数用一个空格隔开。

其次,对于两个不同的行,对应下标的数一一比较,字典序较小的排在前面(例如 1 3 5 7 排在 1 3 6 8 前面)。

数据范围
n>0 ,
0≤m≤n ,
n+(n−m)≤25

输入样例:

5 3

输出样例:

1 2 3 
1 2 4 
1 2 5 
1 3 4 
1 3 5 
1 4 5 
2 3 4 
2 3 5 
2 4 5 
3 4 5 

思考题:如果要求使用非递归方法,该怎么做呢?

解析:为了得到一个字典序,只需要从最小的1为根开始往下面搜索即可

#include <iostream>
#include <vector>

using namespace std;

const int N = 30;

int n, m;
bool st[N];
vector<int>ans;

void dfs(int u)
{
    
    
    if(ans.size() == m)
    {
    
    
        for (auto p : ans)
            cout << p << ' ';
            
        cout << endl;
        return ;
    }
    for (int i = u; i <= n; i ++ )
    {
    
    
        if(!st[i])
        {
    
    
            st[i] = true;
            ans.push_back(i);
            dfs(i);
            ans.pop_back();
            st[i] = false;
        }
    }
}

int main()
{
    
    
    cin >> n >> m;
    
    dfs(1);
    
    return 0;
}

1_2:AcWing 129. 火车进栈(dfs + stack)

题目链接
这里有 n 列火车将要进站再出站,但是,每列火车只有 1 节,那就是车头。

这 n 列火车按 1 到 n 的顺序从东方左转进站,这个车站是南北方向的,它虽然无限长,只可惜是一个死胡同,而且站台只有一条股道,火车只能倒着从西方出去,而且每列火车必须进站,先进后出。

也就是说这个火车站其实就相当于一个栈,每次可以让右侧头火车进栈,或者让栈顶火车出站。

车站示意如图:

        出站<——    <——进站
                 |车|
                 |站|
                 |__|

现在请你按《字典序》输出前 20 种可能的出栈方案。

输入格式
输入一个整数 n,代表火车数量。

输出格式
按照《字典序》输出前 20 种答案,每行一种,不要空格。

数据范围
1≤n≤20

输入样例:

3

输出样例:

123
132
213
231
321


解析:
在这里插入图片描述

#include <iostream>
#include <vector>
#include <stack>

using namespace std;

int n;
vector<int>ans;
stack<int>st;
int u = 1, cnt = 20;

void dfs()
{
    
    
    if(!cnt) return ; //  方案最多输出20种
    
    if(ans.size() == n)
    {
    
    
        cnt --;
        for (auto x : ans)
            cout << x;
            
        cout << endl;
        return ;
    }
    if(st.size()) // 栈里面不空
    {
    
    
        ans.push_back(st.top());
        st.pop();
        dfs();
        st.push(ans.back());
        ans.pop_back();
    }
    if(u <= n)
    {
    
    
        st.push(u);
        u ++;
        dfs();
        st.pop();
        u --;
    }
}
int main()
{
    
    
    cin >> n;
    
    dfs();
    
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_52331221/article/details/127657041
今日推荐