Average+Best Cow Fences（二分区间平均值+前缀和）

Best Cow Fences（poj-2018）

原题链接

题目描述

输入描述

输出描述

输入样例

10 6
6 
4
2
10
3
8
5
9
4
1

输出样例

6500

---

题意及思路

题意：给定一个非负序列，求该序列中长度大于 F 的连续子序列的最大平均值

思路：二分平均数 mid ，判断数组 a 中是否存在长度大于 F 且平均数大于等于 mid 的子序列，并对二分区间进行调整。

本题可构造 b 数组并令 b [ i ] = a [ i ] - mid ，由此易得对于数组 b 的某一段区间和 ≥ 0 时，对应的数组 a 相应区间内的平均数 ≥ mid。而对于数组 b 的最大区间和，可通过前缀和的方式求解，即：

最大区间和 = 前缀和 - 前面的最小前缀和（须同时保证当前区间长度大于 F ，且平均值大于等于 0 ）

参考代码

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int arr[N];
double sum[N],mid;
int n,F;

void add(double mid,int t){
    for(int i=1;i<=t;i++)
        sum[i]=sum[i-1]+arr[i]-mid;
}

int main(){
    cin>>n>>F;
    double l=0,r=0;
    for(int i=1;i<=n;++i){
        cin>>arr[i];
        r=max(r,double(arr[i]));
    }
    while(r-l>1e-3){
        mid=(l+r)/2;
        add(mid,n);
    	double minv=0;
    	for(int i=F;i<=n;++i){
        	minv=min(minv,sum[i-F]);
        	if(sum[i]>=minv) l=mid;
    	}
        if(l!=mid)r=mid;
    }
    double ans=0;
    ans+=r;
    
    cout<<int(ans*1000)<<endl;
    return 0;
}

Average（2021牛客多校第四场J）

题目描述

输入描述

输出描述

输入样例

3 4 2 2
3 1 2
4 1 3 2

输出样例

4.6666666667

---

题意与思路

题意：给定两个一维数组 a 、b ，二维数组 w，其中 w [ i ] [ j ] = a [ i ] + b [ j ] ，求矩阵 w 中长宽分别大于 x 、y 的连续子矩阵的最大平均值。

思路：本题与上一题类型相同，只是将一维转化为了二维，由于矩阵 w 构造的特殊性，因此可分别在 a 、b 数组中求出长度分别大于 x 、y 的连续子序列的最大平均值相加即可。需要注意的是，由于精度问题，r - l 需要限制在 1e - 7 。

参考代码

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int arr[N];
double sum[N],mid;
int n,m,x,y;

void add(double mid,int t){
    for(int i=1;i<=t;i++)
        sum[i]=sum[i-1]+arr[i]-mid;
}

int main(){
    cin>>n>>m>>x>>y;
    double l=0,r=0;
    for(int i=1;i<=n;++i){
        cin>>arr[i];
        r=max(r,double(arr[i]));
    }
    while(r-l>1e-7){
        mid=(l+r)/2;
        add(mid,n);
    	double minv=0;
    	for(int i=x;i<=n;++i){
        	minv=min(minv,sum[i-x]);
        	if(sum[i]>=minv) l=mid;
    	}
        if(l!=mid)r=mid;
    }
    double ans=0;
    ans+=r;
    l=r=0;
    for(int i=1;i<=m;++i){
        cin>>arr[i];
        r=max(r,double(arr[i]));
    }
    while(r-l>1e-7){
        mid=(l+r)/2;
        add(mid,m);
    	double minv=0;
    	for(int i=y;i<=m;++i){
        	minv=min(minv,sum[i-y]);
        	if(sum[i]>=minv) l=mid;
    	}
        if(l!=mid)r=mid;
    }
    ans+=r;
    printf("%.7lf\n",ans);
    return 0;
}