[01背包+分组背包] hdu6968

题目

题目
有$n$个科目，$m$本书，学习一本书可以让这门课期末分数增加$s$分，但是需要$d$天来学习。
现在一共有$t$天，最多能够挂$p$门课，求最高分

思路

先对每门科目进行$01$背包，$dp[i][j]$表示第$i$门课使用$j$天获得的分数，然后对每门科目的结果进行分组背包，$ans[i][j][p]$表示到第$i$门课一共学习了$j$天，挂了$p$门，最后得出结果

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define ll long long
using namespace std;
const int INF = 0x3f3f3f3f;
string s[60];
int dp[60][510];
int ans[60][510][4];
int cnt[60];
struct sut{
	int sc,da;
}a[60][15010];
int main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int t;
	cin>>t;
	while(t--){
		int n;
		cin>>n;
		memset(dp,-INF,sizeof dp);
		memset(ans,-INF,sizeof ans);
		memset(cnt,0,sizeof cnt); 
		map<string,int> mp;
		for(int i=1;i<=n;i++){
			cin>>s[i];
			mp[s[i]]=i;
		}
		int m;
		cin>>m;
		for(int i=1;i<=m;i++){
			string s;
			cin>>s;
			++cnt[mp[s]];
			cin>>a[mp[s]][cnt[mp[s]]].sc>>a[mp[s]][cnt[mp[s]]].da;	
		}
		int t,p;
		cin>>t>>p;
		int maxn=0,num=0;
		for(int i=1;i<=n;i++){
			dp[i][0]=0;
			for(int j=1;j<=cnt[i];j++){
				for(int k=t;k>=a[i][j].da;k--){
					dp[i][k]=max(dp[i][k],dp[i][k-a[i][j].da]+a[i][j].sc);
					dp[i][k]=min(dp[i][k],100);
				}
			}
		}
		bool flag=0;
		ans[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 pp=0;pp<=p;pp++){
						if(pp&&dp[i][j]<60&&dp[i][j]>=0) ans[i][k][pp]=max(ans[i][k][pp],ans[i-1][k-j][pp-1]+dp[i][j]);
						else if(dp[i][j]>=60){
							ans[i][k][pp]=max(ans[i][k][pp],ans[i-1][k-j][pp]+dp[i][j]);
						}
					}
					
				}
			}
		}
		maxn =-1;
		for(int i=0;i<=p;i++){
			for(int j=1;j<=t;j++){
				maxn= max(maxn,ans[n][j][i]);
			}
		}
		cout<<maxn<<endl;
	}
    return 0;
}