To Add or Not to Add(二分+前缀和)
A piece of paper contains an array of n integers a1, a2, ..., an. Your task is to find a number that occurs the maximum number of times in this array.
However, before looking for such number, you are allowed to perform not more than k following operations — choose an arbitrary element from the array and add 1 to it. In other words, you are allowed to increase some array element by 1 no more than k times (you are allowed to increase the same element of the array multiple times).
Your task is to find the maximum number of occurrences of some number in the array after performing no more than k allowed operations. If there are several such numbers, your task is to find the minimum one.
The first line contains two integers n and k (1 ≤ n ≤ 105; 0 ≤ k ≤ 109) — the number of elements in the array and the number of operations you are allowed to perform, correspondingly.
The third line contains a sequence of n integers a1, a2, ..., an (|ai| ≤ 109) — the initial array. The numbers in the lines are separated by single spaces.
In a single line print two numbers — the maximum number of occurrences of some number in the array after at most k allowed operations are performed, and the minimum number that reaches the given maximum. Separate the printed numbers by whitespaces.
5 3
6 3 4 0 2
3 4
3 4
5 5 5
3 5
5 3
3 1 2 2 1
4 2
In the first sample your task is to increase the second element of the array once and increase the fifth element of the array twice. Thus, we get sequence 6, 4, 4, 0, 4, where number 4 occurs 3 times.
In the second sample you don't need to perform a single operation or increase each element by one. If we do nothing, we get array 5, 5, 5, if we increase each by one, we get 6, 6, 6. In both cases the maximum number of occurrences equals 3. So we should do nothing, as number 5 is less than number 6.
In the third sample we should increase the second array element once and the fifth element once. Thus, we get sequence 3, 2, 2, 2, 2, where number 2 occurs 4 times.
题意:给定一个数组,每次可以给任意元素加1,这样的操作不能超过k次,求操作不超过k次后数组中一个数能够出现的最多次数,以及能够以这个次数出现的最小的数
很显然这个题可以从小到大排一下序,判断i,就是在[1,i]中的最小的 j ,能使得a[j]到a[i]都变成a[i],
怎么判断这个呢,这个可以利用前缀和(前缀和具有单调性),
int judge(int l,int r){//看看数组能不能在k次操作中把a[l]到a[r]都变成a[r] ll sum1=sum[r]-sum[l-1]; ll sum2=(r-l+1)*a[r]; if(sum2-sum1<=k){ return 1; } else{ return 0; } }
#include<iostream> #include<algorithm> using namespace std; typedef long long ll; const int maxn=1e6+100; ll a[maxn],n,k; ll sum[maxn]; int judge(int l,int r){//看看数组能不能在k次操作中把a[l]到a[r]都变成a[r] ll sum1=sum[r]-sum[l-1]; ll sum2=(r-l+1)*a[r]; if(sum2-sum1<=k){ return 1; } else{ return 0; } } int lower(int s){//把数组中变成a[s]能有多少个 int l=1,r=s; int ans; while(r>=l){ int mid=(l+r)/2; if(judge(mid,s)){ r=mid-1; ans=mid; } else{ l=mid+1; } } return s-ans+1; } int main(){ cin>>n>>k; for(int i=1;i<=n;i++){ cin>>a[i]; } sort(a+1,a+n+1); for(int i=1;i<=n;i++){ sum[i]=sum[i-1]+a[i]; } ll ans=0; ll temp=0; ll ma=0; for(int i=1;i<=n;i++){ temp=lower(i);//变成a[i] if(temp>ma){ ma=temp; ans=a[i]; } } cout<<ma<<" "<<ans<<endl; }