选数问题
一、题目
给定n个数,从中选出K个数和为S,求一共有多少种取法。
二、输入
第一行是一个整数T<=100,表示测试用例的数量。对于每个样例,输入有两行。第一行,三个整数代表n K s。第二行,n个整数代表正数。
三、输出
对于每个样例,输出一个整数在独立的行中表示答案。
四、样例输入输出
Input
1
10 3 10
1 2 3 4 5 6 7 8 9 10
Output
4
五、解题思路
利用dfs思想遍历加可行性剪枝解决。
枚举所有可能的子集情况,当和为1时,即为符合题意的一组数据,但是这样时间复杂度较大。所以添加条件,当元素个数大于K或者元素和大于S时跳出来达到剪枝的目的。
六、代码样例
#include<iostream>
#include<list>
using namespace std;
int cot=0;//ans 个数
int t,n,k;
double s;
void judge(int i,double ss,list<double> &b,double *a)
{
if(b.size()==k&&ss==s)
{
cot++;
// cout<<b.front()<<b.back()<<endl;
return;
}
// cout<<b.size()<<" "<<ss<<endl;
if(i>=n) return;
if(b.size()>k||ss>s) return;
judge(i+1,ss,b,a);
b.push_back(a[i]);
judge(i+1,ss+a[i],b,a);
b.pop_back();
}
int main()
{
cin>>t;
for(int i=0;i<t;i++)
{
cin>>n>>k>>s;
cot=0;
double ss=0;
list<double> b; //声明一个n个元素的列表,每个元素都是0
double a[n]={0};
b.clear();
for(int j=0;j<n;j++)
cin>>a[j];
judge(0,ss,b,a);
cout<<cot<<endl;
}
}