当前位置: 代码迷 >> 综合 >> 洛谷-P1036 选数
  详细解决方案

洛谷-P1036 选数

热度:86   发布时间:2024-01-30 17:16:17.0

题目描述

已知 nn 个整数 x_1,x_2,…,x_nx1?,x2?,…,xn?,以及11个整数kk(k<nk<n)。从nn个整数中任选kk个整数相加,可分别得到一系列的和。例如当n=4,k=3n=4,k=3,44个整数分别为3,7,12,193,7,12,19时,可得全部的组合与它们的和为:

3+7+12=223+7+12=22

3+7+19=293+7+19=29

7+12+19=387+12+19=38

3+12+19=343+12+19=34。

现在,要求你计算出和为素数共有多少种。

例如上例,只有一种的和为素数:3+7+19=293+7+19=29。

输入格式

键盘输入,格式为:

n,kn,k(1 \le n \le 20,k<n1≤n≤20,k<n)

x_1,x_2,…,x_n (1 \le x_i \le 5000000)x1?,x2?,…,xn?(1≤xi?≤5000000)

输出格式

屏幕输出,格式为: 11个整数(满足条件的种数)。

输入输出样例

输入 #1复制

4 3
3 7 12 19

输出 #1复制

1

 

分析:

这题不难,就是有很多细节需要考虑到,尤其是一些边界数据。首先用a数组存数,并从小到大排序,然后设定一个mark数组,mark数组只有两个值,0或者1,mark[i]==1代表对应的a[i]是k个数中的一个,初始化mark为0,0,0,0...1,1,1,1,1,其中含k个1,n-k个0,然后通过while循环,对mark数组进行全排列,实质上是遍历k个数取值的所有情况,然后针对每一种情况,判断k个数的和是否为素数。

题主写这道题的时候,出了两个小毛病,希望各位引以借鉴

1.next_permutation()这个函数的结束是降序排列,所以初始化的时候0在前,1在后,若是反过来了,那么对不起,这个遍历函数只能执行一次,因为执行一次之后他就发现已经是降序排列了,就停止了,所以这点要注意

2.1不是素数也不是合数,记得单独拉出来

上AC代码:

#include<iostream>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;
int n,k;
int a[25];
int mark[25];
int check();
int ifsushu(unsigned long long int t);
int ans=0;
int main()
{memset(mark,0,sizeof(mark));cin>>n>>k;for(int i=1;i<=n;i++)cin>>a[i];for(int i=1;i<=n-k;i++)mark[i]=0;for(int i=n-k+1;i<=n;i++)mark[i]=1;sort(a+1,a+n+1);//cout<<"a is "<<endl;//for(int i=1;i<=n;i++)//cout<<a[i]<<" ";//cout<<endl;do{//for(int i=1;i<=n;i++)//cout<<mark[i]<<" ";//cout<<endl;int flag=check();if(flag)ans++;}while(next_permutation(mark+1,mark+n+1));cout<<ans;return 0;
}
int check()
{unsigned long long int sum=0;for(int i=1;i<=n;i++)if(mark[i])sum+=a[i];//cout<<"sum is "<<sum<<endl;int flag=ifsushu(sum);//cout<<"flag is "<<flag<<endl;return flag;}
int ifsushu(unsigned long long int t)
{if(t==1)return 0;int flag=1;int temp=sqrt(t);//cout<<"t is "<<t<<" temp is "<<temp<<endl;for(int i=2;i<=temp;i++){if(t%i==0){flag=0;break;}}return flag;
}