[USACO 2009 Feb Gold] Fair Shuttle (贪心+优先队列)

题目大意：有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+,cow+k+,cmp);
     priority_queue<node1>q1;
     priority_queue<node2>q2;
     int j=,cnt=;
     for(int i=;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]=;
             }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]=;
             }else{
                 sum[p.id]-=(cnt-c);
                 cnt=c;
             }
         }
     }
     return ans;
 }

 int main()
 {
     scanf("%d%d%d",&k,&n,&c);
     for(int i=;i<=k;i++) scanf("%d%d%d",&cow[i].x,&cow[i].y,&cow[i].w);
     printf("%d",solve());
     return ;
 }