题目来自牛客网(蓝桥杯集训第3期)
重点:<深度搜索DFS><回溯算法:组合问题>
题目描述
已知 n 个整数 x1,x2,…,xn,以及一个整数 k(k<n)。从 n 个整数中任选 k 个整数相加,可分别得到一系列的和。例如当 n=4,k=3,4 个整数分别为 3,7,12,19 时,可得全部的组合与它们的和为:
3+7+12=22 3+7+19=29 7+12+19=38 3+12+19=34。
现在,要求你计算出和为素数共有多少种。
例如上例,只有一种的和为素数:3+7+19=29)。
输入描述:
输入格式为:n , k(1<=n<=20,k<n) x1,x2,…,xn (1<=xi<=5000000)
输出描述:
输出格式为:一个整数(满足条件的种数)。
示例1:
//输入
4 3
3 7 12 19
//输出
1
思路
1.判断是否是素数
求素数的方法:接收一个整数 x
作为参数,然后通过遍历2到sqrt(x)的范围,检查是否有除1和自身以外的因子,若有则返回 false
,否则返回 true
。
int prime(int x)
{
for(int i=2;i<=sqrt(x);i++)
{
if(x%i==0)
return 0;
}
return x;
}
2.深度搜索
(组合问题:回溯算法)(深度搜索DFS)
这是一个深度优先搜索(DFS)的递归函数。
它有三个参数:
sum
表示当前子序列的和;
d
表示当前子序列的长度;
t
表示当前考虑的元素的下标。
如果d
等于k
(在主函数中输入的值),并且sum
是一个质数,则将ans
的值加1。否则,对于数组中的每一个元素(从下标t
到n-1
),递归调用DFS
函数。
void DFS(int sum,int d,int t)
{
if(d==k)
{
if(prime(sum))
ans++;
}
else
{
for(int i=t;i<n;i++)
{
DFS(sum+arr[i],d+1,i+1);
}
}
}
3.主函数
int main()
{
scanf("%d %d",&n,&k); // 从标准输入读取两个整数n和k。
for(int i=0;i<n;i++)
{
scanf("%d",&arr[i]); // 从标准输入读取n个整数并存储在数组arr中。
}
DFS(0,0,0); // 调用DFS函数,开始搜索。搜索的起始条件是sum=0, d=0, t=0。
printf("%d",ans); // 打印出搜索结果。即所有长度为k且和为质数的子序列的数量。
return 0; // 主函数结束,返回0。
}
完整代码
#include<stdio.h>
#include <math.h>
int n,k,ans;
int arr[1024];
int prime(int x)
{
for(int i=2;i<=sqrt(x);i++)
{
if(x%i==0)
return 0;
}
return x;
}
void DFS(int sum,int d,int t)
{
if(d==k)
{
if(prime(sum))
ans++;
}
else
{
for(int i=t;i<n;i++)
{
DFS(sum+arr[i],d+1,i+1);
}
}
}
int main()
{
scanf("%d %d",&n,&k);
for(int i=0;i<n;i++)
{
scanf("%d",&arr[i]);
}
DFS(0,0,0);
printf("%d",ans);
return 0;
}