第三周作业——选数问题

选数问题

一、题目
给定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;
	}
}