102. 最佳牛围栏 - AcWing题库

102. 最佳牛围栏 - AcWing题库

题目大意

N块田中各有一定数量的牛[1,2000]头，现在要围起来大于等于F个连续的田地，求围起来的田地最大平均值*1000下取整，（平均值就是围起来的田地，平均有都少头牛）

做法一

知识点

        二分+前缀和+双指针

思路

        二分单调性是存在平均值mid，那么一定存在0~mid-1，二分的范围题目已给出：数量不会少于 1 头，也不会超过 2000 头。然后check就是检查围起来的田地是否可以存在平均值mid。

check过程

1、将所有田地的牛减去平均值，计算前缀和。

2、然后就是找大于等于F个连续田地的和大于等于0，这里只需要找到0~i-1田地的sum最小值minn即可，问题就转化成是否存在。 

时间复杂度

#include<iostream>
#include<cmath>
using namespace std;
const int N = 100010;
int n,f,a[N];
double sum[N];
bool check(double avg){//是否存在大于等于平均值avg
    for(int i = 1; i <= n; i++) 
        sum[i] = sum[i-1] + a[i] - avg;
    double minn = 0;
    for(int i = f, j = 0; i <= n; i++, j++){
        minn = min(minn,sum[j]);//s[0]~s[i-1]的最小值
        if(sum[i] - minn >= 0) return true;
    }
    return false;
}
int main(){
    cin>>n>>f;
    for(int i = 1; i <= n; i++) cin>>a[i];
    double l = 1, r = 2000; //数量不会少于 1 头，也不会超过 2000 头
    while(r-l > 1e-5){
        double mid = (l+r)/2;
        if(check(mid)) l = mid; //二分单调性是存在平均值mid，那么一定存在0~mid-1。
        else r = mid;
    }
    printf("%d",(int)(r*1000)); //必须是r*1000,因为l~r范围内都是正解而且我们需要的是最大值
    return 0;
}

做法二

参考：AcWing 102. 最佳牛围栏 O(n)线性做法 - AcWing

知识点

        凸包+队列

思路

1、先处理出原数组的前缀和，然后可以发现，平均数其实是前缀和函数图像割线的斜率。

2、维护一个前缀和函数的上凸包，延迟为 f 地插入每个点。

3、将队尾出队的条件为 即维护凸包

4、将队首出队的条件为维护队首最优决策，如果现在用不到以后一定用不到，不会产生更优的答案，因为明显更优秀。

可以在纸上画个凸包推一推。

时间复杂度

#include<iostream>
#include<algorithm>
#define lb long double
using namespace std;
const int N = 100010;
int n,f,q[N],a[N];

lb slope(int x,int y) {
    return (1.0*a[x]-a[y])/(1.0*x-y);
}

int main()
{
    cin>>n;
    cin>>f;
    for(int i=1;i<=n;++i) {
        cin>>a[i];
        a[i]+=a[i-1];
    }
        
    int head = 1 , tail = 0;
    lb ans = 0 ; 
    //队列中存的是转移左端点-1
    for(int i = f;i <= n; i++) {
        int ys = i-f ; //延迟f个点插入
        while(head<tail && slope(q[tail],ys) <= slope(q[tail-1],ys)) tail--;
        q[++tail]=ys;
        while(head<tail && slope(q[head],i) <= slope(q[head+1],i)) head++; 
        ans = max(ans , slope(q[head],i));
    }
    printf("%d\n",(int)(1000*ans));
}