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

题目大意:有N个站点的轻轨站,有一个容量为C的列车起点在1号站点,终点在N号站点,有K组牛群,每组数量为Mi(1≤Mi≤N),行程起点和终点分别为Si和Ei(1≤Si<Ei≤N)。计算最多有多少头牛可以搭乘轻轨。

一道经典的贪心题目,每当一头牛上车的时候,如果超载,我们就优先踢出去行程终点比较远的那部分牛

而 踢出哪些行程终点较远的牛 以及 哪些在车上的牛在这站到达了终点,都可以用优先队列来维护,复杂度约为O(klogn)

贴上代码:

#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;
}

 

阅读更多
文章标签: 贪心 优先队列
个人分类: usaco
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭