poj 3622 Gourmet Grazers

最新推荐文章于 2021-07-16 16:34:56 发布

原创最新推荐文章于 2021-07-16 16:34:56 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

线段树专栏收录该内容

22 篇文章

订阅专栏

本文介绍了一种利用贪心算法解决特定牧草分配问题的方法。通过对牧草按成本排序并使用线段树来查找满足条件的最大值，实现了快速求解。代码示例展示了如何通过二分查找和线段树维护来实现这一过程。

我的做法，首先对于这样的数据范围应该想到不能去套用模型，只能往贪心什么的靠，容易证明一个结论:

每次选择需要b最大的cow，然后从满足的grass中选择一个最便宜的，这样肯定能够得出最优解。

问题转化成了怎么去快速的得出这个数，我的做法是先对grass按照a排序，那么就能二分找到那些a满足条件的grass，那么现在需要的就是从那些grass中找出b满足要求的并且最靠近左边的，这个用线段树来保存最大值，然后就能询问每个区间的最大值，那么就能二分的找到这个值。当然找到了之后还要记得把这个值给去除，我的做法是把值给成-1.

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define ls (t<<1)
#define rs (t<<1|1)
#define midt (tr[t].l+tr[t].r>>1)
using namespace std;
const int maxn=1e5+9;
struct
{
    int l,r,max;
}tr[maxn<<2];
struct D
{
    int a,b;
    bool operator <(const D & xx) const
    {
        return b>xx.b;
    }
}cow[maxn];
struct T
{
    int a,b;
    bool operator <(const T &xx) const
    {
        return a<xx.a;
    }
}grass[maxn];

void maketree(int t,int l,int r)
{
    tr[t].l=l;
    tr[t].r=r;
    if(l==r)
    {
        tr[t].max=grass[l].b;
        return ;
    }
    int mid=midt;
    maketree(ls,l,mid);
    maketree(rs,mid+1,r);
    tr[t].max=max(tr[ls].max,tr[rs].max);
}

void modify(int t,int x)
{
    if(tr[t].l==x&&tr[t].r==x)
    {
        tr[t].max=-1;
        return ;
    }
    int mid=midt;
    if(x<=mid) modify(ls,x);
    else modify(rs,x);
    tr[t].max=max(tr[ls].max,tr[rs].max);
}

int query(int t,int l,int r)
{
    if(tr[t].l==l&&tr[t].r==r)
    return tr[t].max;
    int mid=midt;
    if(r<=mid)
    return query(ls,l,r);
    else if(mid+1<=l)
    return query(rs,l,r);
    else
    {
        int ret=query(ls,l,mid);
        ret=max(ret,query(rs,mid+1,r));
        return ret;
    }
}

int main()
{
    int n,m;
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;i++)
        scanf("%d %d",&cow[i].a,&cow[i].b);
        for(int i=1;i<=m;i++)
        scanf("%d %d",&grass[i].a,&grass[i].b);
        sort(cow+1,cow+1+n);
        sort(grass+1,grass+1+m);
        maketree(1,1,m);
        long long ans=0;
        bool flag=false;
        for(int k=1;k<=n;k++)
        {
            if(grass[m].a<cow[k].a)
            {
                flag=true;
                break;
            }
            int st=1,ed=m,mid;
            while(st<ed)
            {
                mid=st+ed>>1;
                if(grass[mid].a>=cow[k].a) ed=mid;
                else st=mid+1;
            }
            if(query(1,st,m)<cow[k].b)
            {
                flag=true;
                break;
            }
            ed=m;
            while(st<ed)
            {
                mid=st+ed>>1;
                if(query(1,st,mid)>=cow[k].b) ed=mid;
                else st=mid+1;
            }
            ans+=grass[st].a;
            modify(1,st);
        }
        if(!flag)
        cout<<ans<<endl;
        else
        printf("-1\n");
    }
    return 0;
}