选数--回溯法（洛谷）

@[TOC]选数（洛谷）
添加链接描述

这道洛谷的选数题，有两个问题需要考虑。一是在n个数中选出k个数，二是判断k个数的和是否为素数（当然上一个问题解决了，这个便问题不大(@.@).
我用的是万能的dfs深搜解决的选数问题，而后答案就呼之欲出了。废话不多说，下面贴代码。

#include <iostream>
#include <algorithm>
#include <set>
using namespace std;
int n,k,a[21],book[21]={0},c[21],vis[10]={0},num=0,flag=0;
set<int> it,itt;
int sushu(int n)//判断是否为素数
{
 if(n==1)
 return 0;
 else {
  for(int i=2;i<n;i++)
  {
   if(n%i==0)
   return 0;
  }
  return 1;
 }
}
void dfs(int step)//深搜
{
 int i,sum=0,p=0,q=1,w=0,m;
 if(step==k+1)
 {
  for(i=1;i<=k;i++)
  sum+=c[i];
  if(sushu(sum))
  {
   for(i=1;i<=k;i++)
  {
   p+=vis[i];
   q=q*(vis[i]%20);
  }
  w=p+q;
  pair<set<int>::iterator,bool> result=it.insert(w);
  if(!result.second)
  return;
  num++;
  }
  return;
 }
 for(i=1;i<=n;i++)
 {
  if(book[i]==0)
  {
   book[i]=1;
   c[step]=a[i];
   vis[step]=i;
   dfs(step+1);
   book[i]=0;
  }
 }
 return;
}
int main()
{
 int i;
 cin>>n>>k;
 for(i=1;i<=n;i++)
 cin>>a[i];
 sort(a+1,a+n+1);
 dfs(1);
 cout<<num<<endl;
 return 0;
}

这道题就是酱紫，欢迎大家与我交流~