思考:明显题意说给一个数组,然后给m次操作,每次可以让一个数+1或者-1,求最后使数组中相同的数最多。很明显需要先排序一下,然后可能会思考让那个数成为目标数,然后再左右去处理,但是这样复杂度太高了不可行。但是又想想肯定是让一段区间都变成这个目标数,又根据最小定理,当选定这个区间的时候,变成的数就是这个区间的中间的那个数。然后看看左边全部变成目标数花费多少,右边花费多少即可。这个右区间r不是每次都枚举的,这样就变成n*n的复杂度了,这个r是一直上升的,因为l越大,r肯定会越往右走。
结论:枚举左端点,求出最大右端点,然后看看是否可行即可。
代码:
#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 2010;
int T,n,m,k;
int va[N];
int sum[N];
bool get(int l,int r)
{
int mid = (l+r)>>1;
int suml = va[mid]*(mid-l+1)-(sum[mid]-sum[l-1]);
int sumr = sum[r]-sum[mid]-va[mid]*(r-mid);
return suml+sumr<=m;
}
signed main()
{
IOS;
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>va[i];
sort(va+1,va+1+n);
for(int i=1;i<=n;i++) sum[i] = sum[i-1]+va[i];
int maxn = 0,r = 0;
for(int i=1;i<=n;i++)
{
while(r+1<=n&&get(i,r+1)) r++;
maxn = max(maxn,r-i+1);
}
cout<<maxn;
return 0;
}
总结:
当一个思路方式不能再优化的时候,就要去别的方面想,但也不要想太复杂,合理合情即可。