bzoj 1878: [SDOI2009]HH的项链（主席树）

1878: [SDOI2009]HH的项链

Time Limit: 4 Sec Memory Limit: 64 MB
Submit: 5317 Solved: 2624
[Submit][Status][Discuss]

Description

HH有一串由各种漂亮的贝壳组成的项链。HH相信不同的贝壳会带来好运，所以每次散步完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH不断地收集新的贝壳，因此他的项链变得越来越长。有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答。。。因为项链实在是太长了。于是，他只好求助睿智的你，来解决这个问题。

Input

第一行：一个整数N，表示项链的长度。

第二行：N个整数，表示依次表示项链中贝壳的编号（编号为0到1000000之间的整数）。

第三行：一个整数M，表示HH询问的个数。

接下来M行：每行两个整数，L和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

N ≤ 50000，M ≤ 200000。

Output

M行，每行一个整数，依次表示询问对应的答案。

Sample Input

1 2 3 4 3 5

1 2

3 5

2 6

Sample Output

据说当时出这道题的时候，还没有主席树这东西

现在有了主席树，就成水题了

记L[i]为第i个数字上一次出现的位置，那么对于每次询问[x, y]，答案就是这个区间内满足L[i]<x的数的个数

#include<stdio.h>
#include<algorithm>
using namespace std;
typedef struct
{int l, r;int val;
}Tree;
Tree s[4000005];
int n, cnt, pos[1000005], L[100005], a[100005], t[100005];
int Build(int l, int r)
{int m, root;m = (l+r)/2;root = ++cnt;if(l==r){s[root].val = 0;return root;}s[root].l = Build(l, m);s[root].r = Build(m+1, r);s[root].val = s[s[root].l].val+s[s[root].r].val;return root;
}
int Update(int root, int k)
{int now, temp, m, l, r;now = temp = ++cnt;l = 1, r = n+1;while(l<r){m = (l+r)/2;s[now].val = s[root].val+1;if(k<=m){s[now].l = ++cnt;s[now].r = s[root].r;root = s[root].l;r = m;now = cnt;}else{s[now].r = ++cnt;s[now].l = s[root].l;root = s[root].r;l = m+1;now = cnt;}}s[now].val = s[root].val+1;return temp;
}
int Query(int root, int l, int r, int a, int b)
{int ans, m;ans = 0;if(l>=a && r<=b)return s[root].val;m = (l+r)/2;if(a<=m)ans += Query(s[root].l, l, m, a, b);if(b>=m+1)ans += Query(s[root].r, m+1, r, a, b);return ans;
}
int main(void)
{int i, m, l, r;scanf("%d", &n);for(i=2;i<=n+1;i++){scanf("%d", &a[i]);L[i] = max(pos[a[i]], 1);pos[a[i]] = i;}t[1] = Build(1, n+1);for(i=2;i<=n+1;i++)t[i] = Update(t[i-1], L[i]);scanf("%d", &m);while(m--){scanf("%d%d", &l, &r);l += 1, r += 1;printf("%d\n", Query(t[r], 1, n+1, 1, l-1)-Query(t[l-1], 1, n+1, 1, l-1));}return 0;
}