动态规划--“破锣摇滚”乐队

题目描述

你刚刚继承了流行的“破锣摇滚”乐队录制的尚未发表的N(1 <= N <= 20)首歌的版权。你打算从中精选一些歌曲，发行M(1 <= M <= 20)张CD。每一张CD最多可以容纳T(1 <= T <= 20)分钟的音乐，一首歌不能分装在两张CD中。
不巧你是一位古典音乐迷，不懂如何判定这些歌的艺术价值。于是你决定根据以下标准进行选择：
歌曲必须按照创作的时间顺序在CD盘上出现。选中的歌曲数目尽可能地多。

输入格式

第一行： 三个整数：N, T, M.
第二行： N个整数，分别表示每首歌的长度，按创作时间顺序排列。

输出格式

一个整数，表示可以装进M张CD盘的乐曲的最大数目。

样例输入

4 5 2
4 3 4 2

样例输出

3

解题思路

依旧是动态规划。 首先，我们定义一个3维的f数组，f[i][j][k]的意义是：截至第i首歌，用到了第j张CD，剩余时间为k时能装的歌曲数目。

状态转移方程

重点来了！！！
这里有3种情况：
(1):不选第i首歌，即f[i][j][k]=f[i-1][j][k];
(2):选第i首歌，并且第i首歌和其他歌一起装在第j张CD中，即f[i][j][k]=f[i-1][j][k-s[i]]+1;
(3):选第i首歌，并且第i首歌"独占"第j张CD，即f[i][j][k]=f[i-1][j-1][t];
3者区最大者即可
实现成代码:

if(k<s[i]) f[i][j][k]=f[i-1][j][k];
else f[i][j][k]=max(f[i][j][k],max(f[i-1][j][k-s[i]]+1,f[i-1][j-1][t]+1));

初始化

因为本蒟蒻把f数组定义在main函数外了，所以就不用初始化为0了。

循环次数

一共三层循环
i:1~n;
j:m~1;
k:0~t-1;

完整代码

#include<bits/stdc++.h>
using namespace std;
int a[25],f[25][25];
int main()
{
	int n,t,m;
	scanf("%d %d %d",&n,&t,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=m;j>=1;j--)
		{
			for(int k=0;k<=t;k++)
			{
				if(a[i]>k) f[j+1][t-a[i]]=max(f[j+1][t-a[i]],f[j][k]+1);
				else f[j][k-a[i]]=max(f[j][k-a[i]],f[j][k]+1);
			}
		}
	}
	printf("%d",*max_element(f[m],f[m]+t+1));
	return 0;
}

如果觉得本蒟蒻写得还可以的话，请点一个赞👍哟，谢谢啦~