【蓝桥杯基础算法】dfs(上)组合数,全排列

刚接触算法,有没有被递归又循环的dfs吓到?没关系,几个例题就可以彻底掌握!

1.全排列

1-n的全排列,如输入3,按顺序对1-3进行排列

cpp 复制代码
//枚举
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=10;
int n;
bool st[N];//true选过 ,false是没选过 
int arr[N];//存的是答案  
void dfs(int x)
{
	if(x>n)
	{for(int i=1;i<=n;i++)
	{
		printf("%5d",arr[i]);//长宽5
	}
	printf("\n");
	return ;
	}
	for(int i=1;i<=n;i++)
	{
		if(!st[i])
		{
			st[i]=true;
			arr[x]=i;
			dfs(x+1);
			st[i]=false;
			arr[x]=0;
		}
	}
}
int main()
{
	scanf("%d",&n);
	dfs(1);
	
	return 0;
}

2.组合数

题目:

答案:

cpp 复制代码
//选数
//剪枝 
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=30;
int k,n;
int q[N];
int arr[N];
int res=0;

bool is_prim(int sum)
{
	if(sum<2)return false;
	for(int i=2;i<=sum/i;i++)
	{
		if(sum%i==0)return false;
	}
	return true;
}
//求组合数,x表示当前到了哪个位置
//start表示从几开始枚举 
void dfs(int x,int start)
{
	if(((x-1)+n-start+1)<k)
	{
		return ;
	}
	if(x>k){
		int sum=0
		for(int i=1;i<=k;i++)
		{
			sum+=arr[i];
		}
		if(is_prim(sum))//是素数+1 
		{
			res++;
		}
	}
	for(int i=start;i<=n;i++)
	{
		arr[x]=q[i];
		dfs(x+1,i+1);//继续向下,深度优先 
		arr[x]=0;//恢复现场 
	}
}

int main()
{
	scanf("%d %d",&n,&k);
	dfs(1,1);
	printf("%d\n",res);
	return 0;
}
相关推荐
在等晚安么14 分钟前
力扣面试经典150题打卡
java·数据结构·算法·leetcode·面试·贪心算法
AndrewHZ1 小时前
【图像处理基石】图像滤镜的算法原理:从基础到进阶的技术解析
图像处理·python·opencv·算法·计算机视觉·滤镜·cv
lxmyzzs1 小时前
【图像算法 - 30】基于深度学习的PCB板缺陷检测系统: YOLOv11 + UI界面 + 数据集实现
人工智能·深度学习·算法·yolo·缺陷检测
报错小能手1 小时前
C++笔记(面向对象)详解单例模式
c++·笔记·单例模式
gihigo19981 小时前
基于萤火虫算法(FA)优化支持向量机(SVM)参数的分类实现
算法·支持向量机·分类
吗~喽1 小时前
【C++】内存管理
c++
py有趣1 小时前
LeetCode算法学习之移动0
学习·算法·leetcode
lixinnnn.1 小时前
算法总结篇(枚举-分治)
算法·1024程序员节
on_pluto_2 小时前
【基础复习3】决策树
算法·决策树·机器学习
熬了夜的程序员2 小时前
【LeetCode】90. 子集 II
数据结构·算法·leetcode·链表·职场和发展·排序算法