【Week3作业 A】选数问题【dfs】【剪枝】

题意：

给定n个数，从里面挑选K个数，使得它们的总和为S。输入共有T组数据（T<=100），每组数据有两行，第一行为n,K,S，第二行为n个数。输出共几种总和为S的数的组合。

---

思路：

使用dfs思想进行搜索。对每个数都有选择与不选择两种情况，然后在每种情况下对下一个数进行选择，直到满足条件或者到达边界。这里需要进行可行性剪枝，即如果当前选的数大于K个或者总和大于S时，直接返回，不再进行下一步选择。

---

总结：

一道很典型dfs剪枝题目。剪枝条件一般由题目条件判断得出。

---

代码：

#include <iostream>
#include <algorithm>
using namespace std;

long long int n,K,S;
long long int ans=0;

//i为数组第i个,sum为在当前已选择数的情况下还要选择总值为sum的数
//size为已经选了几个数,*a为数组 
void solve(long long int i,long long int sum,long long int size,long long int *a) 
{
	if(size==K&&sum==0)
	{
		ans++;
		return;
	}
	//搜索边界
	if(i>=n) return;
	//可行性剪枝
	if(size>K||sum<0)	return;
	//是否选择a[i] 
	solve(i+1,sum,size,a);		//不选
	solve(i+1,sum-a[i],size+1,a); //选 
}
int main()
{
	int T;
	cin>>T;
	for(int i=0;i<T;i++)
	{
		cin>>n>>K>>S;
		long long int *a=new long long int [n];
		for(long long int j=0;j<n;j++)
			cin>>a[j];
		sort(a,a+n);
		ans=0;
		solve(0,S,0,a);
		cout<<ans<<endl;
	}
}