总结一下,之后写题也更顺手~
eg:
给定N个整数,有正有负,从中选K个数使得和等于给定的整数X,选择他们中元素平方和最大的那个。
const int maxn=10020;
int n,k,x,maxSqu=-1,A[maxn];
vector<int> temp,ans;//要使用vector进行
void DFS(int index,int nowK,int sum,int sumSqu)//sumSqu 是已选的平方和
{
if(nowK==k&&sum==x){
if(sumSqu>maxSqu){
maxSqu=sumSqu;
ans=temp;//更新最优方案
}
return;
}
if(index==n||nowK>k|||num>x){
return;
}
temp.push_back(A[index]);//选index
DFS(index+1,nowK+1,sum+A[index],sumSqu+A[index]*A[index]);//进行递归
temp.pop_back();
DFS(index+1,nowK,sum+sumSqu);//不选index
}
倘若每个整数都可以被重复选中,选择K个使得和为X,这个代码只要在递归迭代时(选index号数)改为DFS(index,nowK+1,sum+A[index],sumSqu+A[index]*A[index]);