DFS的简单例子

例题：给定N个整数，从中选择K个数，使得这K个数之和恰好等于一个给定的整数X；如果有多种方案，选择它们中元素平方和最大的一个。

如：从4个整数{2，3，3，4}中选择2个数，使它们的和为6。平方和最大的方案为{2，4}，和为20。

 

其中，递归函数需有以下4个部分：

1.满足条件

2.越界

3.未满足条件，进入下一层

4.走完该路径，换走另一条

 

#include<cstdio>
#include<iostream>
#include<vector>

using namespace std;

const int maxn = 50;
int n,k,x,maxSumSqu = -1,A[maxn];

vector<int> temp,ans;

void DFS(int index,int nowK,int sum,int sumSqu){
	if(nowK == k && sum == x){                       //1.满足条件
		if(sumSqu > maxSumSqu){
			maxSumSqu = sumSqu;
			ans = temp;
		}	
		return;
	}
	if(index == n || nowK > k || sum > x) return;    //2.越界
	temp.push_back(A[index]);                        //3.未满足条件，进入下一层
	DFS(index + 1, nowK + 1,sum + A[index],sumSqu + A[index]*A[index]);
	temp.pop_back();                                 //4.走完该路径，换走另一条
	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);

	return 0;
}