luogu 1471 方差 (线段树区间更新)

题目背景

滚粗了的HansBug在收拾旧数学书，然而他发现了什么奇妙的东西。

 

题目描述

蒟蒻HansBug在一本数学书里面发现了一个神奇的数列，包含N个实数。他想算算这个数列的平均数和方差。

 

输入输出格式

输入格式：

第一行包含两个正整数N、M，分别表示数列中实数的个数和操作的个数。

第二行包含N个实数，其中第i个实数表示数列的第i项。

接下来M行，每行为一条操作，格式为以下两种之一：

操作1：1 x y k ，表示将第x到第y项每项加上k，k为一实数。

操作2：2 x y ，表示求出第x到第y项这一子数列的平均数。

操作3：3 x y ，表示求出第x到第y项这一子数列的方差。

输出格式：

输出包含若干行，每行为一个实数，即依次为每一次操作2或操作3所得的结果（所有结果四舍五入保留4位小数）。

 

输入输出样例

输入样例#1： 复制

5 5
1 5 4 2 3
2 1 4
3 1 5
1 1 1 1
1 2 2 -1
3 1 5

输出样例#1： 复制

3.0000
2.0000
0.8000

说明

题目链接：https://www.luogu.org/problemnew/show/P1471

题目分析：将方差公式打开，维护区间和及区间平方和即可

#include <cstdio>
#include <cstring>
#include <algorithm>
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
using namespace std;
int const MAX = 1e5 + 5;
int n, m;
double tmp, sum[MAX << 2], sum2[MAX << 2], lazy[MAX << 2];

void PushUp(int rt) {
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
    sum2[rt] = sum2[rt << 1] + sum2[rt << 1 | 1];
}

void PushDown(int ln, int rn, int rt) {
    if (lazy[rt]) {
        sum2[rt << 1] += 2.0 * sum[rt << 1] * lazy[rt] + 1.0 * ln * lazy[rt] * lazy[rt];
        sum2[rt << 1 | 1] += 2.0 * sum[rt << 1 | 1] * lazy[rt] + 1.0 * rn * lazy[rt] * lazy[rt];
        sum[rt << 1] += lazy[rt] * 1.0 * ln;
        sum[rt << 1 | 1] += lazy[rt] * 1.0 * rn;
        lazy[rt << 1] += lazy[rt];
        lazy[rt << 1 | 1] += lazy[rt];
        lazy[rt] = 0;
    }
}

void Build(int l, int r, int rt) {
    lazy[rt] = sum[rt] = sum2[rt] = 0;
    if (l == r) {
        scanf("%lf", &tmp);
        sum[rt] = tmp;
        sum2[rt] = tmp * tmp;
        return;
    }
    int mid = (l + r) >> 1;
    Build(lson);
    Build(rson);
    PushUp(rt);
}

void Update(int L, int R, double val, int l, int r, int rt) {
    if (L <= l && r <= R) {
        double len = (r - l + 1) * 1.0;
        sum2[rt] += 2.0 * sum[rt] * val + len * val * val;
        sum[rt] += len * val;
        lazy[rt] += val;
        return;
    }
    int mid = (l + r) >> 1;
    PushDown(mid - l + 1, r - mid, rt);
    if (L <= mid) {
        Update(L, R, val, lson);
    }
    if (mid < R) {
        Update(L, R, val, rson);
    }
    PushUp(rt);
}

double QuerySum(int tp, int L, int R, int l, int r, int rt) {
    if (L <= l && r <= R) {
        return tp == 1 ? sum[rt] : sum2[rt];
    }
    int mid = (l + r) >> 1;
    PushDown(mid - l + 1, r - mid, rt);
    double ans = 0;
    if (L <= mid) {
        ans += QuerySum(tp, L, R, lson);
    }
    if (mid < R) {
        ans += QuerySum(tp, L, R, rson);
    }
    return ans;
}

int main() {
    scanf("%d %d", &n, &m);
    Build(1, n, 1);
    int tp, x, y;
    double k;
    while (m--) {
        scanf("%d", &tp);
        if (tp == 1) {
            scanf("%d %d %lf", &x ,&y, &k);
            Update(x, y, k, 1, n, 1);
        } else if (tp == 2) {
            scanf("%d %d", &x, &y);
            double s = QuerySum(1, x, y, 1, n, 1);
            double len = (y - x + 1) * 1.0;
            printf("%.4f\n", s / len);
        } else {
            scanf("%d %d", &x, &y);
            double len = (y - x + 1) * 1.0;
            double s = QuerySum(1, x, y, 1, n, 1);
            double s2 = QuerySum(2, x, y, 1, n, 1);
            double ave = s / len;
            printf("%.4f\n", (s2 - 2.0 * ave * s + ave * ave * len) / len);
        }
    }
}