2002年分区联赛普级组之二 选数
Description
已知 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)。
Input
n , k (1<=n<=20,k<n)
x1,x2,…,xn (1<=xi<=5000000)
Output
一个整数(满足条件的种数)。
分析:直接暴力搜索所有情况然后判断是否满足条件。
代码
var
a:array[0..20] of longint;
n,k,i,ans:longint;
function check(x:longint):boolean;
var
i:longint;
begin
for i:=2 to trunc(sqrt(x)) do
if x mod i=0 then exit(false);
exit(true);
end;
procedure dfs(dep,x,s:longint);
var
i:longint;
begin
if dep>k then
begin
if check(s) then inc(ans);
exit;
end;
for i:=x+1 to n do
dfs(dep+1,i,s+a[i]);
end;
begin
readln(n,k);
for i:=1 to n do
read(a[i]);
dfs(1,0,0);
writeln(ans);
end.