【CodeForces】671 B. Robin Hood

最新推荐文章于 2025-08-13 10:25:05 发布

weixin_33736048

最新推荐文章于 2025-08-13 10:25:05 发布

阅读量168

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/onioncyc/p/8067989.html

本文针对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;
}