C - Om Nom and Candies

思路:贪心+思维(或者叫数学)。假设最大值max(wr,wb)为wr,当c/wr小于√c时,可以枚举r糖的数量(从0到c/wr),更新答案,复杂度√c;否则,假设hr/wr<hb/wr,得到hr*wb<hb*wr,由这个等式可知,在有wb*wr重量限制的情况下,买wb个r糖没有买wr个b糖划算,当需要买超过wb个r糖时,不如去买b糖,可以枚举r糖的数量(从0到wb-1),更新答案,复杂度√c。

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+;
const int INF=0x3f3f3f3f;
int main()
{
ll c,hr,hb,wr,wb;
cin>>c>>hr>>hb>>wr>>wb;
if(wr<=wb)swap(wr,wb),swap(hr,hb);
ll n=c/wr;
ll ans=;
if(n<=sqrt(c))
{
for(ll i=;i<=n;i++)
{
ll temp=c-i*wr;
ll j=temp/wb;
ans=max(ans,(ll)i*hr+(ll)j*hb);
}
}
else
{
if(hr*wb>=hb*wr)swap(wr,wb),swap(hr,hb);
for(ll i=;i<wb;i++)
{
ll temp=c-i*wr;
ll j=temp/wb;
ans=max(ans,i*hr+j*hb);
}
}
cout<<ans<<endl;
return ;
}
05-11 22:24