牛客练习赛63 C - 牛牛的揠苗助长（货仓选址+二分答案）

牛牛的揠苗助长

题目描述

牛牛有一块长度大小为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;
}