【洛谷 P2392】kkksc03考前临时抱佛脚题解（动态规划+01背包）

HEX9CF

于 2023-10-05 16:38:37 发布

阅读量406

点赞数

分类专栏： Algorithm Problems 文章标签：动态规划算法 c++

本文链接：https://blog.csdn.net/qq_34988204/article/details/133580828

版权

Algorithm Problems 专栏收录该内容

453 篇文章 12 订阅

订阅专栏

kkksc03考前临时抱佛脚

题目背景

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

题目描述

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

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

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

输入格式

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

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

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

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

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

输出格式

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

样例 #1

样例输入 #1

样例输出 #1

提示

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

$1\leq A_1,A_2,\ldots,A_{s_1},B_1,B_2,\ldots,B_{s_2},C_1,C_2,\ldots,C_{s_3},D_1,D_2,\ldots,D_{s_4}\leq60$ 。

思路

在每个科目的循环中，计算每个科目的学习总时间 sum，并遍历每个学习时间，使用当前学习时间更新 dp 数组。

状态转移方程：

dp[k] = max(dp[k], dp[k - hw[j]] + hw[j]);

当每个科目左脑时间与右脑时间尽量接近时，总时间最小。

最后，累加每科所有作业总时间减去 dp[sum/2] 的值，得到完成所有科目的最短总时间。

AC代码

#include <iostream>
#include <algorithm>
#include <cstring>
#define AUTHOR "HEX9CF"
using namespace std;

const int N = 1e4 + 5;

int ans;
int s[5], hw[N];

int main()
{
    for (int i = 0; i < 4; i++)
    {
        cin >> s[i];
    }
    for (int i = 0; i < 4; i++)
    {
        int sum = 0;
        int dp[N];
        for (int j = 0; j < s[i]; j++)
        {
            cin >> hw[j];
            sum += hw[j];
        }
        memset(dp, 0, sizeof(dp));
        for (int j = 0; j < s[i]; j++)
        {
            for (int k = sum / 2; k >= hw[j]; k--)
            {
                dp[k] = max(dp[k], dp[k - hw[j]] + hw[j]);
            }
        }
        ans += sum - dp[sum / 2];
    }
    cout << ans << endl;
    return 0;
}