[HDU 3333] Turing Tree （线段树+离线算法）

链接

HDU 3333

---

题意

给出一个长度为N的序列和Q组询问，对于每组询问[l, r]，输出区间内不同值的和（即每个数的贡献只算一次）。

---

思路

一般区间询问问题都会相当用线段树去做，然而这题在线查询是不可做的。
做法是将所有区间记录下来（即“离线化”），并且按照右端点进行排序。对位置建立线段树，然后从左向右扫一遍序列，把当前位置的数a[i]插入线段树的i位置中，并且记录a[i]的新位置为i，如果a[i]在之前出现过，需要在线段树中将之前的位置置0，防止重复计算。这样当i位置存在区间的右端点时，就可以查询[l , r]，由于我们将每个值保存在它出现的最右位置上并且只保存了一次，所以不会漏解或者重复计算。
保存值的位置可以开一个map，相当于离散化。

PS：刚做过regional的网赛被虐了，想补5869那题，结果看题解也看不懂，就一步一步来先做的这题。觉得线段树+区间gcd这种类型的题目值的注意一下。

---

代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <map>
using namespace std;
typedef long long lint;
#define maxn (30003)
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
lint seg[maxn << 2];
void build(int l, int r, int rt)
{
    static int m;
    seg[rt] = 0;
    if(l == r) return;
    m = (l + r) >> 1;
    build(lson), build(rson);
}
inline void _pushup(int rt) { seg[rt] = seg[rt<<1] + seg[rt<<1|1]; }
void update(int x, int value, int l, int r, int rt)
{
    static int m;
    if(l == r) { seg[rt] = value; return ; }
    m = (l + r) >> 1;
    if(x <= m) update(x, value, lson);
    else update(x, value, rson);
    _pushup(rt);
}
lint query(int L, int R, int l, int r, int rt)
{
    if(L <= l && r <= R) { /*printf("[%d, %d] = %I64d\n", l, r, seg[rt]);*/ return seg[rt]; }
    int m = (l + r) >> 1;
    lint ret = 0;
    if(L <= m) ret += query(L, R, lson);
    if(R > m)  ret += query(L, R, rson);
    return ret;
}
#define maxm (100010)
struct _range
{
    int l, r, index;
} range[maxm];
lint o[maxm];
bool cmp(_range &a, _range &b) { return a.r < b.r; }
int a[maxn];
int main()
{
    int T;
    cin >> T;
    while(T--)
    {
        int N;
        cin >> N;
        build(1, N, 1);
        for(int i = 1; i <= N; i++)
            scanf("%d", &a[i]);
        int Q;
        cin >> Q;
        for(int i = 0; i < Q; i++)
        {
            scanf("%d%d", &range[i].l, &range[i].r);
            range[i].index = i;
        }
        sort(range, range + Q, cmp);
        map<int, int> x;
        for(int i = 1, j = 0; i <= N; i++)
        {
            if(x.count(a[i])) update(x[a[i]], 0, 1, N, 1);
            update(i, a[i], 1, N, 1);
            x[a[i]] = i;

            while(j < Q && range[j].r == i)
            {
                //printf("l = %d, r = %d\n", range[j].l, range[j].r);
                o[range[j].index] = query(range[j].l, range[j].r, 1, N, 1);
                //printf("o[%d] = %I64d\n", range[j].index, o[range[j].index]);
                j++;
            }
        }
        for(int i = 0; i < Q; i++)
            printf("%I64d\n", o[i]);
    }
    return 0;
}