链接:https://ac.nowcoder.com/acm/contest/5531/C
题目描述
牛牛有一块长度大小为n的菜园,他首先对这块菜园从1到n进行了编号,每一块地分别为1号、2号…n号菜地,然后他往每块菜地中都种下了一些水稻,一开始,第i块菜地中的水稻高度均为a[i]个单位。然后我们知道水稻的生长周期都是n天,也就是说每逢n天水稻就会长高一个单位。但是不巧的是整个菜园中每一块菜地的生长周期都错开了,具体来说,第1天的时候第1块菜地中的水稻长高一个单位,第2天的时候第2块菜地中的水稻长高一个单位…第n天的时候第n块菜地中的水稻长高一个单位,接下来第n+1天,又轮到第1块菜地中的水稻长高一个单位以此类推。
每天在水稻进行自然生长之后,牛牛可以施展他神奇的魔法,这个魔法可以让任意一块菜地中的水稻长高一个单位,或者让任意一块菜地中的水稻缩短一个单位,当然啦,他也可以不进行任何操作。
牛牛看到菜园中的水稻参差不齐十分难受,请问至少在第几天,他能够让所有的水稻都长到同一个高度?
输入描述:
第一行是一个正整数n(1\leq n \leq 10^51≤n≤10
5
),表示有菜园有n块菜地。
接下来一行输入n个正整数,表示每块菜地上水稻的高度,水稻的高度1 \leq a[i] \leq 10^91≤a[i]≤10
9
。
保证一开始输入时水稻的高度不全都相同(数据保证答案至少为1)。
输出描述:
输出一个正整数表示问题的答案。
思路:
发现答案满足二分性即比答案大的天数都可以使得水稻保持平衡比答案小的天数都不能使水稻保持平衡。
所以二分天数check在这些天内最少需要使用多少的魔法即货仓选址问题sort取中间点求所有的差。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100005;
typedef long long LL;
LL a[N], n, b[N];
bool check(LL t){
int k = t % n;
memcpy(b, a, sizeof a);
LL sum = 0;
for(int i = 0; i < n; i ++){
if(i < k) a[i] ++;
}
sort(a, a + n);
LL avg = a[n / 2];
LL res = 0;
for(int i = 0; i < n; i ++) res += abs(avg - a[i]);
memcpy(a, b, sizeof a);
return res <= t;
}
int main(){
cin >> n;
for(int i = 0; i < n; i ++) cin >> a[i];
LL l = 0, r = 1e14;
while(l < r){
LL mid = (l + r) / 2;
if(check(mid)) r = mid;
else l = mid + 1;
}
cout << r;
return 0;
}