Educational Codeforces Round 120 (Rated for Div. 2) C. Set or Decrease

算法：排序 前缀和 二分

链接：Problem - C - Codeforces

时间复杂度：O（nlogn） 快排（nlogn） 二分（logn） check函数O（n）

题意简介：给定一个数组n个数 a1到an 和一个 k

每次操作你可以 将一个数减一 ai=ai-1 或 选择两个数 让一个数等于另一个数 ai=aj

询问最少的操作次数 可以将数组之和小于等于k sum<=k

思路：对于最小操作次数的问题 我们可以使用二分 来寻找 对于每次操作 如何使得结果最优呢

假设我们目前的操作次数为m 我们可以选择先让最小数自减x次 然后 将最大的m-x个数变成最小的那个数min 我们选择用后缀和来记录这m-x个数的和 这m-x个数的贡献为 这m-x个数的和减去m-x个最小数min 再加上我们让最小的数减去的x 即本次操作的贡献

我们目前的操作次数为m 我们可以遍历后m个数（如果m>=n则 先让最小的数mix减去m-n+1 然后操作次数m变成n-1 ）每次循环让变成最小数的个数往后推一个 但是相应的 我们可以对最小数自减的次数多了 每次让最小的数min进行自减 每次记录一下当前的贡献 并取一个最大值 当所有数的总和减去当前操作次数下的最大贡献 即当前操作次数下的最小数组和 如果最小数组和<=k 则合法

如果m>=n则 先让最小的数mix减去m-n+1 然后操作次数m变成n-1 

二分的区间无脑0到1e18

#include<bits/stdc++.h>
#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
using namespace std;using ll = long long;using PLL = pair<ll,ll>;
const ll MAX = 1e18;const ll MIN = -1e18;const ll INF=0x3f3f3f3f;
const ll Q = 2e5+9;const ll MOD = 1e9 + 7;
ll a[Q],b[Q];ll n,k;ll sum=0;
bool check(ll m){
     ll mi=a[1];ll ans=m;
     if(m>=n) mi-=m-n+1,m=n-1;
     for (ll i = n-m+1; i <= n; i++,mi--)
     {
          ans=max(ans,b[i]-(n-i+1)*mi+a[1]-mi);
     }
     if(sum-ans<=k) return true;
     else return false;
}
void solve(){
   cin>>n>>k;sum=0;
   for (ll i = 1; i <= n; i++)
     cin>>a[i],sum+=a[i];
     sort(a+1,a+n+1);
     b[n+1]=0;
     for (ll i = n; i >= 1; i--)
     {
          b[i]=b[i+1]+a[i];
     }
   ll l=0,r=1e18,ans=0;
   while(l<=r){
     ll mid=(l+r)>>1;
     if(check(mid)){
          ans=mid;r=mid-1;
     }else l=mid+1;
   }
   cout<<ans<<'\n';
}
int main(){
    ios;ll _=1;cin>>_;
    while (_--)solve();
    return 0;
}