hnoi 2016
标签:题解
莫队
考虑左端点左移以及右端点右移产生的贡献
这样就可以由 \([l, r]\) 转移到另外的 \(4\) 个区间
\(f_{l, r}\) 表示右端点在 \(r\), 左端点在 \([l, r]\) 内的答案,这个可以压掉一维
\(pre_{i}\) 表示 \(i\) 前面第一个比 \(a_{i}\) 小的数的位置
\(bhd_{i}\) 表示 \(i\) 后面第一个比 \(a_{i}\) 小的数的位置
考虑计算贡献 \(S\)
也就是右端点右移产生的贡献
显然新增加贡献的区间的右端点只有 \(r + 1\),
发现,在 \((pre_{r} + 1, r + 1]\) 这段区间是单调递减的
那么所有左端点在 \((pre_{r} + 1, r + 1]\) 产生的贡献为 \(a_{r + 1} * Len, Len\) 表示区间长度
我们现在另 \(r2 = pre_{r}\), 那么所有左端点在 \((pre_{r2} + 1, r2]\) 产生的贡献为 \(a_{r2} * Len\)
一直这样下去的话就会找到一个 \(x\) 使得 \(a_{pre_{x}}\) 为区间 \([l, r + 1]\) 的最小值, 记 \(p = pre_{x}\)
记此时的总和为 \(S_{1}\) 就是之前的区间产生的贡献
并且所有左端点在 \([l, p]\) 产生的贡献为 \(S_{2} = a_{p} \times ({p - l + 1})\)
那么
\[f_{l, r + 1} = a_{r + 1} \times (r + 1 - pre_{r + 1}) + \cdots + a_{x} \times (x - pre_{x}) + f_{l, p}
\]
\]
压掉第一维
\[f_{r + 1} = a_{r + 1} \times (r + 1 - pre_{r + 1}) + \cdots + a_{x} \times (x - pre_{x}) + f_{p}
\]
\]
增量
\[S_{2} = f_{r + 1} - f_{p}
\]
\]
对于 \(f_{i}\) 预处理, \(O(1)\) 查询增量
\[S = S_{1} + S_{2}
\]
\]
对于左端点左移
对称处理 \(f_{}\)
#include <bits/stdc++.h>
const int N = 1e5 + 10;
int A[N], pos[N], size;
int n, m;
struct Node {
int l, r, id;
bool operator < (Node a) const {
if(pos[l] == pos[a.l]) return r < a.r;
return l < a.l;
}
} Ask[N];
int Log[N], Pow[N];
int f[N][30];
int Stack[N], topp, pre[N], bhd[N];
int block;
#define LL long long
LL fl[N], fr[N], Answer[N];
inline int Ask_min(int l, int r) {
int t = Log[r - l + 1];
return (A[f[l][t]] < A[f[r - Pow[t] + 1][t]] ? f[l][t] : f[r - Pow[t] + 1][t]);
}
inline LL Left(int l, int r) {
int p = Ask_min(l - 1, r);
return 1ll * A[p] * (r - p + 1) + fl[l - 1] - fl[p];
}
inline LL Right(int l, int r) {
int p = Ask_min(l, r + 1);
return 1ll * A[p] * (p - l + 1) + fr[r + 1] - fr[p];
}
#define Rep(i, a, n) for(int i = a; i <= n; i ++)
int main() {
std:: cin >> n >> m;
for(int i = 1; i <= n; i ++) std:: cin >> A[i];
A[0] = A[n + 1] = (1 << 30);
for(int i = 1; i <= m; i ++) {
std:: cin >> Ask[i].l >> Ask[i].r;
Ask[i].id = i;
}
block = sqrt(n);
for(int i = 1; i <= n; i ++) pos[i] = (i - 1) / block + 1;
for(int i = 1; i <= n; i ++) f[i][0] = i;
std:: sort(Ask + 1, Ask + m + 1);
for(int i = 0; (Pow[i] = (1 << i)) <= n; i ++);
for(int i = 2; i <= n; i ++) Log[i] = Log[i >> 1] + 1;
for(int i = 1; Pow[i] <= n; i ++)
for(int j = 1; j <= n - Pow[i - 1] + 1; j ++)
f[j][i] = (A[f[j][i - 1]] > A[f[j + Pow[i - 1]][i - 1]] ? f[j + Pow[i - 1]][i - 1] : f[j][i - 1]);
for(int i = 1; i <= n; i ++) {
while(topp && A[Stack[topp]] > A[i]) {
bhd[Stack[topp]] = i;
topp --;
}
pre[i] = Stack[topp];
Stack[++ topp] = i;
}
while(topp) bhd[Stack[topp --]] = n + 1;
for(int i = 1; i <= n; i ++) fr[i] = fr[pre[i]] + 1ll * A[i] * (i - pre[i]);
for(int i = n; i >= 1; i --) fl[i] = fl[bhd[i]] + 1ll * A[i] * (bhd[i] - i);
int l = 1, r = 0;
LL Ans = 0;
for(int i = 1; i <= m; i ++) {
int x = Ask[i].l, y = Ask[i].r;
while(l > x) Ans += Left(l, r), l --;
while(r < y) Ans += Right(l, r), r ++;
while(l < x) Ans -= Left(l + 1, r), l ++;
while(r > y) Ans -= Right(l, r - 1), r --;
Answer[Ask[i].id] = Ans;
}
for(int i = 1; i <= m; i ++) std:: cout << Answer[i] << "\n";
return 0;
}