《算法竞赛进阶指南》——最佳牛围栏

与我梦中

已于 2022-10-09 11:11:07 修改

阅读量775

点赞数 2

分类专栏：刷题笔记文章标签：算法 c++ 数据结构

于 2022-10-09 11:09:09 首次发布

本文链接：https://blog.csdn.net/qq_64637770/article/details/127218505

版权

刷题笔记专栏收录该内容

4 篇文章 0 订阅

订阅专栏

1.题目描述

农夫约翰的农场由 N 块田地组成，每块地里都有一定数量的牛，其数量不会少于 1 头，也不会超过 2000 头。

约翰希望用围栏将一部分连续的田地围起来，并使得围起来的区域内每块地包含的牛的数量的平均值达到最大。

围起区域内至少需要包含 F 块地，其中 F 会在输入中给出。

在给定条件下，计算围起区域内每块地包含的牛的数量的平均值可能的最大值是多少。

输出一个整数，表示平均值的最大值乘以 1000 再 向下取整 之后得到的结果。

2.题目分析

方案1

首先，如果我们用cows[N]来存储每个田地的牛的数量的话，要求田地i到田地j中牛的数量，我们很容易想到前缀和。设前缀和数组为sum[N]，那牛的数量为：sum[j]-sum[i-1]。

但注意到题目中说田地的数量为“至少包含F块地”，最简单的做法就是把所有数量大于F的连续区间都求一遍：

for(int i=1;i+f-1<=n;i++){
        for(int j=f;i+j-1<=n;j++){
            if((a[i+j-1]-a[i-1])/j>res) res=(a[i+j-1]-a[i-1])/j;
        }
    }

然后再用二分法把不大于这个浮点数的1000倍的最大整数求出来：

 while(left<right){
        int mid=(left+right+1)/2;
        if(mid<=res){
            left=mid;
        }
        else right=mid-1;
    }

但是运行会超时…事实上，对于输入的n和f，求前缀和的时间复杂度为O(n)，将所有长度大于f的区间都遍历一遍，那比如i=1时，要从1遍历到n-f+1，n-f+1次，在i=n-f+1时为1次，所以消耗的时间为T=1+2+3+……+n-f+1=(n-f)(n-f-1)/2=O(n²），所以是很花费时间的。

方案2

在基于二分法的前提下，这题的思路居然是给出一个平均值avg，看是否有大于F的区间的平均值大于avg，从而不断缩小范围。

我们知道，在整个区间上，左边的一部分满足某个性质1，右边的一部分满足性质2，我们可以用二分法求出分界点。结合到题目给出的，每个田地牛的最小值为1，最大值为2000，也就是说最大均值一定∈[1,2000]，那么在这个数轴上，小于等于avg的为左半边部分，大于avg的为右半边。

这个单调性是显然的，若我给出一个均值a，我们不存在区间长度大于等于F的连续区间的均值大于等于a，那么我们就可以把范围缩小为[left,a）；相反，若存在区间长度大于等于F的连续区间均值大于等于a，那么最大均值avg一定满足avg>=a，那么我们又可以把范围缩小为[a,right]。

在这里插入图片描述
那现在的一个问题是，对于给定的mid，我如何判断mid属于左半边还是右半边？也就是说，是否存在长度大于等于F的区间的均值大于等于mid？

我们现在假设区间cows[i:j]的均值是会大于等于mid的，那么(sum[j]-sum[i-1])/(j-i)≥F，即sum[j]-sum[i-1]≥mid(j-i+1)，这时候我们若将cows[N]都减去平均值mid，得到新的sum，那么sum[j]+j*mid-sum[i-1]-(i-1)*mid=sum[j]-sum[i-1]+(j-i+1)*mid≥mid(j-i+1)，即sum[j]-sum[i-1]≥0

所以，我们会想，是否要让i从1遍历到n-f+1，区间长度从f遍历到n-i+1？
我们需要注意到，当j等于F时，我只有一个i可以去比较，即i=1；当j等于F+1时，我可以与两个i比较，即i=1或i=2,同样的，当j等于F+n时，我可以与i=1，2，……n比较。但注意，sum[j]-sum[i-1]≥0是一个存在问题，对于j=i+f-1，我只需要判断sum[j]是否会大于min{ sum[0],sum[1]……sum[i-1]}，因而每次j向后移动一位，只需要因为i的选择可以同样往后移动一位而更新最小值即可。即minv=min{minv,sum[i-1]},此时sum[j]只需要和minv进行比较。非常巧妙…

接下来就是二分法，具体代码如下：

#include<iostream>
#include<algorithm>
using namespace std;
const int N=100010;
int cows[N];
double sum[N];//前缀和
int n,f;
bool check(double mid){
    double minv=0;
    for(int i=1;i<=n;i++){
        sum[i]=sum[i-1]+cows[i]-mid;
    }
    for(int i=0,j=f;j<=n;i++,j++){//sum[j]总是和前sum[1:j-f+1]中的最小值比较，一定保证了长度大于f
        minv=min(sum[i],minv);
        if(sum[j]>=minv) return true;
    }
    return false;
}
int main(){
    cin>>n>>f;
    for(int i=1;i<=n;i++){
        scanf("%d",&cows[i]);
    }
    double left=1;
    double right=2000;
    while(right-left>1e-5){
        double middle=(left+right)/2;
        if(check(middle)){//如果满足条件
            left=middle;
        }
        else{
            right=middle;
        }
    }
    `cout<<int(right*1000)`;
}

关于代码实现，有以下几个问题需要注意：

1.这里不是整数上的二分问题，所以不可以写成

if(check(middle)){//如果满足条件
            left=middle+1;
        }

那另一种写法也是不可以的：

while(right-left>1e-5){
        double middle=(left+right+1)/2;
        if(check(middle)){//如果满足条件
            left=middle;
        }
        else{
            right=middle-1;
        }
    }