hdu 3333 Turing Tree(線段樹)

題目連結：hdu 3333 Turing Tree

題目大意：給定一個長度為N的序列，有M次查詢，每次查詢l，r之間元素的總和，相同元素隻算一次。

解題思路：漲姿勢了，線段樹的一種題型，離線操作，将查詢按照右區間排序，每次考慮一個詢問，将mv ~ r的點全部标記為存在，并且對于每個位置i，如果A[i]在前面已經出現過了，那麼将前面的那個位置減掉A[i]，目前位置添加A[i]，這樣做維護了每個數盡量做，那麼碰到查詢，用sum[r] - sum[l-1]即可。

#include <cstdio>
#include <cstring>
#include <vector>
#include <map>
#include <algorithm>

using namespace std;

typedef long long ll;

const int maxn = ;

int N, Q;
ll A[maxn+], ans[];
map<ll, int> G;

#define lson(x) ((x)<<1)
#define rson(x) (((x)<<1)|1)

int lc[maxn << ], rc[maxn << ];
ll s[maxn << ];

inline void pushup (int u) {
    s[u] = s[lson(u)] + s[rson(u)];
}

void build (int u, int l, int r) {
    lc[u] = l;
    rc[u] = r;
    s[u] = ;

    if (l == r)
        return;

    int mid = (lc[u] + rc[u]) / ;
    build(lson(u), l, mid);
    build(rson(u), mid + , r);
    pushup(u);
}

void modify(int u, int x, ll d) {
    if (x == lc[u] && rc[u] == x) {
        s[u] += d;
        return;
    }

    int mid = (lc[u] + rc[u]) / ;
    if (x <= mid)
        modify(lson(u), x, d);
    else
        modify(rson(u), x, d);
    pushup(u);
}

ll query(int u, int l, int r) {
    if (l <= lc[u] && rc[u] <= r)
        return s[u];

    ll ret = ;
    int mid = (lc[u] + rc[u]) / ;
    if (l <= mid)
        ret += query(lson(u), l, r);
    if (r > mid)
        ret += query(rson(u), l, r);
    pushup(u);
    return ret;
}

struct Seg {
    int l, r, id;
    Seg (int l = , int r = , int id = ) {
        this->l = l;
        this->r = r;
        this->id = id;
    }
    friend bool operator < (const Seg& a, const Seg& b) {
        return a.r < b.r;
    }
};
vector<Seg> vec;

void init () {
    int l, r;
    G.clear();
    vec.clear();

    scanf("%d", &N);
    for (int i = ; i <= N; i++)
        scanf("%I64d", &A[i]);

    scanf("%d", &Q);
    for (int i = ; i <= Q; i++) {
        scanf("%d%d", &l, &r);
        vec.push_back(Seg(l, r, i));
    }
    sort(vec.begin(), vec.end());
}

void solve () {
    build (, , N);
    int k = ;
    for (int i = ; i < Q; i++) {
        for ( ; k <= vec[i].r; k++) {
            if (G[A[k]])
                modify(, G[A[k]], -A[k]);
            G[A[k]] = k;
            modify(, k, A[k]);
        }
        ans[vec[i].id] = query(, vec[i].l, vec[i].r);
    }
    for (int i = ; i <= Q; i++)
        printf("%I64d\n", ans[i]);
}

int main () {
    int cas;
    scanf("%d", &cas);
    while (cas--) {
        init();
        solve();
    }
    return ;
}