题目描述
牛牛有一块长度大小为n的菜园,他首先对这块菜园从1到n进行了编号,每一块地分别为1号、2号…n号菜地,然后他往每块菜地中都种下了一些水稻,一开始,第i块菜地中的水稻高度均为a[i]个单位。然后我们知道水稻的生长周期都是n天,也就是说每逢n天水稻就会长高一个单位。但是不巧的是整个菜园中每一块菜地的生长周期都错开了,具体来说,第1天的时候第1块菜地中的水稻长高一个单位,第2天的时候第2块菜地中的水稻长高一个单位…第n天的时候第n块菜地中的水稻长高一个单位,接下来第n+1天,又轮到第1块菜地中的水稻长高一个单位以此类推。 每天在水稻进行自然生长之后,牛牛可以施展他神奇的魔法,这个魔法可以让任意一块菜地中的水稻长高一个单位,或者让任意一块菜地中的水稻缩短一个单位,当然啦,他也可以不进行任何操作。 牛牛看到菜园中的水稻参差不齐十分难受,请问至少在第几天,他能够让所有的水稻都长到同一个高度?
输入描述:
第一行是一个正整数n,1 ≤ n ≤ 1 0 5 1\leq n \leq 10^51≤n≤10
5
,表示有菜园有n块菜地。接下来一行输入n个正整数,表示每块菜地上水稻的高度,水稻的高度1 ≤ a [ i ] ≤ 1 0 9 1 \leq a[i] \leq 10^91≤a[i]≤10
9
。保证一开始输入时水稻的高度不全都相同(数据保证答案至少为1)。
输出描述:
输出一个正整数表示问题的答案。
示例1
输入
3
1 2 3
输出
1
说明
第一天,第一块菜地的水稻高度增加1,牛牛使用他的魔法使得第三块菜地的水稻再减少1。此时三块水稻的高度为2 2 2.所有水稻的高度均相同。所以答案为1天。
题意实际上就是一个求一个时刻,使得当前的序列通过±次数最少 -> 离全部相同的步数最小 -> “货仓选址” -> 求中位数+二分这个时刻即为答案
能用二分/即单调性 的证明:随着天数的增多,一定不会往预期目标偏离,至少也是原地不动,不难发现其实每过一天,这天自动调的不是往预期靠近1,就是往预期偏离1,而我们有魔法,也就是可以使得往预期靠近1变成往预期靠近2,往预期偏离1变成原地不动,所以随着天数的增多,这个时刻下能满足条件的可能性一定更大,即单调。
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<string>
#include<cstdio>
#include<algorithm>
#include<map>
#include<vector>
#include<malloc.h>
#include<numeric>
#include<cmath>
#include<iomanip>
using namespace std;
typedef long long ll;
const int N = 1e5 + 7;
/*题意实际上就是一个求一个时刻,使得当前的序列通过+-次数最少——>离全部相同的步数最小->货仓选址->求中位数+二分这个时刻即为答案*/
int n, m;
int a[N], b[N];
bool check(ll x) {
ll res = 0;
for(int i=1;i<=n;i++)
b[i] = a[i] + x / n + (x % n >= i);//精妙的处理,即求出这个二分时刻下每块田的草高.若想改为从i=0开始存则这里写改为x%n>i即可
sort(b+1 , b + n+1);//排序便于求中位数
for (int i = 1; i <= n; i++)
res += abs(b[i] - b[n / 2+1]);//这里是货舱选址模型,但注意由于是下标从1开始存到n,所以中位数是b[n/2+1]而不是b[2]
return res <= x;//若res<=x则说明当前二分的时刻满足条件.
}
int main()
{
scanf("%d", &n);
for (int i = 1; i <=n; i++)scanf("%d", &a[i]);
ll l = 0, r = 1e14 + 7;//左端点即都不用动,但0是取不到的因为题目说初始高度一定不都相等;右端点取得是1e9*1e5,即在这个天数下,1e5块田地的草高能随意从1取到1e9,则一定能符合条件,若取1e13就WA了。
while (l < r) {
long long mid = (l + r) / 2;
if (check(mid)) r = mid;//满足条件,为了找尽量小的答案所以r=mid
else l = mid + 1;
}
printf("%lld\n", l);
return 0;
}