题目传送门:[NOIP2002 普及组] 选数 - 洛谷
题目描述
已知 n 个整数 x1,x2,⋯ ,xn,以及 1个整数 k(k<n)。从 n 个整数中任选 k 个整数相加,可分别得到一系列的和。例如当 n=4,k=3,44 个整数分别为 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)。
第二行 nn 个整数,分别为 x1,x2,⋯ ,xn(1≤xi≤5×106)。
输出格式
输出一个整数,表示种类数。
输入输出样例
输入 #1复制
4 3 3 7 12 19
输出 #1复制
1
说明/提示
【题目来源】
NOIP 2002 普及组第二
先上我的代码
#include <iostream>
#include <cstdio>
using namespace std;
bool isprime(int a){
if(a == 1) return false;
for(int i = 2; i * i <= a; i++)
if(a % i == 0)
return false;
return true;
}
int n,k;
int a[25];
long long ans;
void dfs(int m, int sum, int startx){
if(m == k){
if(isprime(sum))
ans++;
return ;
}
for(int i = startx; i < n; i++)
dfs(m + 1, sum + a[i], i + 1);
return ;
}
int main(){
scanf("%d%d",&n,&k);
for(int i = 0; i < n; i++)
scanf("%d",&a[i]);
dfs(0, 0, 0);
printf("%d\n",ans);
return 0;
}
//本蒟蒻的第一篇题解,求过
其实这里的难点是:如何去重?
答案是:不降原则
不降原则是个神马意思呢
举个例子:
比如说在6里面随便选5个数,那么选法都是什么呢?
瞎枚举?
12345
12346
前两个还不会弄混
然后很可能就乱了
少点数可能不会乱
但是多了就不好整了
比如说在100里随便选50个数。
1 2 3 4 5 6 7 8 9 10 11 12......
Die.
所以我们可以运用不降原则:
保证枚举的这些数是升序排列
其实真正的不降原则还可以平
比如 1 2 2 3 3 4......
但是请注意这道题也不能平
否则就有重复数字了
拿6个里面选3个举例子
1 2 3
1 2 4
1 2 5
1 2 6
第一轮枚举完毕。
第二个数加一
1 3 ?
这个“?”应该是4,因为是升序排列
1 3 4
1 3 5
1 3 6
接着,就是这样
1 4 5
1 4 6
1 5 6
第一位是1枚举完毕
第一位是2呢?
2 3 4
2 3 5
2 3 6
2 4 5
2 4 6
2 5 6
就是这样的,枚举还是蛮清晰的吧
以此类推.....
3 4 5
3 4 6
3 5 6
4 5 6
然后就枚举不了了,结束。
所以说,这样就可以避免判重了。
知道了不降原则,咱们再来see see
代码
代码中还是一样的老套路
一样的dfs
一样的参数
咱们最主要看其中不降原则的部分
在dfs函数中,我们明显能看到一个参数 startx
是startx而不是start是为了避免关键字
这是个什么东西呢?
就是当前的初始值
也就是最小的,符合不降原则的参数
这样一来,就可以去重了!
好的,代码说完了,最后给个带有注释版的程序:
#include <iostream>
#include <cstdio>
using namespace std;
bool isprime(int a){//判断素数
/*
吐槽:题中n的数据范围很奇怪,
n还有可能=1.....那k<n......
*/
if(a == 1) return false;
for(int i = 2;i * i <= a; i++)//不想用sqrt,还要头文件
if(a % i == 0)//如果整除
return false;//扔回false
//程序都到这里的话就说明此为素数
//否则就被扔回了
return true;//扔回true
}
int n,k;
int a[25];
long long ans;
void dfs(int m, int sum, int startx){//最重要的递归
//m代表现在选择了多少个数
//sum表示当前的和
//startx表示升序排列,以免算重
if(m == k){//如果选完了的话
if(isprime(sum))//如果和是素数
ans++;//ans加一
return ;
}
for(int i = startx; i < n; i++)
dfs(m + 1, sum + a[i], i + 1);//递归
//步数要加一,和也要加
//升序起始值要变成i+1,以免算重
return ;//这一个步骤下,所有的都枚举完了
//直接返回去
}
int main(){
scanf("%d%d",&n,&k);//输入
for(int i = 0; i < n; i++)
scanf("%d",&a[i]);//循环读入
dfs(0,0,0);//调用函数
printf("%d\n",ans);//输出答案
return 0;//结束程序
}
此解为歪解,仅供观赏,切勿模仿
发现原来漏考虑题中的k等于一的情况了
思路:题中的k等于几就几个for语句
768行代码见下:
#include <bits/stdc++.h>
#define ff(i