SPOJ DQUERY - D-query（主席树-区间不同数的个数）

大体题意：

给你n 个数，给你q个询问，每个询问问你某个区间上不同数的个数是多少？

思路：

主席树入门题：

简单记录一下：

这里先建立一个完整的线段树，这里的区间就代表区间了，不再是第几大了，

定义的sum 是这个区间上的不同数的个数有几个。

因为是主席树嘛，所以肯定要建立n 棵线段树，每个线段树是以每个位置的数为根，比如说该建立第i 个线段树了，如果这个数字之前没有出现过，那么我们直接以位置为划分依据，在线段树上包含这个位置的加1即可，表示这个区间上又多了一种数。

但是如果这个数字出现过了，我们先求出上一个同样数在哪里出现，我们就在第i 个线段树上以那个位置为划分依据给它减去1，在在第i 个线段树上 包含位置的i 的区间加1，这样我们就保证了 区间中数字不重复，只保留最后一个。

update代码：

int update(int pos,int c,int v,int l,int r){
int nc = ++cnt;
p[nc] = p[c];
p[nc].sum += v;
if (l == r) return nc;
int m = l+r>>1;
if (m >= pos){
p[nc].l = update(pos,p[c].l,v,l,m);
}
else {
p[nc].r = update(pos,p[c].r,v,m+1,r);
}
return nc;
}


在来说查询：
比如说我们要查询[L,R]这个区间上不同数的个数，我们就以L 为划分依据，在第R个线段树上进行查询，当发现往左走时，右边是一个完整的，我们直接加上右儿子的sum即可，在递归左儿子，如果发现是往右走，那么直接递归右边就好了，左边的不用加，因为左边的比L小，肯定会加多。

这样加到底，我们就可以得到一个完整区间[L,R]上不同数的个数。

int query(int pos,int c,int l,int r){
if (l == r) return p[c].sum;
int m = l + r >> 1;
if (m >= pos){
return p[p[c].r ].sum + query(pos,p[c].l,l,m);

}
else return query(pos,p[c].r,m+1,r);
}


完整代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn = 30000 + 10;

int n,q;
int cnt = 0;
struct Node{
int l,r,sum;
}p[maxn*40];

int la[1000000 + 10];

int a[maxn];
int root[maxn];

int build(int l,int r){
int nc = ++cnt;
p[nc].sum = 0;
p[nc].l = p[nc].r = 0;
if (l == r) return nc;
int m = l + r >> 1;
p[nc].l = build(l,m);
p[nc].r = build(m+1,r);
return nc;
}

int update(int pos,int c,int v,int l,int r){
int nc = ++cnt;
p[nc] = p[c];
p[nc].sum += v;
if (l == r) return nc;
int m = l+r>>1;
if (m >= pos){
p[nc].l = update(pos,p[c].l,v,l,m);
}
else {
p[nc].r = update(pos,p[c].r,v,m+1,r);
}
return nc;
}

int query(int pos,int c,int l,int r){
if (l == r) return p[c].sum;
int m = l + r >> 1;
if (m >= pos){
return p[p[c].r ].sum + query(pos,p[c].l,l,m);

}
else return query(pos,p[c].r,m+1,r);
}

int main(){
scanf("%d",&n);
memset(la,-1,sizeof la);
for (int i = 1; i <= n; ++i){
scanf("%d",a+i);
}
root[0] = build(1,n);

for (int i = 1 ; i <= n; ++i){
int v = a[i];
if (la[v] == -1){
root[i] = update(i,root[i-1],1,1,n);
}
else{
int t = update(la[v],root[i-1],-1,1,n);
root[i] = update(i,t,1,1,n);
}
la[v] = i;
}

scanf("%d",&q);
while(q--){
int x,y;
scanf("%d %d",&x, &y);
printf("%d\n",query(x,root[y],1,n));
}
return 0;
}

DQUERY - D-query

#sorting #tree

English

Vietnamese

Given a sequence of n numbers a1, a2,
..., an and a number of d-queries. A d-query is a pair (i, j) (1 ≤ i ≤ j ≤ n). For each d-query (i, j), you have to return the number of distinct elements in the
subsequence ai, ai+1, ..., aj.

Input

Line 1: n (1 ≤ n ≤ 30000).
Line 2: n numbers a1, a2, ..., an (1
≤ ai ≤ 106).
Line 3: q (1 ≤ q ≤ 200000), the number of d-queries.
In the next q lines, each line contains 2 numbers i, j representing a d-query (1 ≤ i ≤ j ≤ n).

Output

For each d-query (i, j), print the number of distinct elements in the subsequence ai, ai+1,
..., aj in a single line.

Example

Input
5
1 1 2 1 3
3
1 5
2 4
3 5

Output
3
2
3

 Submit
solution!

hide comments

< Previous 1 2 3 4 5 6 7 8 9