[算法]PAT-A_DFS模板

#include<vector>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;

const int maxn = 10010;
vector<int> temp, ans;
int A[maxn];
int k, sum;//题目要求的，选择数字的个数，以及要求数字之和
int max_squ = -11;
int n;

void DFS(int index, int nowk, int nowsum, int nowsqu){
    if(nowk==k && nowsum==sum){//找到了正确答案
        if(nowsqu>max_squ){//判断是否可以更新最优解
            max_squ = nowsqu;
            ans = temp;//得到最优解序列
        }
        return;//在DFS中考虑了出口，必须及时retun
    }
    if(nowk==n || nowsum>sum || nowk>k) return;//如果处理完了n个数，或者数字之和超过要求，或者选择数字个数超过要求，则return
    //否则的话，正常的递归操作
    temp.push_back(A[index]);
    DFS(index+1, nowk+1, nowsum+A[index], nowsqu+A[index]*A[index]);//选择这个数
    temp.pop_back();
    DFS(index+1, nowk, nowsum, nowsqu);//不选这个数
}
//由以上程序可知
//一般来讲DFS都是先将出口考虑清楚，比如if(满足条件)则更新解决方案 返回
//if(不再满足题目要求) 返回
//在判断出口条件并进行相应处理后，一定要记得及时return
//然后设计本轮递归要进行的正常操作，比如将index所代表的元素保存起来，进行下一步迭代；或者直接进行下一步迭代（不同的分支）