题意
元素的 频数 是该元素在一个数组中出现的次数。
给你一个整数数组 nums 和一个整数 k 。在一步操作中,你可以选择 nums 的一个下标,并将该下标对应元素的值增加 1。
执行最多 k 次操作后,返回数组中最高频元素的 最大可能频数 。
思路
我们第一个性质很容易想到,那就是最终结果频数最多的那个数肯定是原数组中的数。
我们先给数组排一下序。
第一层循环枚举n个原数组中的数,把其他数都尽量构造成这个数。
我们如果想把其他数都构造成这个数,那必须让小于它的数加上一些数变成这个数,我们肯定每次要选尽量接近枚举的这个数,这样可以每次使用最少的加,来构造更多的数。
但是我们怎么判断加到左边第几个数呢?
因为我们向左加的时候,需要的数是递增的,我们可以二分着来找这些数。
#include <bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 2000010;
const int mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;
using namespace std;
typedef pair<int,int> PII;
typedef pair<ll,ll> PLL;
int nums[N];
int sum[N];
int n,k;
bool check(int c,int x)
{
int cha=(x-c+1)*nums[x]-(sum[x]-sum[c-1]);
if(cha>k) return false;
else return true;
}
int find(int x)
{
int l=1,r=x;
while(l<r)
{
int mid=l+r>>1;
if(check(mid,x)) r=mid;
else l=mid+1;
}
return l;
}
int main()
{
cin>>n>>k;
for(int i=1 ; i<=n ; i++) cin>>nums[i];
sort(nums+1,nums+1+n);
for(int i=1 ; i<=n ; i++) sum[i]=sum[i-1]+nums[i];
int ans=0;
for(int i=1 ; i<=n ; i++)
{
int x=nums[i];
int l=find(i);
cout<<l<<endl;
ans=max(ans,i-l+1);
}
cout<<ans<<endl;
return 0;
}