Codeforces GYM 100753J: Souvenirs 题解

这题是显然的dp

设dp[i][j][k]为考虑到第i个商人（第i个商人已经处理完），当前有j个金币和k个银币时，最多能买到多少纪念品

1.不买第i个纪念品，dp[i][j][k]=max(dp[i][j][k],dp[i-1][j][k])

2.用金币买第i个纪念品，dp[i][j][k]=max(dp[i][j][k],dp[i-1][j+1][k-get*pack[i]]+1)

get指的是通过商人的种类算出的商人会给你的钱袋数,pack[i]表示第i个商人一个钱袋里装多少银币

当k-get*pack[i]>=price[i]且第i个商人是generous时，不能用金币购买

3.用银币买第i个纪念品，dp[i][j][k]=max(dp[i][j][k],dp[i-1][j][k+price[i]]+1)

注意数组要滚动

我做这题的时候总是TLE，后来发现在循环里判断s[i]=="generous","greedy","honest"非常慢，因为是字符串的比较

所以提前把类型转换成数字后判断效率提升8倍左右#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cstdlib>
#include <utility>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <map>
#include <vector>
#include <cmath>
#define Pair pair<int,int>
#define LOWBIT(x) x & (-x)
#define LL long long
#define mp make_pair
#define pb push_back
#define x first
#define y second
using namespace std;

const int MOD=1e9+7;
const int INF=0x7ffffff;
const int magic=348;
const double eps=1e-9;

int g,c,n;
string s[101];
int type[101];
int pack[101],price[101];

int dp[2][101][20148];

int main ()
{
int i,j,k;
scanf("%d%d%d",&g,&c,&n);
for (i=1;i<=n;i++)
{
cin>>s[i]>>pack[i]>>price[i];
if (s[i]=="generous") type[i]=1;
if (s[i]=="greedy") type[i]=2;
if (s[i]=="honest") type[i]=3;
}
for (i=0;i<=1;i++)
for (j=0;j<=c;j++)
for (k=0;k<=20100;k++)
dp[i][j][k]=-INF;
dp[0][c][0]=0;
int p1=0,p2=1;
int remain,get;
double t;
int ans=0;
for (i=1;i<=n;i++)
{
for (j=0;j<=c;j++)
for (k=0;k<=min(i*100,10010);k++)
{
dp[p2][j][k]=dp[p1][j][k];
/*if (k+price[i]<=20100)
{
dp[p2][j][k]=max(dp[p2][j][k],dp[p1][j][k+price[i]]+1);
if (type[i]==1 && dp[p1][j][k+price[i]]>=0)
{
if (i==n) ans=max(ans,dp[p2][j][k]);
continue;
}
}*/
remain=g-price[i];
t=remain*1.0/pack[i];
if (type[i]==2) get=int(t);
if (type[i]==3) get=int(t+0.5);
if (type[i]==1)
{
get=int(t+1);
if (t==int(t)) get--;
}
if (j+1<=c && k-get*pack[i]>=0)
if (!(type[i]==1 && k-get*pack[i]>=price[i]))dp[p2][j][k]=max(dp[p2][j][k],dp[p1][j+1][k-get*pack[i]]+1);
if (k+price[i]<=min(i*100,10010))
dp[p2][j][k]=max(dp[p2][j][k],dp[p1][j][k+price[i]]+1);
if (i==n) ans=max(ans,dp[p2][j][k]);
}
p1^=1;p2^=1;
}
printf("%d\n",ans);
return 0;
}