NOIP2002选数
题目描述
已知 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 2 |
4 3 3 7 12 19 |
样例输出
1 |
1 |
代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
#include<iostream> #include<cmath> using namespace std; int a[21],ans=0,used[21]; int n,k; bool ss(int x)//判断素数 { for(int i=2;i<=sqrt(x);i++) if(x%i==0)return 0; return 1; } void search(int t,int s,int l)//t是当前选取的数个数,s为总和,l为使用到第l个数 { if(t==k){if(ss(s))ans++;} else for(int i=l;i<=n;i++) { if(!used[i]){used[i]=1;search(t+1,s+a[i],i);} } used[l]=0; } int main() { cin>>n>>k; for(int i=1;i<=n;i++) cin>>a[i]; search(0,0,1); cout<<ans; return 0; } |
used数组貌似是多余的….改一下
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
#include<iostream> #include<cmath> using namespace std; int a[21],ans=0; int n,k; bool ss(int x) { for(int i=2;i<=sqrt(x);i++) if(x%i==0)return 0; return 1; } void search(int t,int s,int l) { if(t==k){if(ss(s))ans++;} else for(int i=l;i<=n;i++) search(t+1,s+a[i],i+1); } int main() { cin>>n>>k; for(int i=1;i<=n;i++) cin>>a[i]; search(0,0,1); cout<<ans; return 0; } |
Subscribe