题解 P5142 【区间方差】

under 题解 P5142


题目要求维护模意义下的区间方差,显然是数据结构题。

考虑方差公式:

而算术平均数

可以发现,只要维护序列的区间和区间平方和,就可以维护平均数和方差。

区间和区间平方和满足结合律,因此可以用线段树维护。


题目要求对 $ 1e9+7 $ 取模,而 $ 1e9+7 < \frac{INT _ MAX}{2} $,完全可以不使用 long long 变量维护。

注:此处 MathJax 出锅,实际为 INT_MAX

于是写了一发代码,看看毒瘤真正的取模和强制类型转换的大常数是什么样子的:

代码(c++11) 含注释:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
#include<cstdio>

using namespace std;
typedef long long ll;

constexpr int mod = 1e9+7;

int N, M, op, x, y, s1, s2, inv, ave, ans, z[100003];

int qpow(int b, int p = mod - 2, int m = mod) {
// 快速幂用于费马小定理求逆元
b %= m;
int s = 1 % m;
for(; p; p >>= 1, b = (ll)b * b % m)
if(p & 1) s = (ll)s * b % m;
return s;
}

namespace SGT {
#define ls p<<1
#define rs p<<1|1
#define sr(x) ((ll)(x)*(x)%mod) // 注意这个宏

int L[400003], R[400003], s1[4000003], s2[400003];
// s1[] 存储区间和,s2[] 存储区间平方和
// 由于无区间修改,不需要 lazytag 和 pushdown 操作

inline void pushup(int p) {
s1[p] = (s1[ls] + s1[rs]) % mod;
s2[p] = (s2[ls] + s2[rs]) % mod;
}

void build(int p, int l, int r) {
L[p] = l, R[p] = r;
if(l == r) {
s1[p] = z[l] % mod;
s2[p] = sr(z[l]) % mod;
return;
}
int m = (l + r) >> 1;
build(ls, l, m);
build(rs, m + 1, r);
pushup(p);
}

void modify(int p, int k, int v) {
// 单点修改
if(L[p] == R[p]) {
s1[p] = v % mod;
s2[p] = sr(v) % mod;
return;
}
int m = (L[p] + R[p]) >> 1;
if(k <= m) modify(ls, k, v);
else modify(rs, k, v);
pushup(p);
}

int query1(int p, int l, int r) {
// 询问区间和
if(l == L[p] && r == R[p]) return s1[p] % mod;
int m = (L[p] + R[p]) >> 1;
if(r <= m) return query1(ls, l, r) % mod;
if(l > m) return query1(rs, l, r) % mod;
return (query1(ls, l, m) + query1(rs, m + 1, r)) % mod;
}

int query2(int p, int l, int r) {
// 询问区间平方和
if(l == L[p] && r == R[p]) return s2[p] % mod;
int m = (L[p] + R[p]) >> 1;
if(r <= m) return query2(ls, l, r) % mod;
if(l > m) return query2(rs, l, r) % mod;
return (query2(ls, l, m) + query2(rs, m + 1, r)) % mod;
}
}

int main() {
scanf("%d%d", &N, &M);
for(int i = 1; i <= N; ++i) scanf("%d", &z[i]);
SGT::build(1, 1, N);

while(M--) {
scanf("%d%d%d", &op, &x, &y);
if(op == 1) SGT::modify(1, x, y % mod);
else {
// 以下各变量均在模意义下
// 强制类型转换 (ll) 一个也不能少!
s1 = SGT::query1(1, x, y) % mod; // 区间和
s2 = SGT::query2(1, x, y) % mod; // 区间平方和
inv = qpow(y - x + 1); // 区间长度(分母)的逆元
ave = (ll)s1 * inv % mod; // 区间算术平均数
ans = (ll)s2 * inv % mod - (ll)ave * ave % mod;
ans = (ans % mod + mod) % mod; // 区间方差,前文有减法操作,防止出现负数
printf("%d\n", ans);
}
}
return 0;
}

文章作者: fa_555
文章链接: https://blog.fa555.tech/2019/%E9%A2%98%E8%A7%A3-P5142-%E3%80%90%E5%8C%BA%E9%97%B4%E6%96%B9%E5%B7%AE%E3%80%91/
版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 fa_555's Blog