dfs练习

最新推荐文章于 2021-02-21 19:42:33 发布

石窟寺前写代码

最新推荐文章于 2021-02-21 19:42:33 发布

阅读量290

点赞数

分类专栏：杂七杂八练习

本文链接：https://blog.youkuaiyun.com/Toby_Cho/article/details/56851331

版权

杂七杂八练习专栏收录该内容

1 篇文章

订阅专栏

//序列A中n个数选k个数使得和为x，最大平方和是maxsumsqu
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
using namespace std;

const int maxn=30;
vector<int> ans,temp;
int maxsumsqu=0;
int a[maxn];
int k,x,n;
void dfs(int index,int nowk,int sum,int sumsqu)	
{
	if(nowk==k&&sum==x)
	{
		if(sumsqu>maxsumsqu)
		{
			maxsumsqu=sumsqu;
			ans=temp;
		}
		return;
	}
	if(nowk>k||sum>x||index==n)
	{
		return;
	}
	temp.push_back(a[index]);
	dfs(index+1,nowk+1,sum+a[index],sumsqu+a[index]*a[index]);
	temp.pop_back();
	dfs(index+1,nowk,sum,sumsqu);
}
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);
	printf("%d",maxsumsqu);
	for(int i=0;i<ans.size();i++)
	{
		printf(" %d",ans[i]);
	}
	system("pause");
	return 0;
}