P2392 kkksc03考前临时抱佛脚【dp、01背包】

题目背景

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

题目描述

这次期末考试，kkksc03 需要考 4 科。因此要开始刷习题集，每科都有一个习题集，分别有 s1​,s2​,s3​,s4​ 道题目，完成每道题目需要一些时间，可能不等（A1​,A2​,…,As1​​，B1​,B2​,…,Bs2​​，C1​,C2​,…,Cs3​​，D1​,D2​,…,Ds4​​）。

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

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

输入格式

本题包含 5 行数据：第 1 行，为四个正整数 s1​,s2​,s3​,s4​。

第 2 行，为 A1​,A2​,…,As1​​ 共 s1​ 个数，表示第一科习题集每道题目所消耗的时间。

第 3 行，为 B1​,B2​,…,Bs2​​ 共 s2​ 个数。

第 4 行，为 C1​,C2​,…,Cs3​​ 共 s3​ 个数。

第 5 行，为 D1​,D2​,…,Ds4​​ 共 s4​ 个数，意思均同上。

输出格式

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

输入输出样例

输入

1 2 1 3		
5
4 3
6
2 4 3

输出

20

说明/提示

1≤s1​,s2​,s3​,s4​≤20

1≤A1​,A2​,…,As1​​,B1​,B2​,…,Bs2​​,C1​,C2​,…,Cs3​​,D1​,D2​,…,Ds4​​≤60。

 解题思路

这道题可以将题目转化为01背包问题后使用动态规划进行求解。

要想两个大脑同时开工，使得最后的解题时间最短，那么我们就得将所有题目分成两组，使得两组解题时间和最接近。那么两组的总时间肯定是越接近总时间的一半越好。那么我们假设右脑的容量为sum/2，然后只要尽可能的往其中放入题目，使得时间不超过sum/2即可。然后该科目所用的时间应该就是sum-dp[sum/2]，因为所用时间得按照用时最长的那一组计算。

这样这道题目就转化为了一个01背包问题。

代码

#include<cstdio>
#include<algorithm>
using namespace std;

int arr[4][25] = { 0 };  //其中0用于保存总时间
int dp[4][1500] = { 0 };

int main()
{
	int s[4];
	int ans = 0;
	scanf("%d%d%d%d", &s[0], &s[1], &s[2], &s[3]);
	for (int i = 0; i < 4; i++)
	{
		for (int j = 1; j <= s[i]; j++)
		{
			scanf("%d", &arr[i][j]);
			arr[i][0] += arr[i][j];
		}
		for (int j = 1; j <= s[i]; j++)                   //01背包问题模板
		{
			for (int k = arr[i][0] / 2; k >= arr[i][j]; k--)
			{
				dp[i][k] = max(dp[i][k], dp[i][k - arr[i][j]] + arr[i][j]);      //状态转移方程
			}
		}
		ans += (arr[i][0] - dp[i][arr[i][0] / 2]);
	}
	printf("%d", ans);
	return 0;
}