【CodeForces】671 B. Robin Hood

本文针对RobinHood问题,提出了一种高效的解决方法。通过排序和二分查找技巧,实现对序列的最大值减一和最小值加一的操作,直至达到指定次数k。此方法能够快速计算出序列的最终极差。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

【题目】B. Robin Hood

【题意】给定n个数字的序列和k次操作,每次将序列中最大的数-1,然后将序列中最小的数+1,求最终序列极差。n<=5*10^5,0<=k<=10^9,1<=ai<=10^9。

【算法】模拟

【题解】关键在于,增加和减少可以分开操作。

将数列排序,从小到大增加前面若干个数到同一个数直到k次,再从大到小减少后面若干个数到同一个数直到k次,就可以得到结果。

也可以不排序直接二分”同一个数”找到恰好k次的位置。

#include<cstdio>
#include<algorithm>
int n,k,a[500010];
void solve(){
    std::sort(a+1,a+n+1);
    int t=0,b=0,s=k;
    while(t<n&&1ll*t*(a[t+1]-b)<=s)s-=1ll*t*(a[t+1]-b),b=a[++t];
    b=s/t;s%=t;
    for(int i=1;i<=t;i++)a[i]=a[t]+b+(i>t-s);
    for(int i=1;i<=n;i++)a[i]=-a[i];
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    solve();solve();
    printf("%d",a[1]-a[n]);
    return 0;
}
View Code

 

转载于:https://www.cnblogs.com/onioncyc/p/8067989.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值