[BNUOJ]Squared Permutation（线段树，单点更新）

题目链接：https://www.bnuoj.com/v3/problem_show.php?pid=51636

第i个数在线段树中存放的是第i个数作为位置的数字，两数交换的时候会影响区间和，画一棵线段树会发现会有四个点影响这个区间和。

#include <bits/stdc++.h>
using namespace std;

#define lrt rt << 1 
#define rrt rt << 1 | 1
typedef long long LL;

const int maxn = 100100;
const int maxm = maxn << 3;
int n, q;
int a[maxn], p[maxn];
LL seg[maxm];

void pushup(int rt) {
    seg[rt] = seg[lrt] + seg[rrt];
}

void build(int rt, int l, int r) {
    if(l == r) {
        seg[rt] = a[a[l]];
        return;
    }
    seg[rt] = 0;
    int m = (l + r) >> 1;
    build(lrt, l, m);
    build(rrt, m+1, r);
    pushup(rt);
}

void update(int rt, int l, int r, int pos, int val) {
    if(l == r) {
        seg[rt] = val;
        return;
    }
    int m = (l + r) >> 1;
    if(pos <= m) update(lrt, l, m, pos, val);
    if(pos > m) update(rrt, m+1, r, pos, val);
    pushup(rt);
}

LL query(int rt, int L, int R, int l, int r) {
    if(l >= L && R >= r) return seg[rt];
    int m = (l + r) >> 1;
    LL ret = 0;
    if(m >= L) ret += query(lrt, L, R, l, m);
    if(m < R) ret += query(rrt, L, R, m+1, r);
    return ret;
}

int main() {
    // freopen("in", "r", stdin);
    int T, o, l, r;
    scanf("%d", &T);
    while(T--) {
        scanf("%d", &n);
        for(int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
            p[a[i]] = i;
        }
        build(1, 1, n);
        scanf("%d", &q);
        while(q--) {
            scanf("%d%d%d", &o, &l, &r);
            if(o == 2) printf("%lld
", query(1, l, r, 1, n));
            else {
                swap(a[l], a[r]);
                swap(p[a[l]], p[a[r]]);
                update(1, 1, n, l, a[a[l]]);
                update(1, 1, n, r, a[a[r]]);
                update(1, 1, n, p[l], a[l]);
                update(1, 1, n, p[r], a[r]);
            }
        }
    }
    return 0;
}