计蒜客 dfs 买书

题目：

蒜头君去书店买书，他有 $m$ 元钱，书店里面有 $n$ 本书，每本书的价格为 $p_i$ 元。蒜头君很爱学习，想把身上钱都用来买书，并且刚好买 $k$ 本书。请帮蒜头君计算他是否能刚好用 $m$ 元买 $k$本书。

输入格式

第一行输入 $3$ 个整数 $m(1\le m\le 100000000)$，$n(1\le n\le 30)$，$k(1\le k\le min(8,n))$

接下来一行输入 $n$ 个整数，表示每本书的价格 $p_i(1\le p_i\le 100000000)$。

输出格式

如果蒜头君能 刚好 用 $m$ 元买 $k$ 本书，输入一行"Yes", 否则输出"No"。

样例输入1

10 4 4
1 2 3 4

样例输出1

Yes

样例输入2

10 4 3
1 2 3 4

样例输出2

No

最开时做这题时直接想的是二进制枚举，，但超时了，，，后来又改为dfs。

先附上二进制枚举代码(会超时)：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
/* run this program using the console pauser or add your own getch, system("pause") or input loop */


int main(int argc, char *argv[])
{
int flag=0,count=0;
int s=0,i,j;
long long int m;
int n,k;
scanf("%d%d%d",&m,&n,&k);
int q=m;
long long int p[n+1];
for(i=1;i<=n;i++)
{
scanf("%lld",&p[i]);
}
for(i=0;i<(1<<n);i++)
{
for(j=0;j<n;j++)
{
if(i&(1<<j))
{
m-=p[j+1];
count++;
}
}
if(m==0&&count==k)
{
flag=1;
}
m=q;
count=0;
s=0;
}
if(flag)
{
printf("Yes");
}else
{
printf("No");
}


return 0;
}

10个测试点只过了5 6个

下面是dfs：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
int s=0;//当前总价
int count= 0;//当前数量 
int flag=0;
int main(int argc, char *argv[]) 
{
int i;
int m,n,k;
scanf("%d%d%d",&m,&n,&k);
int p[n+100];
int c[n+100];
memset(c,0,sizeof(c));
for(i=1;i<=n;i++)
{
scanf("%d",&p[i]);
}
c[0]=1;
c[1]=0;
dfs(0,c,k,m,n,p);
if(flag)
{
printf("Yes");
}else
{
printf("No");
}
return 0;
}
void dfs (int x,int c[],int k,int m,int n,int p[])
{
int i;
if(s==m&&count==k)
{
flag=1;
return;
}
if((s>m)||(count>k))
{
return;
}
int tx = x+1;
 for(i=0;i<=1;i++)
 {
  if(i==0&&tx<=n&&tx>0)//买 
  {
  c[tx]=1;
  s+=p[tx];
  count++;
  dfs(tx,c,k,m,n,p);
  s-=p[tx];
  count--;
 
  }else if(i==1&&tx<=n&&tx>0)//不买
 {
  dfs(tx,c,k,m,n,p);
 } 
 
 }
 
}

代码写得很丑,,,,