DFS——选数字

版权声明：关中大侠Lv轻侯 https://blog.csdn.net/weixin_44312186/article/details/88732820

题目概述：

          如从4个数字{2，3，3，4 }中选2个数字使其和为6，求平方和最大的方案。

思路：

        类似于背包问题，但不同的是要输出最佳的方案，因此需要使用STL中的vector（“变长数组”）来存储方案。关键在于如何使选中“index”的方案不影响不选中“index”的方案。

代码：

#include<cstdio>
#include<vector>
using namespace std;
const int maxn=30;

int n,k,x,maxsumsqu=0,a[maxn];//在a中的n个数中选k个数字，使其和为x，最大平方和为maxsumsqu
vector<int> temp,ans;
void dfs(int index,int sum,int sumsqu,int nowk)
{
	if(sum==x&&nowk==k)
	{
		if(sumsqu>maxsumsqu)
		{
			maxsumsqu=sumsqu;
			ans=temp;
		}
		return;
	}
	if(index==n||nowk>k||sum>x)  return;//3中退出方案 
	temp.push_back(a[index]);//当选定a[index]时 
	dfs(index+1,sum+a[index],sumsqu+a[index]*a[index],nowk+1);
	temp.pop_back();
	dfs(index+1,sum,sumsqu,nowk); 
}

int main()
{
	scanf("%d%d%d",&n,&k,&x);
	for(int i=0; i<n; i++)
	{
		scanf("%d",&a[i]);
	}
	dfs(0,0,0,0);
	for(int i=0;i<ans.size();i++){
		printf("%d ",ans[i]);	
	}
	return 0;
}