题目大意:有N个站点的轻轨站,有一个容量为C的列车起点在1号站点,终点在N号站点,有K组牛群,每组数量为Mi(1≤Mi≤N),行程起点和终点分别为Si和Ei(1≤Si<Ei≤N)。计算最多有多少头牛可以搭乘轻轨。
一道经典的贪心题目,每当一头牛上车的时候,如果超载,我们就优先踢出去行程终点比较远的那部分牛
而 踢出哪些行程终点较远的牛 以及 哪些在车上的牛在这站到达了终点,都可以用优先队列来维护,复杂度约为
贴上代码:
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cstring>
#define N 50005
#define ll long long
using namespace std;
int k,n,c,ans;
int sum[N];
struct COW{
int x,y,w;
}cow[N];
int cmp(COW s1,COW s2) {return s1.x<s2.x;}
struct node1{
int id;
friend bool operator<(const node1 &s1,const node1 &s2)
{return cow[s1.id].y > cow[s2.id].y;}
};
node1 ins1(int idd){node1 w;w.id=idd;return w;}
struct node2{
int id;
friend bool operator<(const node2 &s1,const node2 &s2)
{return cow[s1.id].y < cow[s2.id].y;}
};
node2 ins2(int idd){node2 w;w.id=idd;return w;}
int solve()
{
sort(cow+1,cow+k+1,cmp);
priority_queue<node1>q1;
priority_queue<node2>q2;
int j=1,cnt=0;
for(int i=1;i<=n;i++)
{
while(!q1.empty())
{
node1 p=q1.top();
if(cow[p.id].y==i)
{
q1.pop();
ans+=sum[p.id];
cnt-=sum[p.id];
sum[p.id]=0;
}else{
break;
}
}
while(cow[j].x==i)
{
q1.push(ins1(j));
q2.push(ins2(j));
cnt+=cow[j].w;
sum[j]=cow[j].w;
j++;
}
while(cnt>c)
{
node2 p=q2.top();
if(cnt-sum[p.id]>c)
{
q2.pop();
cnt-=sum[p.id];
sum[p.id]=0;
}else{
sum[p.id]-=(cnt-c);
cnt=c;
}
}
}
return ans;
}
int main()
{
scanf("%d%d%d",&k,&n,&c);
for(int i=1;i<=k;i++) scanf("%d%d%d",&cow[i].x,&cow[i].y,&cow[i].w);
printf("%d",solve());
return 0;
}