计蒜客 dfs 买书

最新推荐文章于 2019-08-04 23:05:45 发布

xiadada2

最新推荐文章于 2019-08-04 23:05:45 发布

阅读量558

点赞数

CC 4.0 BY-SA版权

文章标签： ds 算法计蒜客

本文链接：https://blog.youkuaiyun.com/xiadada2/article/details/78980356

题目：

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

输入格式

第一行输入 $3$ 个整数 $\le m \le 100000000)$ ， $\le n \le 30)$ ， $\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);
}

}

}

代码写得很丑,,,,