Problem J: 数字选择
Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 820 Solved: 198
Description
给出一个没有重复元素的集合,如{1, 2, 3, 4},可从这个集合中选取任意个元素,求有多少种选取方法使得这些元素相加以后等于K。
Input
输入包含多组数据测试,每组数据第一行是整数N和K ( 1 <= N <= 5,1 <= K <= 1000),接下来一行是N个整数。
Output
对于每组数据输出在符合条件的情况下,有多少种选取方法。
Sample Input
4 5
1 2 3 4
Sample Output
2
HINT
这题要dfs深度搜索,正常来说如果数据大需要标记已经走过的节点,并且目前没有碰过。可以简化,因为数据处在1-5.所以可以不用标记递归处理。
代码:
#include<stdio.h>
int a[100],count,N,K;
void dfs(int now,int sum);
int main(void)
{
int i,sum;
while(scanf("%d%d", &N, &K)!=EOF)
{
sum=0;
for(i=0;i<N;i++)
{
scanf("%d", &a[i]);sum+=a[i];
}
if(sum==K)
printf("1\n");
else if(sum<K)
printf("0\n");
else
{count=0;dfs(0,0);printf("%d\n",count);}
}
return 0;
}
void dfs(int now, int sum)
{
if(now>=N) return; ///这个return回到之前的dfs。
if(sum+a[now]==K) ///递归的理解建议自己草稿纸上用 3 6 1 2 3 去算一下就明白了
count++;
dfs(now+1,sum+a[now]);//处理完后再进行dfs(now+1,sum);处理完了再处理之前的 //dfs(now+1,sum);从而达到遍历搜索完
dfs(now+1,sum);
}