bzoj 1577: [Usaco2009 Feb]庙会捷运Fair Shuttle——小根堆+大根堆+贪心
Description
公交车一共经过N(1<=N<=20000)个站点,从站点1一直驶到站点N。K(1<=K<=50000)群奶牛希望搭乘这辆公交车。第i群牛一共有Mi(1<=Mi<=N)只.
他们希望从Si到Ei去。
公交车只能座C(1<=C<=100)只奶牛。而且不走重复路线,请计算这辆车最多能满足多少奶牛听要求。
注意:对于每一群奶牛,可以部分满足,也可以全部满足,也可以全部不满足。
Input
第1行: 三个整数: K,N,C。 由空格隔开。
第2..K+1行:第i+1行,告诉你第i组奶牛的信息: S_i, E_i and M_i。由空格隔开。
Output
一行:可以在庙会乘坐捷运的牛的最大头数
Sample Input
8 15 3
1 5 2
13 14 1
5 8 3
8 14 2
14 15 1
9 12 1
12 15 2
4 6 1
1 5 2
13 14 1
5 8 3
8 14 2
14 15 1
9 12 1
12 15 2
4 6 1
Sample Output
10
HINT
捷运可以把2头奶牛从展台1送到展台5,3头奶牛从展台5到展台8,
2头奶牛从展台8 到展台14,1头奶牛从展台9送到展台12,
一头奶牛从展台13送到展台14, 一头奶牛从 14送到15。
——————————————————————————————
这道题我们维护两个堆 按终点距离为键值 维护一个大根堆 一个小根堆
我们每次到一个点 就把所有到这个点(作为终点)的奶牛扔出堆(大根堆小根堆的信息要互通)
表示他们已经下车了 到一个点之后把所有在这个点上车的牛全部扔进堆 表示他们要上车辣QAQ
如果这个时候车上的人太多了(会出事的2333) 我们就贪心地把终点最远的扔出堆
这样就可以得到最优解了2333
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
#include<cstdio> #include<cstring> #include<algorithm> #include<vector> #include<queue> const int M=3e4+7; int read(){ int ans=0,f=1,c=getchar(); while(c<'0'||c>'9'){if(c=='-') f=-1; c=getchar();} while(c>='0'&&c<='9'){ans=ans*10+(c-'0'); c=getchar();} return ans*f; } int f[2*M],cnt,cost,k,n,c,ans; struct pos{ int id,h,ed; bool operator <(const pos &x)const{return x.ed<ed;} }; std::vector<pos>e[M]; std::priority_queue<pos>q1; struct Q{ int id,h,ed; bool operator <(const Q &x)const{return x.ed>ed;} }; std::priority_queue<Q>q2; int main(){ int v,x,y; k=read(); n=read(); c=read(); for(int i=1;i<=k;i++) x=read(),y=read(),v=read(),e[x].push_back((pos){++cnt,v,y}); for(int i=1;i<=n;i++){ while(!q1.empty()){ pos x=q1.top(); if(x.ed>i) break; q1.pop(); if(f[x.id]) continue; f[x.id]=1; cost-=x.h; ans+=x.h; } int mx=e[i].size(); for(int j=0;j<mx;j++){ q1.push((pos){e[i][j].id,e[i][j].h,e[i][j].ed}); q2.push((Q){e[i][j].id,e[i][j].h,e[i][j].ed}); cost+=e[i][j].h; } while(cost>c){ Q x=q2.top(); q2.pop(); if(f[x.id]) continue; f[x.id]=1; if(cost-c>=x.h){cost-=x.h; continue;} int now=cost-c; cnt++; q1.push((pos){cnt,x.h-now,x.ed}); q2.push((Q){cnt,x.h-now,x.ed}); cost=c; } } printf("%d\n",ans); return 0; }