Week3 作业 A - 选数问题

问题描述：

n个数，选k个，要求这k个数的和是s，问有几种可行方案

思路：

直接DFS即可，注意边界和可行性剪枝。

总结：

可行性剪枝：当 选的数的个数大于k个、选的数的和大于s、选了k个数但和小于s、选了和为s的数但个数小于k ，这些情况都不必再继续搜索。

代码：

#include <cstdio>
#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
int n,k,s,ans;
int a[100005];
//num表示从a[num]-a[n]中选数，existed表示已经选了existed个数，sum是这些数的和 
void dfs(int num,int existed,int sum) 
{
    if( existed==k && sum==s ) ans++;
    else if( existed>=k || sum>=s ) return;
    
    for(int i=num;i<=n;i++)   //从a[num]-a[n]中选数
        dfs(i+1,existed+1,sum+a[i]);  //尝试选a[i]
}
int main()
{
    int T; cin>>T;
    while(T--)
    {
        scanf("%d %d %d",&n,&k,&s);
        memset( a,0,sizeof(a) );
        for(int i=1;i<=n;i++)
            scanf("%d",a+i);
        ans=0;
        dfs(1,0,0);
        cout<<ans<<endl;
    }
}