度度熊很喜欢数组!!
我们称一个整数数组为稳定的,若且唯若其同时符合以下两个条件:
1. 数组里面的元素都是非负整数。
2. 数组里面最大的元素跟最小的元素的差值不超过 1 。
举例而言,[1,2,1,2] 是稳定的,而 [−1,0,−1] 跟 [1,2,3] 都不是。
现在,定义一个在整数数组进行的操作:
* 选择数组中两个不同的元素 a 以及 b ,将 a 减去 2 ,以及将 b 加上 1 。
举例而言,[1,2,3] 经过一次操作后,有可能变为 [−1,2,4] 或 [2,2,1] 。
现在给定一个整数数组,在任意进行操作后,请问在所有可能达到的稳定数组中,拥有最大的『数组中的最小值』的那些数组,此值是多少呢?
Input
输入的第一行有一个正整数 T ,代表接下来有几组测试数据。
对于每组测试数据:
第一行有一个正整数 N 。
接下来的一行有 N 个非负整数 xi ,代表给定的数组。
* 1≤N≤3×105
* 0≤xi≤108
* 1≤T≤18
* 至多 1 组测试数据中的 N>30000
Output
对于每一组测试数据,请依序各自在一行内输出一个整数,代表可能到达的平衡状态中最大的『数组中的最小值』,如果无法达成平衡状态,则输出 −1 。
Sample Input
2
3
1 2 4
2
0 100000000
Sample Output
2
33333333
思路:首先找到数组中的最大值 最小值,然后对答案进行二分,每次以大于mid的为后半边(要-2的),反之为前面的。统计后面的能为前面贡献多少个+1,记为y,再用前缀和求出前面少多少个1,记为x,若x>y则不合法,否则一定合法。
#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<map>
#define N 0x3f3f3f
using namespace std;
#define ll long long
int n;
ll a[300010];
int find(int m)
{
ll x=0,y=0;
for(int i=1;i<=n;i++)
{
if(a[i]<m)
{
x+=m-a[i];
}
else
{
y+=(a[i]-m)/2;
}
}
if(x<=y)
return 1;
else
return 0;
}
int main()
{
ios::sync_with_stdio(false);
int t;
cin>>t;
while(t--)
{
cin>>n;
ll l=N,r=-N;
for(int i=1;i<=n;i++)
{
cin>>a[i];
if(a[i]<l)
{
l=a[i];
}
if(a[i]>r)
{
r=a[i];
}
}
ll ans=-1;
while(l<=r)
{
ll mid=(l+r)/2;
if(find(mid))
{
ans=mid;
l=mid+1;
}
else
r=mid-1;
}
cout<<ans<<endl;
}
return 0;
}