线段树 hdu3874 Necklace

这题有两种方法，，一种是莫队算法，，感觉莫队算法是一眼就能看出来可以用的，，只是写起来也略有麻烦

另外一种就是线段树，话说这题线段树写起来超级爽不知道为何

首先询问需要离线，然后按照右区间从大到小排序

pre[i]表示上一个数字i出现的位置

然后从1开始枚举i到n，每一次在区间[pre[A[i]]+1,i]上利用线段树成段更新加上A[i]

因为pre[A[i]]是上一个数字出现的位置，那么加1后的位置都是没出现过A[i]这个数字的，所以都需要加上

顺便解释一下当前线段树单节点的含义，，线段树[x,x]区间也就是x的单节点，存放的是[x,i]区间内不重复数字之和，也就是答案

当发现i与询问中的R相同时候，就单点询问对应查询的左端点在线段树的值，按照线段树的含义，得到的就是答案了

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<cstdio>
#include<string>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>

using namespace std;
typedef long long LL;
typedef pair<int, int> PII;

const int MX = 5e4 + 5;
const int MP = 1e6 + 5;
const int MQ = 2e5 + 5;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define root 1,n,1

struct Qust {
    int L, R, id;
    bool operator<(const Qust &b)const {
        return R < b.R;
    }
} Q[MQ];

LL ans[MQ];
int pre[MP], A[MX];
LL S[MX << 2], col[MX << 2];

void push_up(int rt) {
    S[rt] = S[rt << 1] + S[rt << 1 | 1];
}

void push_down(int rt, int len) {
    if(col[rt]) {
        col[rt << 1] += col[rt];
        col[rt << 1 | 1] += col[rt];
        int wei = len >> 1;
        S[rt << 1] += col[rt] * (len - wei);
        S[rt << 1 | 1] += col[rt] * wei;
        col[rt] = 0;
    }
}

void update(int L, int R, int d, int l, int r, int rt) {
    if(L <= l && r <= R) {
        S[rt] += (LL)d * (r - l + 1);
        col[rt] += d;
        return;
    }

    int m = (l + r) >> 1;
    push_down(rt, r - l + 1);
    if(L <= m) update(L, R, d, lson);
    if(R > m) update(L, R, d, rson);
    push_up(rt);
}

LL query(int x, int l, int r, int rt) {
    if(l == r) {
        return S[rt];
    }

    int m = (l + r) >> 1;
    push_down(rt, r - l + 1);

    LL ret;
    if(x <= m) ret = query(x, lson);
    else ret = query(x, rson);
    return ret;
}

int main() {
    //freopen("input.txt", "r", stdin);
    int T, n, Que;
    scanf("%d", &T);
    while(T--) {
        memset(S, 0, sizeof(S));
        memset(col, 0, sizeof(col));
        memset(pre, 0, sizeof(pre));

        scanf("%d", &n);
        for(int i = 1; i <= n; i++) {
            scanf("%d", &A[i]);
        }

        scanf("%d", &Que);
        for(int i = 1; i <= Que; i++) {
            scanf("%d%d", &Q[i].L, &Q[i].R);
            Q[i].id = i;
        }
        sort(Q + 1, Q + 1 + Que);

        for(int i = 1, cur = 1; i <= n; i++) {
            update(pre[A[i]] + 1, i, A[i], root);
            pre[A[i]] = i;

            while(cur <= Que && Q[cur].R == i) {
                ans[Q[cur].id] = query(Q[cur].L, root);
                cur++;
            }
        }

        for(int i = 1; i <= Que; i++) {
            printf("%I64d\n", ans[i]);
        }
    }
    return 0;
}