2002年分区联赛普级组之二 选数(dfs)

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.