转化为分组背包 zoj 3769

时间:2022-08-26 00:09:23

题目链接:https://vjudge.net/problem/ZOJ-3769

题意:现在你要去打怪,你有13种装备,每件装备会有伤害和防御两种属性,一般来说,每种装备只可以装备一件,但是特别的,戒指(Finger)你可以同时装备两个,左右手各一个,然后对于“Two-Handed”类的装备,如果你装备这种装备,那么你就不可以装备"Shield", "Weapon"这两种,反之,如果你装备了"Shield", "Weapon"中的任意一种,那么你就不可以装备“Two-Handed”类的装备,现在需要我们求出在达到m点以上防御值的情况下可以达到的最大攻击值。如果无法达到m点防御值,则输出-1。

思路:这题可以转化为分组背包,但是又两个问题要解决,一个是Finger类装备,我们可以把Finger类的装备两两组合,加入Finger类里面,这样就把两个Finger类的装备转化为了一个,然后对于"Shield", "Weapon"这两类,我们也是可以两两组合加入“Two-Handed”类里面去的,当然,为了避免其中一种装备的数量为0,所以可以先将"Shield", "Weapon"这两类分别加入“Two-Handed”,然后在加入两两组合的。我一开始是用结构体数组储存的,一直超时,看了大佬博客,说要先处理数量多的那一类,这样可以节省时间,觉得有道理,但是用结构体数组改了还是一直超时(果然,大佬的代码和我的就是不一样),最后重写代码,改成和他们一样用vector储存就过了,不知道为啥。如果有谁知道的,可以提醒一下我,感谢感谢。

这里开二维数组,其中把防御看做背包容量,但是我们不知道背包容量的上限,题目只说了要大于等于m点防御值,所以我们把m以上的都看成m(为啥别人脑洞就这么大),由于在第k组拿装备的时候需要知道第k-1组的值,所以状态转移方程就是dp[k][i+w[j]]=max(dp[k][i+w[j]],dp[k-1][i]+v[j])。

代码:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<cmath>
#include<vector>
#include<set>
#include<cstdio>
#include<string>
#include<deque>
using namespace std;
typedef long long LL;
#define eps 1e-8
#define INF 0x3f3f3f3f
#define maxn 50005
/*struct point{
int u,w;
};
bool operator <(const point &s1,const point &s2)
{
if(s1.v!=s2.v)
return s1.v>s2.v;
else
return s1.u>s2.u;
}*/
struct node{
int v,w;
};
int n,m,k,t;
map<string,int>mp;
vector<node>ve[];
string ss[]={" ","Two-Handed","Finger", "Head", "Shoulder", "Neck", "Torso",
"Hand", "Wrist", "Waist", "Legs","Feet", "Shield", "Weapon"};
int dp[][maxn];
void init()
{
for(int i=;i<=;i++){//给装备编号
mp[ss[i]]=i;
}
}
int main()
{
init();
cin>>t;
while(t--)
{
cin>>n>>m;
for(int i=;i<=;i++){
ve[i].clear();
}
string s;
int w,v;
for(int i=;i<=n;i++){
cin>>s>>v>>w;
int id=mp[s];
ve[id].push_back((node){v,w});
}
memset(dp,-,sizeof(dp));//初始化所有状态都不可达
dp[][]=;//初始化 for(int i=;i<ve[].size();i++){//"Shield", "Weapon"合并到“Two-Handed”
ve[].push_back(ve[][i]);
}
int num1=ve[].size();
int num2=ve[].size();
for(int i=;i<num2;i++){
ve[].push_back(ve[][i]);
for(int j=;j<num1;j++){
ve[].push_back((node){ve[][i].v+ve[][j].v,ve[][i].w+ve[][j].w});
}
} num1=ve[].size();
for(int i=;i<num1;i++){//"Finger"合并
for(int j=i+;j<num1;j++){
ve[].push_back((node){ve[][i].v+ve[][j].v,ve[][i].w+ve[][j].w});
}
} for(int k=;k<=;k++){//枚举组
for(int i=;i<=m;i++){//枚举防御值
dp[k][i]=max(dp[k][i],dp[k-][i]);//现在状态的初值从前一状态来
if(dp[k-][i]==-)//如果前一状态不可达
continue;
for(int j=;j<ve[k].size();j++){//如果前一状态可以到达,那么可以在前一状态的基础上在当前组
//拿一件装备
int min1=min(m,i+ve[k][j].w);
dp[k][min1]=max(dp[k][min1],dp[k-][i]+ve[k][j].v);
}
}
}
cout<<dp[][m]<<endl;
}
return ;
}

结构体数组一直超时的代码:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<cmath>
#include<vector>
#include<set>
#include<cstdio>
#include<string>
#include<deque>
using namespace std;
typedef long long LL;
#define eps 1e-8
#define INF 0x3f3f3f3f
#define maxn 50005
/*struct point{
int u,w;
};
bool operator <(const point &s1,const point &s2)
{
if(s1.w!=s2.w)
return s1.w>s2.w;
else
return s1.u>s2.u;
}*/
map<string,int>mp;
int dp[][maxn];
int n,m,k,t;
struct node{
int num;
int w[];
int v[];
}zu[];
string s[]={" ", "Two-Handed"," ","Head", "Shoulder", "Neck", "Torso", "Hand", "Wrist", "Waist",
"Legs", "Feet", "Finger", "Shield", "Weapon"};
void init()
{
for(int i=;i<=;i++)
{
mp[s[i]]=i;
}
}
void combine_finger()
{
int c=zu[].num;
if(c==)
return;
if(c==)
{
zu[].num++;
zu[].w[]=zu[].w[];
zu[].v[]=zu[].v[];
return;
}
for(int i=;i<c;i++){
for(int j=i+;j<=c;j++){
zu[].num++;
int k=zu[].num;
zu[].w[k]=zu[].w[i]+zu[].w[j];
zu[].v[k]=zu[].v[i]+zu[].v[j];
}
}
return;
}
void combine_two()
{
int a=zu[].num;
int b=zu[].num;
for(int i=min(a,);i<=a;i++){
for(int j=min(b,);j<=b;j++){
zu[].num++;
int c=zu[].num;
zu[].w[c]=zu[].w[i]+zu[].w[j];
zu[].v[c]=zu[].v[i]+zu[].v[j];
}
}
return;
}
int main()
{
init();
scanf("%d",&t);
while(t--)
{
memset(dp,,sizeof(dp));
scanf("%d%d",&n,&m);
string ss,w,v;
int a,b;
for(int i=;i<=n;i++){
cin>>ss>>a>>b;
int id=mp[ss];
zu[id].num++;
int c=zu[id].num;
zu[id].v[c]=a;
zu[id].w[c]=b;
}
combine_finger();
combine_two();
memset(dp,-,sizeof(dp));
dp[][]=;
for(int k=;k<=;k++){
for(int i=;i<=m;i++){
dp[k][i]=max(dp[k][i],dp[k-][i]);
if(dp[k-][i]==-)
continue;
for(int j=;j<=zu[k].num;j++){
int min1=min(m,i+zu[k].w[j]);
dp[k][min1]=max(dp[k][min1],dp[k-][i]+zu[k].v[j]);
}
}
}
cout<<dp[][m]<<endl;
}
return ;
}