零、前言
前段时间为蓝桥杯省赛做准备,了解到了深度优先搜索(dfs),正好遇到一题,成功AC通过了。
一、题目详细
1.题目描述
请写一个程序,在输入的n个正整数中,判断能否找到三个数总和为m,这三个数可以重复(即n个数中的每个数都可以使用多次)。如果可以找到,则输出为Yes。若找不到则输出No。输入的第一行为n和m两个整数,第二行为n个数字。 其中n<=1000,数字绝对值不超过2000。
输入格式:第一行为n和m两个整数,第二行为n个数字。 其中n<=1000,数字绝对值不超过2000。m<=2000。
输出格式:Yes或者No
2.输入样例
4 15
1 5 3 9
3.输出样例
Yes
4.解题思路
关于这道题,很多人都想到用三重循环来依次判断,但会TE(Time error)。于是,又有人用起了goto跳转,确实简化了,但代码的可读性明显下降,不推荐。
于是,我们就用深搜dfs,以一个深度和一个要处理的数据做参数,定义void类函数,先写一个递归边界(深度达标或有错),接着用循环依次递归。
这里注意,我们的目的是简化时间复杂度,不是单纯地改变程序的结构与代码的可读性,而是需要有排除,在以下代码中,在循环中加入判断即可。
5.AC代码
#include<cstdio>//调用C语言输入输出头文件
using namespace std;//使用标准名字空间
int N[10009];//定义数组N,包含10009个整数类型变量
bool t;//定义布尔类型变量t
int n,m;//定义整数类型变量n,m
void dfs(int deep,int need){//定义函数dfs,参数为搜索深度,还需凑need
if(deep==3){//深度达标
if(need==0){//刚好凑够
t=1;//成功
}return;//返回
}
for(int i=0;i<n;i++){//搜索
if(N[i]>need){//不满足情况(减少时间复杂度)
return;//返回
}
dfs(deep+1,need-N[i]);//进行新一轮的枚举
}
}
int main(){//主函数开始
scanf("%d %d",&n,&m);//输入n,m的值
for(int i=0;i<n;i++){//for循环,计数器i从0自增到n-1,共循环n次
scanf("%d",&N[i]);//输入N的i号元素
}
dfs(0,m);//使用搜索
if(t){//如果t为真
printf("Yes");//输出Yes
}else{//否则
printf("No");//输出No
}
return 0;//主函数结束,返回0
}
总结
那么本期的刷题笔记就到这里了,有问题的可以私信我。