bzoj 5102: [POI2018]Prawnicy 贪心+堆

题意

定义一个区间(l,r)的长度为r-l,空区间的长度为0。
给定数轴上n个区间,请选择其中恰好k个区间,使得交集的长度最大。
n<=1000000

分析

为什么同为一个log的算法跑出来的差别就这么大呢?
看完题就想到了一个线段树做法,打完后发现被卡常了,然后就果断改成了堆。
具体来讲就是枚举左端点,那么最优的右端点显然是把所有左端点不大于该点的区间加入,然后取第k大的右端点。
那么我们只要把区间按照左端点排序,然后用一个堆来维护即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>

const int N=1000005;

int n,m;
struct data{int l,r,id;}a[N];
std::priority_queue<int> que;

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

bool cmp(data a,data b)
{
    return a.l<b.l;
}

int main()
{
    n=read();m=read();
    for (int i=1;i<=n;i++) a[i].l=read(),a[i].r=read(),a[i].id=i;
    std::sort(a+1,a+n+1,cmp);
    int ans=0,a1,a2;
    for (int i=1;i<=n;i++)
    {
        que.push(-a[i].r);
        if (que.size()>m) que.pop();
        if (que.size()==m&&-que.top()-a[i].l>ans) ans=-que.top()-a[i].l,a1=a[i].l,a2=-que.top();
    }
    printf("%d\n",ans);
    for (int i=1;i<=n;i++)
        if (a[i].l<=a1&&a[i].r>=a2)
        {
            printf("%d ",a[i].id);m--;
            if (!m) break;
        }
    return 0;
}
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_33229466/article/details/79951058
个人分类: 贪心
想对作者说点什么? 我来说一句

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

关闭
关闭
关闭