BZOJ 1878 HH的项链 | 主席树

题意

询问区间有多少不同的数。

题解

Luogu 1903一样,这道题也是用pre数组来求区间不同数的个数,这里pre[i]表示a[i]上一次出现的位置 +1,询问相当于查询区间内有多少pre小于等于左端点。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
using namespace std;
typedef long long ll;
#define space putchar(' ')
#define enter putchar('\n')
template <class T>
void read(T &x){
    char c;
    bool op = 0;
    while(c = getchar(), c < '0' || c > '9')
	if(c == '-') op = 1;
    x = c - '0';
    while(c = getchar(), c >= '0' && c <= '9')
	x = x * 10 + c - '0';
    if(op) x = -x;
}
template <class T>
void write(T x){
    if(x < 0) x = -x, putchar('-');
    if(x >= 10) write(x / 10);
    putchar('0' + x % 10);
}
const int N = 50005;
int n, m, l, r, ans, a[N], last[1000005], pre[N];
int idx, ls[N*50], rs[N*50], data[N*50], root[N];
void build(int &k, int l, int r){
    k = ++idx;
    if(l == r) return;
    int mid = (l + r) >> 1;
    build(ls[k], l, mid);
    build(rs[k], mid + 1, r);
}
void change(int old, int &k, int l, int r, int p, int x){
    k = ++idx;
    ls[k] = ls[old], rs[k] = rs[old], data[k] = data[old];
    if(l == r) return (void)(data[k] += x);
    int mid = (l + r) >> 1;
    if(p <= mid) change(ls[old], ls[k], l, mid, p, x);
    else change(rs[old], rs[k], mid + 1, r, p, x);
    data[k] = data[ls[k]] + data[rs[k]];
}
int query(int old, int k, int l, int r, int ql, int qr){
    if(ql <= l && qr >= r) return data[k] - data[old];
    int mid = (l + r) >> 1, ret = 0;
    if(ql <= mid) ret += query(ls[old], ls[k], l, mid, ql, qr);
    if(qr > mid) ret += query(rs[old], rs[k], mid + 1, r, ql, qr);
    return ret;
}
int main(){
    read(n);
    build(root[0], 1, n);
    for(int i = 1; i <= n; i++){
	read(a[i]);
	pre[i] = last[a[i]] + 1;
	last[a[i]] = i;
	change(root[i - 1], root[i], 1, n, pre[i], 1);
    }
    read(m);
    while(m--){
	read(l), read(r);
	write(query(root[l - 1], root[r], 1, n, 1, l)), enter;
    }
    return 0;
}
本文作者:胡小兔
博客地址:http://rabbithu.cnblogs.com
posted @ 2017-11-28 20:22  胡小兔  阅读(581)  评论(0编辑  收藏  举报