搜索——深度优先搜索（DFS）

　　设想我们现在身处一个巨大的迷宫中，我们只能自己想办法走出去，下面是一种看上去很盲目但实际上会很有效的方法。

　　以当前所在位置为起点，沿着一条路向前走，当碰到岔道口时，选择其中一个岔路前进。如果选择的这个岔路前方是一条死路，就退回到这个岔道口，选择另一个岔路前进。如果岔路口存在新的岔道口，那么仍然按上面的方法枚举新岔道口的每一条岔道。这样，只要迷宫存在出口，那么这个方法一定能够找到它。

　　也就是说，当碰到岔道口时，总是以“深度”作为前进的关键词，不碰到死胡同就不回头，因此这种搜索的方式称为深度优先搜索（DFS）。

　　接下来讲解一个例子。

　　有 n 件物品，每件物品的重量为 w[i]，价值为 c[i]。现在需要选出若干件物品放入一个容量为 V 的背包中，使得在选入背包的物品重量和不超过容量 V 的前提下，让背包中物品的价值之和最大，求最大价值。(1≤n≤20)

　　在这个问题中，对每件物品都有选或者不选两种选择，而这就是所谓的“岔道口”。那么什么是“死胡同”呢？题目要求选择的物品重量总和不能超过 V，因此一旦选择的物品重量总和超过 V，就会到达“死胡同”，需要返回最近的“岔道口”。

　　DFS 函数的参数中必须记录当前处理的物品编号 index，和在处理当前物品之前，已选物品的总重量 sumW 与 总价值 sumC。于是 DFS 函数如下：

void DFS(int index, int sumW, int sumC) {...}

　　思路：

• 如果选择不放入 index 号物品，那么 sumW 与 sumC 就将不变，接下来处理 index+1 号物品，即前往 DFS(index+1, sumW, sumC) 这条分支；
• 如果选择放入 index 号物品，那么 sumW=sumW+w[index], sumC=sumC+c[index]，接着处理 index+1 号物品，即前往 DFS(index+1, sumW+w[index], sumC+c[index]) 这条分支；
• 一旦 index 增长到了 n，则说明已经把 n 件物品处理完毕。此时记录的 sumW 和 sumC 就是所选物品的总重量和总价值。如果 sumW 不超过 V 且 sumC 大于记录的最大总价值 maxValue，就说明当前的这种选择方案可以得到更大的价值，于是用 sumC 更新 maxValue。　　　　　　

　　代码如下：

 /*
     搜索_DFS
     有 n 件物品，每件物品的重量为 w[i]，价值为 c[i]。现在需要选出若干件物品
         放入一个容量为 V 的背包中，使得在选入背包的物品重量和不超过容量 V 的前提下，
         让背包中物品的价值之和最大，求最大价值。(1≤n≤20)
 */

 #include <stdio.h>
 #include <string.h>
 #include <math.h>
 #include <stdlib.h>
 #include <time.h>
 #include <stdbool.h>

 #define maxn 30
 int n, V, maxValue;    // 物品减数，背包容量，最大价值
 int w[maxn], c[maxn];    // 每件物品的重量，价值 

 // index 当前处理的物品编号，sumW 和 sumC 为当前总重量和总价值
 void DFS(int index, int sumW, int sumC) {
     if(index == n) {    // 已经把 n 件物品处理完毕(死胡同）
         if(sumW <= V && sumC > maxValue) {
             maxValue = sumC;    // 有更好的选择
         }
         return;
     }
     // 岔道口
     DFS(index+, sumW, sumC);    // 不选 Index 号物品
     DFS(index+, sumW+w[index], sumC+c[index]);    // 选 index 号物品
 }

 int main() {
     scanf("%d %d", &n, &V);
     int i;
     for(i=; i<n; ++i) {
         scanf("%d", &w[i]);        // 每件物品的重量
     }
     for(i=; i<n; ++i) {
         scanf("%d", &c[i]);        // 每件物品的价值
     }
     DFS(, , );
     printf("%d\n", maxValue); 

     return ;
 }

　　在上述代码中，总是把 n 件物品的选择全部确定之后才去更新最大价值，但是事实上忽视了背包容量不超过 V 这个特点。也就是说，完全可以把对 sumW 的判断加入“岔道口”中，只有当 sumW ≤ V 时才进入岔道，这样效率会高很多。代码如下：

 // index 当前处理的物品编号，sumW 和 sumC 为当前总重量和总价值
 void DFS1(int index, int sumW, int sumC) {
     if(index == n) {    // 已经把 n 件物品处理完毕(死胡同）
         return;
     }
     // 岔道口
     DFS(index+, sumW, sumC);    // 不选 Index 号物品
     // 只有加入 index 物品后总重量小于 V 才可以继续
     if(sumW + w[index] <= V) {
         if(sumC + c[index] > maxValue) {
             maxValue = sumC + c[index];
         }
         DFS(index+, sumW+w[index], sumC+c[index]);    // 选 Index 号物品
     }
 }

　　再来看另外一个问题。

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

　　与之前的问题类似，此处仍然需要记录当前处理的整数编号 index；由于要求恰好选择 K 个数，因此需要一个参数 nowK 来记录当前已经选择的数的个数；另外，还需要参数 sum 和 sumSqu 分别记录当前已选整数之和与平方和。于是 DFS 函数如下：

void DFS(int index, int nowK, int sum, int sumSqu) {...} 

　　思路：

• 需要一个数组 temp，用以存放当前选择的整数。
• 当试图进入“选 index 号数”这条分支时，就把 A[index] 加入 temp 中；
• 当这条分支结束时，就还原 temp 数组，使他不影响“不选 index 号数”这条分支。
• 如果当前已选择了 K 个数，且这 K 个数之和恰为 x 时，就将平方和与已有的最大平方和 maxValue 作比较，如果更大，更新 maxValue 和数组 ans。　　

　　代码如下：

 /*
     DFS_N个整数选K个
     给定 N 个整数（可能有负数），从中选择 K 个数，使得这 K 个数之和
     恰好等于一个给定的整数 X；如果有多种方案，选择它们中元素平方和最大的一个。
 */

 #include <stdio.h>
 #include <string.h>
 #include <math.h>
 #include <stdlib.h>
 #include <time.h>
 #include <stdbool.h>

 #define maxn 30
 // 序列A中n个数选k个数使得和为x，最大平方和为maxSumSqu
 int n, k, x, maxSumSqu=-, A[maxn];
 int temp[maxn]={}, ans[maxn]={};    // 临时方案，平方和最大的方案

 void DFS(int index, int nowK, int sum, int sumSqu) {
     if(nowK == k && sum == x) {        // 找到K个数和为x
         if(sumSqu > maxSumSqu) {    // 更优方案
             maxSumSqu = sumSqu;        // 更新 maxValue 和数组 ans
             int i;
             for(i=; i<k; ++i) {
                 ans[i] = temp[i];
             }
         }
     }
     // 已经处理完n个数，选择超过k个数，和大于x
     if(index==n || nowK>k || sum>x)    return;
     // 选 index 号数
     temp[nowK] = A[index];
     DFS(index+, nowK+, sum+A[index], sumSqu+A[index]*A[index]);
     temp[nowK] = ;
     // 不选 index 号数
     DFS(index+, nowK, sum, sumSqu);
 } 

 int main() {
     scanf("%d %d %d", &n, &k, &x);
     int i;
     for(i=; i<n; ++i) {
         scanf("%d", &A[i]);    // n个数
     }
     DFS(, , , );
     for(i=; i<k; ++i) {    // 最优方案
         printf("%d ", ans[i]);
     }
     printf("%d\n", maxSumSqu);    // 最优方案的平方和 

     return ;
 }