（DFS）洛谷P2392 kkksc03考前临时抱佛脚

P2392 kkksc03考前临时抱佛脚

题目背景

kkksc03 的大学生活非常的颓废，平时根本不学习。但是，临近期末考试，他必须要开始抱佛脚，以求不挂科。

题目描述

这次期末考试，kkksc03 需要考 $4$ 科。因此要开始刷习题集，每科都有一个习题集，分别有 $s_1,s_2,s_3,s_4$ 道题目，完成每道题目需要一些时间，可能不等（$A_1,A_2,\dots ,A_{s_1}$，$B_1,B_2,\dots ,B_{s_2}$，$C_1,C_2,\dots ,C_{s_3}$，$D_1,D_2,\dots ,D_{s_4}$）。

kkksc03 有一个能力，他的左右两个大脑可以同时计算 $2$ 道不同的题目，但是仅限于同一科。因此，kkksc03 必须一科一科的复习。

由于 kkksc03 还急着去处理洛谷的 bug，因此他希望尽快把事情做完，所以他希望知道能够完成复习的最短时间。

输入格式

本题包含 $5$ 行数据：第 $1$ 行，为四个正整数 $s_1,s_2,s_3,s_4$。

第 $2$ 行，为 $A_1,A_2,\dots ,A_{s_1}$ 共 $s_1$ 个数，表示第一科习题集每道题目所消耗的时间。

第 $3$ 行，为 $B_1,B_2,\dots ,B_{s_2}$ 共 $s_2$ 个数。

第 $4$ 行，为 $C_1,C_2,\dots ,C_{s_3}$ 共 $s_3$ 个数。

第 $5$ 行，为 $D_1,D_2,\dots ,D_{s_4}$ 共 $s_4$ 个数，意思均同上。

输出格式

输出一行,为复习完毕最短时间。

输入输出样例 #1

输入 #1

1 2 1 3		
5
4 3
6
2 4 3

输出 #1

20

说明/提示

$1\le s_1,s_2,s_3,s_4\le 20$。

$1\le A_1,A_2,\dots ,A_{s_1},B_1,B_2,\dots ,B_{s_2},C_1,C_2,\dots ,C_{s_3},D_1,D_2,\dots ,D_{s_4}\le 60$。

算法核心逻辑

这段代码的核心是使用深度优先搜索（DFS）结合回溯法来解决子集和问题：

1. 问题转化：
   • 对于每组数据，目标是将其分成两个子集 A 和 B，使得 | sum (A) - sum (B)| 最小
   • 转化为寻找子集 A，使得 A 的和不超过总和的一半（sum/2），且尽可能接近
2. DFS 策略：
   • 对于每个元素，有两种选择：加入当前子集或不加入
   • 使用递归回溯遍历所有可能的子集组合
   • 通过剪枝条件（nowtime + a [x] <= sum/2）减少不必要的搜索
3. 最优性保证：
   • 当遍历完所有元素时，记录当前找到的最大子集和
   • 由于每次选择都是基于 “不超过 sum/2” 的条件，最终得到的结果必然是最优解

题解

#include<iostream>
#include<algorithm>
#define Max 30
using namespace std;
int nowtime, maxtime, sum;
int ans, maxdeep;
int s[4], a[Max];
void dfs(int x) {
    // 边界条件：所有元素处理完毕，更新maxtime
    if (x > maxdeep) {
        maxtime = max(nowtime, maxtime);
        return; //当前路径已经探索完毕，需要回到上一层递归，继续探索其他未完成的路径
    }
    // 选择当前元素a[x]（若不超过sum/2）
    if (nowtime + a[x] <= sum / 2) {
        nowtime += a[x];   // 加入当前元素
        dfs(x + 1);        // 递归处理下一个元素
        nowtime -= a[x];   // 回溯：撤销选择
    }
    // 不选择当前元素a[x]
    dfs(x + 1);            // 直接处理下一个元素
}
int main() {
    cin >> s[0] >> s[1] >> s[2] >> s[3];
    for (int i = 0; i < 4; i++) {
        nowtime = 0;
        maxdeep = s[i];
        sum = 0;
        for (int j = 1; j <= s[i]; j++) {
            cin >> a[j];
            sum += a[j];
        }
        maxtime = 0;
        dfs(1);
        ans += (sum - maxtime);
    }
    cout << ans;
    return 0;
}