【经典dfs模板题】选数问题

题目描述

给你n个数，取出k个使得加起来的和为s，求所有情况。

Input

第一行给出输入数据组数T。对于每组数据，共有两行，第一行是n，k，s。第二行是n个数据（保证互不相同）。

output

每组数据输出独立一行的结果。

Example

Input

1
10 3 10
1 2 3 4 5 6 7 8 9 10

Output

4

思路

经典dfs模板问题，dfs每层选取第i个数，直到选取k个数，由于本题结果是无序的，所以每一层的选择范围要求最后选数序列是递增的，故对所有数进行排序，同时dfs函数加一个参数，表示t-1层选取的数的编号pre，这样t层选择的范围是pre+1~n。此外还可以用stl中的list来做，这里不再赘述。最后还有一个小小的剪枝，即在t<k时所有数的和已经大于s（前提：所有数为正数），则搜索提前结束。

总结

又双叒叕忘记每组数据的初始化，ans要清零，改了好多遍，注意思路清晰逻辑严谨。

代码

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iomanip>
using namespace std;
#define rep(i,s,t) for(int i=s;i<=t;i++)
int a[20]={0};
int n,k,s;
int ans = 0;
void dfs(int t,int sum,int pre){
	if(sum > s) 
		return;
	rep(i,pre+1,n){
		sum+=a[i];
		if(t == k){ 
			if(sum == s)
				ans++;
		} 
		else dfs(t+1,sum,i);
		sum-=a[i];
	}
}
int main(){
//	freopen("1.txt","r",stdin);
	int T;
	cin>>T;
	while(T--){
		cin>>n>>k>>s;
		ans=0;
		rep(i,1,n)
			cin>>a[i];
		sort(a+1,a+n+1);
		dfs(1,0,0);
		cout << ans <<endl;
	}
	return 0;
}