分块莫队基础

· · 算法·理论

本文内容较为基础,由普通初中生整理,不喜跳过。

分块

分块思想

描述

具体来说,分块是一种思想,是一种不错的优化时间复杂度的方法。
一般是根据题意,将一段序列分成若干块,从而可以对每块操作,进而优化时间复杂度。

时间复杂度

一般来说,分块可以将时间复杂度由 \mathcal O(n^2) 优化到 \mathcal O(n \sqrt n)

例题1 Luogu P3870 开关

题意

给你一个 01 序列,初始都为 0。对其进行两种操作:翻转一段区间和查询一段区间的 1 的数量。

分块

经典的线段树问题。
考虑将整个序列分成若干块,每一块的大小为 \sqrt n。记录一个数组 ct_i 表示第 i 块的 1 的数量。

using namespace std;

define get(x) ((x - 1) / S + 1) // 找到x对应的块

define get_l(x) ((x - 1) * S + 1) // 第x块的左端点

define get_r(x) min(n, x * S) // 第x块的右端点

int n, m, S; int ct[10005], tag[10005]; int a[100010];

int main(){ scanf("%d%d", &n, &m); S = sqrt(n); // 块长 while(m --){ int c, aa, b; scanf("%d%d%d", &c, &aa, &b); if(c == 0){ if(get(aa) == get(b)){ // 位于同一块中 int k = get(aa); ct[k] = 0; for(int i = get_l(k); i <= get_r(k); i ++) { // 暴力修改 a[i] ^= tag[k]; // 懒标记下传 if(i >= aa && i <= b) a[i] ^= 1; ct[k] += a[i]; // 更新 } tag[k] = 0; // 懒标记清空 } else { int k = get(aa); ct[k] = 0; for(int i = get_l(k); i <= get_r(k); i ++){ // 左边的暴力修改 a[i] ^= tag[k]; if(i >= aa) a[i] ^= 1; ct[k] += a[i]; } tag[k] = 0; for(int j = get(aa) + 1; j < get(b); j ++){ // 完整的块 ct[j] = (get_r(j) - get_l(j) + 1) - ct[j]; // 更新 tag[j] ^= 1; // 懒标记更新 } k = get(b); ct[k] = 0; for(int i = get_l(k); i <= get_r(k); i ++){ // 暴力修改右边的部分 a[i] ^= tag[k]; if(i <= b) a[i] ^= 1; ct[k] += a[i]; } tag[k] = 0; } } else { if(get(aa) == get(b)){ int k = get(aa), res = 0; for(int i = aa; i <= b; i ++) res += (a[i] ^ tag[k]); // 块内单独查询 printf("%d\n", res); } else { int k = get(aa), res = 0; for(int i = aa; i <= get_r(k); i ++) res += (a[i] ^ tag[k]); // 左边暴力 for(int j = get(aa) + 1; j < get(b); j ++) res += ct[j]; // 完整查询 k = get(b); for(int i = get_l(k); i <= b; i ++) res += (a[i] ^ tag[k]); // 右边暴力 printf("%d\n", res); } } } return 0; }


### 例题2 [Luogu P5064 等这场战争结束之后](https://www.luogu.com.cn/problem/P5064)
#### 分析
对于**连通块的维护**, 可以考虑使用并查集。但如果使用暴力查询第 $y$ 小,时间复杂度 $\mathcal O(nm)$,必然会爆掉。考虑如何优化。  

为此,~~我们可以用平衡树~~,我们可以给**权值**(离散化后的)进行**分块**,同时开一个数组 $ct_{i, j}$ 记录第 $i$ 个点所在连通块位于分出来的第 $j$ 块的数量的多少。  
这样一来,每次查询时的操作时间复杂度可以降到 $\mathcal O(\sqrt n)$,时间绰绰有余。

#### 回滚
最让人头疼的回滚该怎么做呢?   

这时候就需要用到一个叫做**操作树**的东西。顾名思义,就是把所有的操作建成一棵树。   

我们以 $0$ 号节点为根,操作的类型为点权,若是普通操作 $i$, 就把操作 $i - 1$ 向 $i$ 连一条单向边。若是回滚操作,就把第 $i$ 次操作接在回滚回的点 $x$ 上。由于除 $0$ 号节点外,所有的节点都有一个 **父亲**,那么建出来的东西就是一棵树。   

最后 **DFS** 一边,求出答案即可。

#### 注意事项
- 空间被卡时,可以适当用 `short` 类型存储。    
- 由于需要 **DFS** 回溯,因此使用并查集时,不可路径压缩。  
- 时间比较严,建议写快读~~快写~~。 
- 并查集合并时,建议采用**启发式合并**。即将点数少的合并到点数大的,以减少时间复杂度。

#### Code

```cpp
#include<bits/stdc++.h>

using namespace std;

const int N = 1e5 + 5, S = 3000;

int n, Q, m, k;
int w[N], b[N], num[N]; //离散化
short iuy[N][N / S + 2]; // 在块中的在块中的数量(必须开short,不然空间会卡)
int fa[N], sz[N]; // 并查集
int number[N]; 
pair<pair<int, int>, int> q[N]; // 询问离线
int ans[N]; // 答案

struct Node{
    int to, nxt;
} tr[N << 1];

int hd[N << 1], idx;

void add(int u, int v){
    tr[++ idx] = {v, hd[u]}, hd[u] = idx;
}

void init(){
    for(int i = 1; i <= n; i ++) fa[i] = i, sz[i] = 1;
}

int find(int x){ // 查找; 不要路径压缩
    if(x == fa[x]) return x;
    return find(fa[x]);
}

void dfs(int u){
    int op = q[u].second, x = q[u].first.first, y = q[u].first.second;
    if(op == 1){ // 加入一条边
        x = find(x), y = find(y);
        if(x != y) {
            if(sz[x] > sz[y]) swap(x, y); 
            fa[x] = y; sz[y] += sz[x]; // 启发式合并
            for(int i = 1; i <= m; i ++){ // 对于每一块分别更新
                iuy[y][i] += iuy[x][i];
            }   
        }
    }
    if(op == 3){
        x = find(x);
        if(y > sz[x]) ans[u] = -1;
        else {
            for(int i = 1; i <= m; i ++){
                if(y <= iuy[x][i]){ // 一块一块的找
                    for(int j = (i - 1) * S + 1; j <= i * S; j ++){ // 单独查找
                        if(find(num[j]) == x){
                            -- y;
                            if(y == 0) {
                                ans[u] = b[j];
                                break; 
                            } 
                        }
                    } 
                    break;
                }
                y -= iuy[x][i];
            }
        }
    }   
    for(int i = hd[u]; i; i = tr[i].nxt){ // 继续操作
        dfs(tr[i].to);
    }
    if(op == 1 && x != y){ // 回溯操作
        sz[y] -= sz[x]; fa[x] = x; 
        for(int i = 1; i <= m; i ++) {
            iuy[y][i] -= iuy[x][i];
        }
    }
}

int main(){
    scanf("%d%d", &n, &Q);
    init();
    m = (n - 1) / S + 1; // 块
    for(int i = 1; i <= n; i ++){
        scanf("%d", &w[i]); b[i] = w[i];
    }
    sort(b + 1, b + n + 1); // 离散化
    for(int i = 1; i <= n; i ++){
        w[i] = lower_bound(b + 1, b + n + 1, w[i]) - b;
        w[i] += number[w[i]] ++;
        num[w[i]] = i;
        iuy[i][(w[i] - 1) /  S + 1] = 1;
    }
    for(int i = 1; i <= Q; i ++){
        int op, x, y; scanf("%d%d", &op, &x);
        if(op != 2) scanf("%d", &y), add(i - 1, i);
        if(op == 2) add(x, i);
        q[i] = {{x, y}, op}; // 离线
    }
    dfs(0);
    for(int i = 1; i <= Q; i ++){
        if(q[i].second == 3) printf("%d\n", ans[i]);
    }
    return 0;
}

块状链表

如上图(from oi-wiki),就是把一个序列变成链表

我们可以把原数组分成 \sqrt n 个节点,每个节点挂一个长度为 \sqrt n 的数组

struct Node{
    char s[N + 20]; // 存的数组
    int len, pre, nxt; // 长度,前驱,后继
} p[M];

作为一个合格的块状链表, 它应该至少支持:分裂插入查找三种操作。下面以 Luogu P4008 文本编辑器为例。

查找

单点(移动到某一个点 k 后面)

我们可以一个节点一个节点的去找,每到一个节点后,判断是否要查找的第 k 个数是否大于当前节点的大小。若大于,则将 k 减去当前节点的大小;若小于等于,则在这块内寻找。时间复杂度 \mathcal O(\sqrt n)

void Move(int k){
    x = 0;
    while(k > p[x].len) k -= p[x].len, x = p[x].nxt; // 一直往后跳
    y = k;
} 

区间

和普通分块差不多,只需先单点找到左端点,再和普通分块一样跳就可以了。

void Get(int lenth){//左端点之前已经找到了
    if(p[x].len - y >= lenth){ // 节点内
        for(int i = y + 1; i <= y + lenth; i ++) putchar(p[x].s[i]);
    } else { // 节点间
        lenth -= p[x].len - y;
        for(int i = y + 1; i <= p[x].len; i ++) putchar(p[x].s[i]);
        int id = p[x].nxt;
        while(lenth > p[id].len){
            for(int i = 1; i <= p[id].len; i ++) putchar(p[id].s[i]);
            lenth -= p[id].len;
            id = p[id].nxt;
        }
        for(int i = 1; i <= lenth; i ++) putchar(p[id].s[i]);
    }
    puts("");
}

分裂

顾名思义,就是将一个节点从某处分成两个节点。我们只需要再开一个新的节点,将分裂出来的节点放到里面就可以了。

插入(在某个点后插入一段区间)

首先,我们要知道,在一个节点的内部是肯定是不可以插入一个节点的。所以,我们就需要先从插入节点的地方分裂开, 再在这两个节点之间插入一段序列。

void Insert(int lenth){
    if(y < p[x].len){ // 如果在一个节点内部,分裂
        int nw = u.top(); u.pop(); // 从还未使用的节点中取出一个
        for(int j = y + 1; j <= p[x].len; j ++) p[nw].s[++ p[nw].len] = p[x].s[j]; // 先复制一份
        p[x].len = y;
        add(x, nw); // 增加分裂后的连边
    }
    int lst = x;
    for(int i = 1; i <= lenth;){ 
        int nw = u.top(); u.pop();
        for(int j = 1; j <= N && i <= lenth; j ++, i ++) p[nw].s[++ p[nw].len] = str[i]; // 将插入的序列分好
        add(lst, nw); // 增加节点
        lst = nw;
    }
}

维护时间正确的关键:合并操作

如果一直按照上面的操作不断分裂,我们将无法保证块长稳定在 \sqrt n 左右,进而时间复杂度骤增。因此,我们需要在每次分裂或者插入的时候,进行合并操作,将块长较小的(\le \sqrt n 的)节点合并起来,以求时间复杂度的正确性。

void Merge(){
    int i = p[0].nxt;
    while(p[i].nxt){
        if(p[i].len + p[p[i].nxt].len <= N){ // 小于 $N = \sqrt n$
            if(x == p[i].nxt) x = i, y += p[i].len;
            for(int k = 1; k <= p[p[i].nxt].len; k ++) p[i].s[++ p[i].len] = p[p[i].nxt].s[k];
            u.push(p[i].nxt); del(p[i].nxt); 
        }
        i = p[i].nxt;
        if(i == 0) break;
    }
} 

Code

#include<bits/stdc++.h>

using namespace std;

const int N = 2000, M = 2010;

int x, y;
stack<int> u;
char str[2000010];

struct Node{
    char s[N + 20];
    int len, pre, nxt;
} p[M];

void Move(int k){
    x = 0;
    while(k > p[x].len) k -= p[x].len, x = p[x].nxt;
    y = k;
} 

void add(int a, int b){
    p[b].nxt = p[a].nxt, p[p[b].nxt].pre = b;
    p[a].nxt = b, p[b].pre = a;
}

void del(int a){
    p[p[a].pre].nxt = p[a].nxt, p[p[a].nxt].pre = p[a].pre;
    p[a].nxt = p[a].len = p[a].pre = 0;
}

void Insert(int lenth){
    if(y < p[x].len){
        int nw = u.top(); u.pop();
        for(int j = y + 1; j <= p[x].len; j ++) p[nw].s[++ p[nw].len] = p[x].s[j];
        p[x].len = y;
        add(x, nw);
    }
    int lst = x;
    for(int i = 1; i <= lenth;){
        int nw = u.top(); u.pop();
        for(int j = 1; j <= N && i <= lenth; j ++, i ++) p[nw].s[++ p[nw].len] = str[i];
        add(lst, nw);
        lst = nw;
    }
}

void Merge(){
    int i = p[0].nxt;
    while(p[i].nxt){
        if(p[i].len + p[p[i].nxt].len <= N){
            if(x == p[i].nxt) x = i, y += p[i].len;
            for(int k = 1; k <= p[p[i].nxt].len; k ++) p[i].s[++ p[i].len] = p[p[i].nxt].s[k];
            u.push(p[i].nxt); del(p[i].nxt); 
        }
        i = p[i].nxt;
        if(i == 0) break;
    }
} 

void Delete(int lenth){
    if(p[x].len - y >= lenth){
        for(int i = y + 1, j = y + lenth + 1; j <= p[x].len; i ++, j ++) p[x].s[i] = p[x].s[j];
        p[x].len -= lenth;
    } else {
        lenth -= p[x].len - y;
        p[x].len = y;
        int id = p[x].nxt;
        while(lenth > p[id].len){
            lenth -= p[id].len;
            u.push(id);
            id = p[id].nxt;
            del(p[id].pre);
        }
        for(int i = 1, j = lenth + 1; j <= p[id].len; i ++, j ++) p[id].s[i] = p[id].s[j];
        p[id].len -= lenth;
    }
}

void Get(int lenth){
    if(p[x].len - y >= lenth){
        for(int i = y + 1; i <= y + lenth; i ++) putchar(p[x].s[i]);
    } else {
        lenth -= p[x].len - y;
        for(int i = y + 1; i <= p[x].len; i ++) putchar(p[x].s[i]);
        int id = p[x].nxt;
        while(lenth > p[id].len){
            for(int i = 1; i <= p[id].len; i ++) putchar(p[id].s[i]);
            lenth -= p[id].len;
            id = p[id].nxt;
        }
        for(int i = 1; i <= lenth; i ++) putchar(p[id].s[i]);
    }
    puts("");
}

void Prev(){
    if(y == 1) y = p[x = p[x].pre].len;
    else y --;
}

void Next(){
    if(y == p[x].len) y = 1, x = p[x].nxt;
    else y ++;
}

int main(){
    for(int i = 2000; i; i --) u.push(i);
    int t; scanf("%d", &t);
    while(t --){
        int a; char op[10]; scanf("\n%s", op + 1);
        if(op[1] == 'I'){
            scanf("%d", &a);
            int n = a, i = 1;
            while(n){
                str[i] = getchar();
                if(str[i] >= 32 && str[i] <= 126) i ++, n --; 
            }
            Insert(a);
            Merge();
        } else if(op[1] == 'D') {
            scanf("%d", &a);
            Delete(a);
            Merge();
        } else if(op[1] == 'M') {
            scanf("%d", &a);
            Move(a);
        } else if(op[1] == 'G') {
            scanf("%d", &a);
            Get(a);
        } else if(op[1] == 'P') Prev();
        else Next();
    }
    return 0;
} 

STL中的块状链表:rope

rope 是STL中的一种数据结构,头文件是 <ext/rope>, (好像不在万能头里)。使用时如下:

#include <ext/rope>
using namespace __gnu_cxx;

它支持块状链表的一般操作,但是时间复杂度较小。

如下,给出一些基本操作: 操作 作用
rope<int> a 初始化一个 rope
a.push_back(x) a 的末尾插入一个数 x
a.insert(pos, x, y) 在位置 pos 插入 y (不写则默认为 1)个数 x
a.erase(pos, x) 在位置 pos 删除 x 个元素
a.at(x)a[x] 访问 a 的第 x 个元素
a.length()a.size() 获取 a 的大小

需要注意的是,rope 的下标由 0 开始。

下面给出一个板子: LOJ 数列分块6

#include <bits/stdc++.h>
#include <ext/rope>

using namespace std;
using namespace __gnu_cxx;

rope<int> a;

int main() {
    int t;
    scanf("%d", &t);
    for (int i = 1; i <= t; i++) {
        int x;
        scanf("%d", &x);
        a.push_back(x);
    }
    while (t--) {
        int op, l, r, c;
        scanf("%d%d%d%d", &op, &l, &r, &c);
        if (op == 0) {
            a.insert(l - 1, r);
        } else {
            printf("%d\n", a[r - 1]);
        }
    }
    return 0;
}

更科技的版本建议去看陈老师的索引树优化块状链表。(常数小到离谱)

数论分块

题意:求 \sum_{i = 1}^{n} \lfloor \frac{1}{n} \rfloorn \le 10^{14}

如果直接求,时间肯定会炸掉,那么,我们考虑优化。

如下图,是上面函数的图像,我们发现,原函数中,有很多的值是相同的。

因此,我们可以把这些相同的数合并在一起统计。

对于一个数 i,我们可以发现他的右端点是 \lfloor \dfrac{n}{\lfloor \frac{n}{i} \rfloor} \rfloor。证明如下:

k = \lfloor \dfrac{n}{i} \rfloor, 则一定有 k \le \dfrac{n}{i}
那么,i = \lfloor \dfrac{n}{k} \rfloor \ge \lfloor \dfrac{n}{\lfloor \frac{n}{i} \rfloor} \rfloor,所以,i 所在块的右端点为 \lfloor \dfrac{n}{\lfloor \frac{n}{i} \rfloor} \rfloor

那么根据这一个东西我们就可以计算了。

时间复杂度

我们分块来看。

对于 i \le \sqrt n,我们可以发现,我们只需要遍历 [1, \sqrt n]

对于 i > \sqrt n,他们的值域为 [1, \sqrt n]

因此,总体的时间复杂度为 \mathcal O(\sqrt n)

Code

LL l = 1ll, r;
while(l <= n){
    r = n / (n / l);
    (ans -= 1ll * (r - l + 1ll) % Mod * (1ll * (n / l) % Mod) % Mod) %= Mod;
    (ans += Mod) %= Mod; 
    l = r + 1ll;
}

例题 [Luogu P2261 余数和]

众所周知, k\% i 可以表示成 k - i \times \lfloor \dfrac{k}{i} \rfloor。那么,原式 = \sum_{i = 1} ^ {n} (k - i \times \lfloor \dfrac{k}{i} \rfloor) = \sum_{i = 1} ^ {n} k - \sum_{i = 1} ^ {n} i \times \lfloor \dfrac{k}{i} \rfloor

如此,我们就可以在 \mathcal O(\sqrt n) 的时间复杂度内完成此题。

#include<bits/stdc++.h>

using namespace std;

long long n, k;

int main(){
    scanf("%lld%lld", &n, &k);
    long long l = 1, r, ans = n * k;
    while(l <= n){
        if(k / l == 0) r = n;
        else r = min(n, k / (k / l));
        ans -= (l + r) * (r - l + 1) / 2 * (k / l);
        l = r + 1;
    }
    printf("%lld\n", ans);
    return 0;
} 

莫队

普通莫队

莫队算法可以解决一类离线区间询问问题,适用性极为广泛。

算法功能:

一个序列 a_1, a_2, a_3, ..., a_n 中,查询区间 [l, r] 的某种信息。把正常复杂度 \mathcal O(n^2) 的算法优化成 \mathcal O(n \sqrt{n})

主题思想

分块

把整个查询的序列离线下来之后,把整个序列分成 \sqrt{n} 个块,每个块内在分别按查询的右端点排序 (节省时间)。

查询

用当前已有[l, r] 序列,推出 [l \pm 1, r \pm 1]的信息,再根据需要查询的区间,加 / 删点。

时间复杂度分析

主要的复杂度由查询中,指针跳跃的时间。首先,左指针由于已经分好了块,最多跳 \sqrt{n} 次,而右指针在每个块中最多最多可以跳 n 次。总复杂度乘起来就是 \mathcal O(n \sqrt{n})

注意事项 & 使用条件

for(int i = 1; i <= m; i ++) b[i].l = read(), b[i].r = read(), b[i].id = i;//离线下来
sort(b + 1, b + m + 1, [](node a, node b)
{ return a.l / S != b.l / S ? a.l < b.l : a.r < b.r; }) ;//排序:不在一个块内的,按块排;在的,按右端点先后排

处理查询操作

考虑已知区间 [l, r] 的信息,该如何推出下一个区间有多少个不同的值。那么,我们可以开一个桶,用来记录每个值在当前区间分别出现了多少次,那么将 l 往左移时,只需要将对应的桶相应的加上一,如果加一以前没有记录过,就说明这是一个新增的值,需要把答案加一;把 l 向右移时,则在桶中删除当前的值,如果桶中归零,那么相对应地,答案要减一。 r 同理。

void add(int x){
    if(!t[a[x]]) tot ++;//以前的区间中没有出现过,更新答案
    t[a[x]] ++;
}

void del(int x){
    if(t[a[x]] == 1) tot --;//删没了,要减一
    t[a[x]] --;
}
for(int i = 1, L = 1, R = 0; i <= m; i ++){
        while(L < b[i].l) del(L ++);//多了要删
        while(L > b[i].l) add(-- L);//少了要加
        while(R > b[i].r) del(R --);//多了要删
        while(R < b[i].r) add(++ R);//少了要加
        ans[b[i].id] = (tot == b[i].r - b[i].l + 1);//答案
    } 

带修莫队

基本思想

莫队算法是暴力离线算法,一般不支持修改操作,如果需要修改操作,则需要在原基础 (l, r) 上增加一个维度——时间戳,用来表示修改操作的时间范围,即 (l, r, t)

具体操作

对于区间 [l, r],采用普通莫队方法即可。对于时间的改变,我们可以直接暴力将其移到对应时间即可。

时间复杂度分析

设分块的块长为 L, 则有 \dfrac {n}{L} 个块。那么 l 指针移动 \dfrac{n}{L} 次,具体时间为 nLr 每块移动 n 次,共计 \dfrac {n ^ 2}{L} 次,具体为 nL + \dfrac{n^2}{L}。对于 t 指针,打了 t 个时间戳,又因为 lr 的块总值 \dfrac {n ^ 2}{L ^ 2}。所以,总时间复杂度最优时,根据均值不等式,在 L 取到 \sqrt[3]{nt} 时最优。总体时间复杂度 \mathcal O(\sqrt[3]{n^4t})

模板(Luogu P1903 [国家集训队] 数颜色 / 维护队列)

#include<bits/stdc++.h>

using namespace std;

const int N = 150010, S = 1000010;

int n, m, mq, mc, len;
int w[N], cnt[S], ans[N];

struct Query { // 离线查询
    int id, l, r, t;
}q[N];

struct Modify { // 修改
    int p, c;
}c[N];

int get(int x)
{
    return x / len;
}

bool cmp(const Query &a, const Query &b) // 排序
{
    int al = get(a.l), ar = get(a.r);
    int bl = get(b.l), br = get(b.r);
    if (al != bl) return al < bl; // 先按块
    if (ar != br) return ar < br; // 再按下标
    return a.t < b.t; // 后按时间
}

void add(int x, int& res) {
    if (!cnt[x]) res ++;
    cnt[x] ++;
}

void del(int x, int &res) {
    cnt[x] --;
    if (!cnt[x]) res --;
}

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++) scanf("%d", &w[i]);
    for (int i = 0; i < m; i ++) {
        char op[2];
        int a, b;
        scanf("%s%d%d", op, &a, &b);
        if (*op == 'Q') mq ++, q[mq] = {mq, a, b, mc};
        else c[++ mc] = {a, b};
    }
    len = cbrt((double)n * max(1 , mc)) + 1; // 三次根号
    sort(q + 1, q + mq + 1, cmp);
    for (int i = 0, j = 1, t = 0, k = 1, res = 0; k <= mq; k ++) {
        int id = q[k].id, l = q[k].l, r = q[k].r, tm = q[k].t;
        while (i < r) add(w[++ i], res);
        while (i > r) del(w[i --], res);
        while (j < l) del(w[j ++], res);
        while (j > l) add(w[-- j], res); // 找区间
        while (t < tm) { // 暴力搞时间
            t ++;
            if (c[t].p >= j && c[t].p <= i)
            {
                del(w[c[t].p], res);
                add(c[t].c, res);
            }
            swap(w[c[t].p], c[t].c);
        }
        while (t > tm) { // * 2
            if (c[t].p >= j && c[t].p <= i) {
                del(w[c[t].p], res);
                add(c[t].c, res);
            }
            swap(w[c[t].p], c[t].c);
            t --;
        }
        ans[id] = res; // 更新答案
    }
    for (int i = 1; i <= mq; i ++) printf("%d\n", ans[i]);
    return 0;
}

附录

对比几种数据结构 名称 时间复杂度 码量 直观性
朴素暴力 \mathcal O(n ^ 2) 不定
线段树/平衡树 \mathcal O(n \log{n}) 较好
分块(莫队) \mathcal O(n \sqrt{n}) 适中
树状数组 \mathcal O(n \log{n})

终于写完了。完结撒花★,°:.☆( ̄▽ ̄)/$:.°★