【SPOJ DQUERY】D-query（主席树、树状数组）

版权声明：湘 ICP 证 66666666 号 | Copyright © 2002-2019, ♂Hany01♂, All Rights Reserved 难道会有人转蒟蒻的博文？？ https://blog.csdn.net/hhaannyyii/article/details/82120572

Description

给出一个长度为$n$ 的数列，$a_{1},a_{2},...,a_{n}$​ ，有$q$ 个询问，每个询问给出数对$(i,j)$，需要你给出$a_i\dots a_j$段中有多少不同的数字。

Solution

比较水的一道题，做法也比较多，比如莫队、树状数组（离线）、主席树（在线）。
直接上莫队比较无脑。
如果用树状数组的话，我们将询问离线并挂在右端点，从左向右扫，每次加入当前的数并在上一次出现该数的位置减去贡献，然后处理挂在该位置上的询问的答案。
用类似的思路，我们可以用主席树来做到强制在线，跟上面一样地从左向右建树时加上、减去贡献即可。

---

Code

/**************************************
 * Au: Hany01
 * Prob: SPOJ DQUERY
 * Date: Aug 27th, 2018
 * Email: [email protected]
**************************************/

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef pair<int, int> PII;
typedef vector<int> VI;
#define File(a) freopen(a".in", "r", stdin), freopen(a".out", "w", stdout)
#define rep(i, j) for (register int i = 0, i##_end_ = j; i < i##_end_; ++ i)
#define For(i, j ,k) for (register int i = (j), i##_end_ = (k); i <= i##_end_; ++ i)
#define Fordown(i, j, k) for (register int i = (j), i##_end_ = (k); i >= i##_end_; -- i)
#define Set(a, b) memset(a, b, sizeof(a))
#define Cpy(a, b) memcpy(a, b, sizeof(a))
#define SZ(a) ((int)(a.size()))
#define ALL(a) a.begin(), a.end()
#define pb(a) push_back(a)
#define mp(a, b) make_pair(a, b)
#define x first
#define y second
#define INF (0x3f3f3f3f)
#define INF1 (2139062143)
#define y1 wozenmezhemecaia 
#ifdef hany01
#define debug(...) fprintf(stderr, __VA_ARGS__)
#else
#define debug(...)
#endif

template<typename T> inline bool chkmax(T &a, T b) { return a < b ? a = b, 1 : 0; }
template<typename T> inline bool chkmin(T &a, T b) { return b < a ? a = b, 1 : 0; }

inline int read() {
    register char c_; register int _, __;
    for (_ = 0, __ = 1, c_ = getchar(); !isdigit(c_); c_ = getchar()) if (c_ == '-')  __ = -1;
    for ( ; isdigit(c_); c_ = getchar()) _ = (_ << 1) + (_ << 3) + (c_ ^ 48);
    return _ * __;
}

const int maxn = 3e5 + 5;

struct Node { int lc, rc, val; }tr[maxn * 60];
int rt[maxn], tot;
map<int, int> las;

#define mid ((l + r) >> 1)

void update(int& t, int las, int l, int r, int x, int dt) {
    tr[t = ++ tot] = tr[las], tr[t].val += dt;
    if (l < r)
        if (x <= mid) update(tr[t].lc, tr[las].lc, l, mid, x, dt);
        else update(tr[t].rc, tr[las].rc, mid + 1, r, x, dt);
}

int query(int t, int l, int r, int x, int y) {
    if (x <= l && r <= y) return tr[t].val;
    if (y <= mid) return query(tr[t].lc, l, mid, x, y);
    if (x >  mid) return query(tr[t].rc, mid + 1, r, x, y);
    return query(tr[t].lc, l, mid, x, y) + query(tr[t].rc, mid + 1, r, x, y);
}

int main()
{
#ifdef hany01
    File("DQUERY");
#endif

    static int n, ai, t;

    n = read();
    For(i, 1, n) {
        ai = read(), update(rt[i], rt[i - 1], 1, n, i, 1);
        if (t = las[ai]) update(rt[i], rt[i], 1, n, t, -1);
        las[ai] = i;
    }

    for (static int m = read(), l, r; m --; )
        l = read(), r = read(), printf("%d\n", query(rt[r], 1, n, l, r));

    return 0;
}