0
点赞
收藏
分享

微信扫一扫

算法笔记深搜

八卦城的酒 2022-03-21 阅读 55
算法

问题 E: 【递归入门】出栈序列统计

实际是一种栈操作,如图

在这里插入图片描述

条件分析:用in、out分别表示入队队列和出队队列

  • 当in为空时,栈内的只能出去,为终止条件,序列已经确定
  • 当栈为空,必须入栈
  • 当栈不空
    • 出栈
    • 或者入栈
#include<iostream>
#include<algorithm>
#include<math.h>
#include <cstring>
#define ll long long
#define pa pair<int,int>
using namespace std;
int cnt=0;
void dfs(int in_num,int out_num){
    if(out_num==0){
        cnt++;
        return;
    }
    if(in_num==0){
        dfs(in_num+1,out_num-1);
    }
    else{
        dfs(in_num+1,out_num-1);
        dfs(in_num-1,out_num);
    }
    return;
}
int main(){
    int n;
    while (cin>>n)
    {
        cnt=0;
        dfs(0,n);
        cout<<cnt<<"\n";
    }
    return 0;
}

问题 C: 【递归入门】组合+判断素数

思路:模板题目思路,深搜

// 组合质数判断
int n;
int r;
int ans;
vector<int> v;
vector<int> src;
bool isprime(int x){
    if(x<2) return false;
    bool flag=true;
    int sqt=sqrt(x*1.0);
    for(int i=2;i<=sqt;i++){//这里包含等于
        if(x%i==0){
            flag=false;
            break;
        }
    }
    return flag;
}
void dfs(int k,int cnt){
    if(k>=n||cnt>=r){
        if(cnt==r){
            ll total=0;
            for(auto i:v) total+=i;
            // cout<<"\n";
            if(isprime(total)) {ans++;}
        }
        return;
    }
    v.push_back(src[k]);
    dfs(k+1,v.size());
    v.pop_back();
    dfs(k+1,v.size());
}
int main(){
    //非递归方法
    // freopen("a1.in","r",stdin);
    int x;
    while (cin>>n>>r)
    {
        ans=0;
        src.clear();
        v.clear();
        for(int i=0;i<n;i++)
            cin>>x,src.push_back(x);
        dfs(0,0);
        cout<<ans<<"\n";
    }
    return 0;
}

error:质数判断包括等于号

举报

相关推荐

0 条评论