I love exam HDU - 6968（01背包的应用）

 

 

题目链接：2021“MINIEYE杯”中国大学生算法设计超级联赛（2） - Virtual Judge (vjudge.net) 

题目大意：给我们n门课程的名称，然后m本复习材料，看完一本复习材料需要花费一定的天数，看完之后该门课能提高一定的分数，初始情况下所有科目的得分均为0，现在我们总共有t天时间来复习，最多挂科p门，问在满足挂科数<=p的前提下所有课程得分总和的最高分是多少，如果不能满足条件就输出-1。

解题思路：对于每一门课程做一个01背包，得出课程i花费j天最多能得到的分数是多少cost[i][j],然后再对所有课程统一做一次01背包，f[i][j][k]表示前i门课程，总共花费j天时间，挂了k门科所能得到的最大分数，具体递归公式见代码，说明一个细节，就是f数组一开始的初始状态是f[0][0][0]=0,其他所有状态都是负无穷，也就是说其他所有状态都必须得从当前初始状态推出并计算，一开始我就把f数组全部初始化为0，结果WA了，还有就是计算f数组的4重循环，后面的三重循环是可以任意调换的，所以可以有不同的写法。

上代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
int cost[60][510],f[60][510][10];
int t,n,m,p;
struct node{
	int day;
	int score;
};
map<string,int>mp;
vector<node>mt[60]; 
int main()
{
	int T;
	cin>>T;
	while(T--)
	{
		cin>>n;
		memset(cost,0,sizeof cost);
		memset(f,-0x3f3f3f3f,sizeof f);
		for(int i=1;i<=n;i++)
		{
			string str;
			cin>>str;
			mp[str]=i;
			mt[i].clear();
		}
		cin>>m;
		for(int i=1;i<=m;i++)
		{
			string str;
			int d,s;
			cin>>str>>s>>d;
			mt[mp[str]].push_back({d,s});
		}
		cin>>t>>p;
		//对每个课程做一次01背包 
		for(int i=1;i<=n;i++)
			for(int j=0;j<mt[i].size();j++)
				for(int k=t;k>=mt[i][j].day;k--)
				{
					cost[i][k]=max(cost[i][k],cost[i][k-mt[i][j].day]+mt[i][j].score);
					cost[i][k]=min(cost[i][k],100);
				}
		f[0][0][0]=0;
		for(int i=1;i<=n;i++)//枚举课程 
			for(int j=t;j>=0;j--)//枚举该门课程的用时 
				for(int k=t;k>=j;k--)//枚举总时间 
					for(int g=0;g<=p;g++)//枚举挂科数量 
					 if(cost[i][j]>=60)
					 f[i][k][g]=max(f[i][k][g],f[i-1][k-j][g]+cost[i][j]);
					 else if(g)
					 f[i][k][g]=max(f[i][k][g],f[i-1][k-j][g-1]+cost[i][j]);
		int ans=-1;
		for(int i=0;i<=p;i++)
		ans=max(ans,f[n][t][i]);
		cout<<ans<<endl;
	}
	return 0;
}