简介
原网址:http://ybt.ssoier.cn:8088/problem_show.php?pid=1673
据说这道题是池塘钓鱼的升级版, O ( n 2 log n ) O(n^2\log n) O(n2logn)的算法无法在这道题通过
思路
这道题的数据范围:
n
,
h
i
≤
1
0
6
n,h_i\le10^6
n,hi≤106 。
看看这么庞大的数据,只能用
O
(
n
log
n
)
O(n \log n)
O(nlogn)复杂度去解决这个庞大的数据。
已知一个序列取出一个最小值序列平均复杂度为
O
(
log
n
)
O(\log n)
O(logn),
从前向后枚举时间复杂度为
O
(
n
)
O(n)
O(n)。
所以最终复杂度是
O
(
n
log
n
)
O(n \log n)
O(nlogn)。
这道题有一个噱头:这栋楼的楼层和答案并没有多大关系。因为一栋楼中每层的价值都一样。所以我们可以把这栋楼变成一个有
n
n
n个宝箱的一个楼,每上一层就相当于取一个宝箱,所以问题被化简成有k个时间,可以取的宝箱的最大价值是多少。
如果可以知道最远到哪栋楼,便可以排序来取出前几大的宝箱。所以我们可以尝试枚举最远到哪一栋楼
这时,我们就用不了排序了,它的时间消耗是在是太可怕了。
所以我们要改进一下取出优先值的策略,我们可以建一个堆,里面存储要取的所有的宝箱在之前,我们已经拿走了从每一栋楼拿走了一个宝箱,所以这栋楼只剩下
(
h
i
−
1
)
(h_i-1)
(hi−1)个宝箱。我们与堆里的最小值比较,如果最小值小于这栋楼的值,则弹出最小值放入宝箱中,如果弹无可弹,就把新的宝箱尽可能地放入堆中。
但是,
M
M
M的数据范围是
M
≤
1
0
9
M\le10^9
M≤109 ! 堆可吃不下这庞然大物。但是,考虑到堆内重复的数据极多,所以我们可以建立一个类来存下它们,就像这样:
class value{
public:
int h;
int v;
value(){return;}
value(const int &h,const int &v):h(t),v(v){return;}
bool operator<(const value &a)const{return this->v>a.v;}
};
通过这个运算符重载,我们可以轻松骗过 STL。
我们保留这个堆的同时,计算目前的价值,同最大值进行比较,目的是求出最大的值。
最终最大值就是我们想要的答案。
代码
这里是题解的最终部分了
#include<bits/stdc++.h>
using namespace std;
char frd[1048576],*ihead(frd+1048576);const char *itail(ihead);
inline char nxtChar(void){
if(ihead==itail)fread(frd,1,1048576,stdin),ihead=frd;
return *ihead++;
}
//此程序仅适用于整型
template<typename T>
inline void get(T &res){
T f(1);char c;res=0;
while(!isdigit(c=nxtChar()))if(c=='-')f=-1;
res=c&15;
while(isdigit(c=nxtChar()))res=(res<<1)+(res<<3)+(c&15);
res*=f;
}
//以上为快读
class value{
public:
long long h;
long long v;
value(){return;}
value(const long long &h,const long long &v):h(h),v(v){return;}
bool operator>(const value &a)const{return this->v>a.v;}
};
priority_queue<value,vector<value>,greater<value> > heap;
//以上为数据结构
int main(){
long long n,k,ans(0),res(0),tot_v(0),tot_h(0);
get(n),get(k),++k;
for(register int i(1);i<=n;++i){
long long h,v;
get(h),get(v),--k,res+=v;
if(!k)break;
if(h>1){
value tmp(h-1,v);
heap.push(tmp);
tot_h+=tmp.h;
tot_v+=tmp.h*tmp.v;
while(!heap.empty()&&tot_h-heap.top().h>=k){
tot_h-=heap.top().h,tot_v-=heap.top().v*heap.top().h;
heap.pop();
}//可以捞到的总油水。![在这里插入图片描述](https://img-blog.csdnimg.cn/20210719220444440.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L01pbmVjcmFmdF8zMTQx,size_16,color_FFFFFF,t_70#pic_center)
}
if(ans<(tot_h>=k?res+tot_v+(k-tot_h)*heap.top().v:tot_v+res))
ans=tot_h>=k?res+tot_v+(k-tot_h)*heap.top().v:tot_v+res;
}
printf("%lld\n",ans);
return 0;
}
附一张P过的截图吧(作者小心翼翼地吧个人信息删去了呢)