D-query SPOJ - DQUERY 查询区间有多少个不同的数（SPOJ DQUERY）

题目链接：

题意：查询区间有多少个不同的数

思路：主席树的模板题

建立主席树的话，每次建立前对之前的那个状态进行判断，判断要加的那个数是否已经出现过，如果没有出现过的话对应位置加1。如果出现过的话，将出现过的该数字的位置清0，将当前的位置加1.

 T[n+1] = build(1,n);//建立一颗空树
map<int,int>mp;
        for(int i = n;i>= 1;i--)
        {
            if(mp.find(a[i]) == mp.end())
            {
                T[i] = update(T[i+1],i,1);
            }
            else
            {
                int tmp = update(T[i+1],mp[a[i]],-1);
                T[i] = update(tmp,i,1);
            }
            mp[a[i]] = i;
        } 

接下来解释下每个线段树中存在的意义。以题目的数据为例子。

1 1 2 1 3

首先T[6]是一颗空树 

T[5]  5的位置是1

T[4]  4 ，5的位置是1

T[3]  3,4,5的位置是1

T[2] 因为2位置的1出现过，所以将4位置的1替换掉。即线段树叶节点2，3,5为1

T[1]以为1位置的1出现过，所以将2位置的1替换掉。即线段树的叶节点1,3,5，为1

那么每棵线段树相当于一个区间的左端点，例如T[1]的话，是以1为左端点的区间，当右端点扩到1位置时，出现第一个不同的数字，右端点到位置3时出现第二个不同的数字，右端点到达位置5的时候出现第三个不同的数字。

其他线段树同理。

那么最后查询区间[l，r]中有多少个不同的数字，即查询线段树T[l]这个线段树叶节点[l,r]的和

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;

/*
 * 给出一个序列，查询区间内有多少个不相同的数
 */
const int MAXN = 30010;
const int M = MAXN * 100;
int n,q,tot;
int a[MAXN];
int T[M],lson[M],rson[M],c[M];
int build(int l,int r)
{
    int root = tot++;
    c[root] = 0;
    if(l != r)
    {
        int mid = (l+r)>>1;
        lson[root] = build(l,mid);
        rson[root] = build(mid+1,r);
    }
    return root;
}
int update(int root,int pos,int val)
{
    int newroot = tot++, tmp = newroot;
    c[newroot] = c[root] + val;
    int l = 1, r = n;
    while(l < r)
    {
        int mid = (l+r)>>1;
        if(pos <= mid)
        {
            lson[newroot] = tot++; rson[newroot] = rson[root];
            newroot = lson[newroot]; root = lson[root];
            r = mid;
        }
        else
        {
            rson[newroot] = tot++; lson[newroot] = lson[root];
            newroot = rson[newroot]; root = rson[root];
            l = mid+1;
        }
        c[newroot] = c[root] + val;
    }
    return tmp;
}
int query(int root,int pos)
{
    int ret = 0;
    int l = 1, r = n;
    while(pos < r)
    {
        int mid = (l+r)>>1;
        if(pos <= mid)
        {
            r = mid;
            root = lson[root];
        }
        else
        {
            ret += c[lson[root]];
            root = rson[root];
            l = mid+1;
        }
    }
    return ret + c[root];
}

int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    while(scanf("%d",&n) == 1)
    {
        tot = 0;
        for(int i = 1;i <= n;i++)
            scanf("%d",&a[i]);
        T[n+1] = build(1,n);//建立一颗空树
        map<int,int>mp;
        for(int i = n;i>= 1;i--)
        {
            if(mp.find(a[i]) == mp.end())
            {
                T[i] = update(T[i+1],i,1);
            }
            else
            {
                int tmp = update(T[i+1],mp[a[i]],-1);
                T[i] = update(tmp,i,1);
            }
            mp[a[i]] = i;
        } 
        scanf("%d",&q);
        while(q--)
        {
            int l,r;
            scanf("%d%d",&l,&r);
            printf("%d\n",query(T[l],r));
        }
    }
    return 0;
}