中位数问题 选m个数,使其和不超过v,并且要使其中位数最大

思路:
分奇数和偶数:
奇数的时候:遍历i,i是中位数,看是否满足条件,维护两个数组,一个是i之前有没有超过m/2个数,一个是i之后有没有超过m/2个数
偶数的时候:你会发现必须选两个挨着的数
处理和奇数一样即可

代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn=100000*4+100;
struct node
{
    long long int w;
    long long int v;
    friend bool operator < (node aa,node bb)
    {
        if(aa.w==bb.w)
            return aa.v<bb.v;
        return aa.w<bb.w;
    }
}a[maxn*2+100];
long long dp1[maxn],dp2[maxn];
int main ()
{
    int v,n,m;
    scanf("%d%d%d",&v,&n,&m);
    for(int i=0;i<n;i++)
    {
        scanf("%lld%lld",&a[i].w,&a[i].v);
    }
    sort(a,a+n);
    if(m%2)
    {
        int c=m/2;
        long long int sum=0;
        priority_queue<long long >q;
        for(int i=0;i<n;i++)
        {
            dp1[i]=sum;
            sum+=a[i].v*1LL;
            q.push(a[i].v);
            if(q.size()>c)
            {
                sum-=q.top();
                q.pop();
            }
        }
        while(!q.empty())
            q.pop();
        sum=0;
        for(int i=n-1;i>=0;i--)
        {
            dp2[i]=sum;
            sum+=a[i].v*1LL;
            q.push(a[i].v);
            if(q.size()>c)
            {
                sum-=q.top();
                q.pop();
            }
        }
        long long int maxx=-1;
        for(int i=c;i<=n-c-1;i++)
        {
            if(dp1[i]+dp2[i]+a[i].v*1LL<=v)
            {
                maxx=max(maxx,a[i].w);
            }
        }
        printf("%lld\n",maxx);
    }
    else
    {
        int c=m/2;
        long long int sum=0;
        priority_queue<long long>q;
        for(int i=0;i<n;i++)
        {
            sum+=a[i].v*1LL;
            q.push(a[i].v);
            if(q.size()>c)
            {
                sum-=q.top();
                q.pop();
            }
            dp1[i]=sum;
        }
        while(!q.empty())
            q.pop();
        sum=0;
        for(int i=n-1;i>=0;i--)
        {
            sum+=a[i].v*1LL;
            q.push(a[i].v);
            if(q.size()>c)
            {
                sum-=q.top();
                q.pop();
            }
            dp2[i]=sum;
        }
        long long maxx=-1;
        for(int i=c-1;i<=n-c-1;i++)
        {
            if(dp1[i]+dp2[i+1]<=v)
            {
                maxx=max(maxx,(a[i].w+a[i+1].w)/2);
            }
        }
        printf("%lld\n",maxx);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值