牛客多校4J二分答案连续子段最大平均值

最新推荐文章于 2021-07-30 16:42:51 发布

hesorchen

最新推荐文章于 2021-07-30 16:42:51 发布

阅读量349

点赞数

分类专栏：题解 # 二分

本文链接：https://blog.csdn.net/hesorchen/article/details/119172656

版权

二分查找平均值子段数组处理算法

关键词由CSDN通过智能技术生成

题解同时被 2 个专栏收录

178 篇文章 9 订阅

订阅专栏

二分

12 篇文章 0 订阅

订阅专栏

题目

Average

问题可以转化为求长度不小于K的子段的最大平均值。POJ2018原题。

其中一种比较简单的方法是，二分出答案 $m i d$ ，对原数组每个数减去 $m i d$ 后做前缀和 $s u m$ ，如果存在一段区间 $[l, r]$ 满足 $sum_r-sum_{l-1}>=0$ ，那么最大平均值 $m i d$ 就存在。

代码

include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 10;
int n, m, x, y;
int a[N], b[N];
double c[N], pre[N];

double eps = 1e-6;
double sov(int a[], int n, int x)
{
    double l = 0, r = 100000;
    while (l + eps <= r)
    {
        double mid = (l + r) / 2;
        double minn = 100000;
        int flag = 0;
        for (int i = 1; i <= n; i++)
        {
            c[i] = 1.0 * a[i] - mid;
            pre[i] = pre[i - 1] + c[i];
            if (i >= x)
            {
                minn = min(minn, pre[i - x]);
                if (pre[i] >= minn)
                    flag = 1;
            }
        }
        flag ? l = mid + eps : r = mid - eps;
    }
    return l;
}

int main()
{
    scanf("%d %d %d %d", &n, &m, &x, &y);
    for (int i = 1; i <= n; i++)
        scanf("%d", &a[i]);
    for (int i = 1; i <= m; i++)
        scanf("%d", &b[i]);
    double ans = sov(a, n, x) + sov(b, m, y);
    printf("%.10f\n", ans);
}