选数问题

问题描述

给n个正整数，可从中选择K个整数使和为S，现在计算有多少种选取方法？
【输入】第一行：T<=100,测试数据的组数
第二行：n，K，S
第三行：给定的n个正整数
【输出】 可选取方法数量

问题分析

典型的子集枚举问题，可以枚举所有子集判断是否合法，显然有很多不可能合法的情况，可以直接跳过
（1）选数的个数超过了K
（2）选的数的和超过了sum
solve（）函数：
（1）在搜索到边界以及有不合法情况时直接跳出
（2）递归 solve函数首先判断当前选择是否成功，满足选择个数以及sum==0，在递归时若选择当前元素则令sum减去该值，若不选a[i]则直接递归下一个i+1号元素,若选择该元素则先将该元素push_back放入list中，为了遍历任意n个元素的情况，在每一层递归后将最后一个放入的元素删除，以尝试其他组合方式达到遍历。
（3）输出，在判断满足条件时令计数的cnt加一，最终跳出循环时为遍历每一种情况之后得到的可行性结果

重点

可行性剪枝和最优性剪枝达到程序的优化

补充list

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

int cnt=0;
int a[20];
int m,p;
void solve(int i,int sum,list<int>&t)
{ //i是目前检测数的下标,m是选数个数,p是所有数的个数 
	if(t.size()==m && sum==0)
	{
		cnt++;
		return; 
	}
	if(i>=p+1) return;
	if(t.size()>m||sum<0) return;//可行性剪枝 
	
	solve(i+1,sum,t);//不选下标为i的数 
	t.push_back(a[i]);//选下标为i的数
	solve(i+1,sum-a[i],t);
	t.pop_back();	
}

int main()
{
	int n;
	list<int> t;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		int sum;   //p是可选数的个数,m是待选数的个数
	    cin>>p>>m>>sum;
		for(int j=1;j<=p;j++)
		{
			cin>>a[j];
		}
		solve(1,sum,t);
		cout<<cnt<<endl;
		t.clear(); 
		cnt=0; 
	}
}