洛谷P1036 选数(DFS解n选m问题)
题目描述
已知 n 个整数 x1,x2,x3....,xn ,以及 1 个整数 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 个整数(满足条件的种数)。
输入输出样例
输入样例#1:
4 3
3 7 12 19
输出样例#1:
1
思路
n选m问题且数据量不大,用DFS枚举所有可能,再求和判断素数即可
注意
注意设全局变量的位置,让函数直接计算结果
代码
#include<bits/stdc++.h>
using namespace std;
int a[21];
int n,k;
bool isPrime(int x)
{
for(int i=2;i<=sqrt(x);i++)
{
if(x%i==0) return false;
}
return true;
}
int ans=0;
void DFS(int index,int nowK,int nowSum)
{
//找到满足条件的解
if(nowK==k&&isPrime(nowSum))
{
ans++;
return;
}
//处理完了全部或k多了(dfs边界)
if(index==n||nowK>k) return;
//选index
DFS(index+1,nowK+1,nowSum+a[index]);
//不选
DFS(index+1,nowK,nowSum);
}
int main()
{
scanf("%d %d",&n,&k);
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
DFS(0,0,0);
printf("%d",ans);
}
拓展
若要输出具体的方案,可用vector容器存对应各个数字,在满足条件后输出。具体看代码
#include<bits/stdc++.h>
using namespace std;
int a[21];
int n,k;
vector<int> temp,ans;
bool isPrime(int x)
{
for(int i=2;i<=sqrt(x);i++)
{
if(x%i==0) return false;
}
return true;
}
void DFS(int index,int nowK,int nowSum)
{
//找到满足条件的解
int s=0;
if(nowK==k&&isPrime(nowSum))
{
ans=temp;
//输出方案
for(int i=0;i<ans.size();i++)
{
printf("%d",ans[i]);
if(i!=ans.size()-1) printf("+");
else printf("=%d\n",nowSum);
}
return;
}
//处理完了全部或k多了(dfs边界)
if(index==n||nowK>k) return;
//选index
temp.push_back(a[index]);
DFS(index+1,nowK+1,nowSum+a[index]);
temp.pop_back();
//不选
DFS(index+1,nowK,nowSum);
}
int main()
{
scanf("%d %d",&n,&k);
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
DFS(0,0,0);
return 0;
}