codeves天梯 选数

已知 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）。

分析：就是暴搜嘛==。

var
  n,k,ans:longint;
  a:array[1..5000000]of longint;
procedure init;
var i:longint;
begin
  read(n,k);
  for i:=1 to n do
    read(a[i]);
  ans:=0;
end;

function prime(x:longint):boolean;
var i:longint;
begin
  prime:=true;
  for i:=2 to trunc(sqrt(x))do
    if x mod i=0 then
      begin
        prime:=false;
	break;
      end;
end;

procedure dfs(s,i,dep:longint);
var j:longint;
begin
  if dep=k+1 then
    begin
      if prime(s) then inc(ans);
      exit;
    end;
  for j:=i+1 to n do
    dfs(s+a[j],j,dep+1);
end;

begin
  init;
  dfs(0,0,1);
  writeln(ans);
end.