主席树算法

主席树(可持久化线段树)

一 定义个作用

 对于主席树而言,这个其实并不能算作是一种数据结构,只能算作是一种思想,将原来的在堆上面的开点改为了现在的动态开点,每当我们要加入一个新的点的时候,我们都需要去在当前点重开一个点,得到一个新的关系

 对每一个加入的新点得到它在原来式子中的排名(从1开始),然后根据这个排名去对他们进行更新,每次进行更新的时候去判断这个点是在前半段还是在后半段,在哪一段就对那一段进行更新,需要更新的次数为 ${\log }_{2}n$

二 基本操作

1.build

int build(int l,int r) {
    int root = ++cnt;
    if(l == r) return cnt;
    int mid = (l + r)>>1;
    ls[root] = build(l, mid);
    rs[root] = build(mid + 1,r);
    return root;
}
// 得到最开始各点之间的关系

2. update

 在更新节点的时候我们每一次都要新建一个节点

int update(int k, int l, int r, int root) {  // 插入操作
  int now = ++cnt;
  ls[now] = ls[root], rs[now] = rs[root], sum[now] = sum[root] + 1;
  // 先去继承原来点的关系,然后再去更新有必要去更新的那一边
  if (l == r) return now;
  int mid = l + r >> 1;
  if (k <= mid)
    ls[now] = update(k, l, mid, ls[now]);
  else
    rs[now] = update(k, mid + 1, r, rs[now]);
  return now;
}
// 插入一个排名为k的数字

3.ask

int query(int u, int v, int l, int r, int k) {  // 查询操作
  int mid = (l + r) >> 1, x = sum[ls[v]] - sum[ls[u]];  // 通过区间减法得到左儿子的信息,左边区间内的大小是为x
  if (l == r) return l; // 如果是当前的点就去返回
  if (k <= x)  return query(ls[u], ls[v], l, mid, k); // 如果在左边的区间内就去访问左边的区间
  else  return query(rs[u], rs[v], mid + 1, r, k - x); // 否则去访问右边的区间
}

4.init

inline void init() {
  scanf("%d%d", &n, &m);
  for (int i = 1; i <= n; ++i) scanf("%d", a + i);
  memcpy(id, a, sizeof id); // 将a中的值拷贝到id中去
  sort(id + 1, id + n + 1); // 再去进行排序,为后面的离散化做准备
  len = unique(id + 1, id + n + 1) - id - 1; // 去重
  rt[0] = build(1, len); // 建树
  for (int i = 1; i <= n; ++i) rt[i] = update(getid(a[i]), 1, len, rt[i - 1]); // 更新
}

code

#include <iostream>
#include <algorithm>
const int N = 2e5;
using namespace std;
int a[N], sum[N<<5], id[N], ls[N<<5], rs[N<<5], rt[N<<5] ,cnt, n, m, len;
int Getid(int x){
    return lower_bound(id+1, id + 1 + len, x) - id;
}
int build(int l, int r) {
    int root = ++cnt;
    if(l == r) return root;
    int mid = (l + r)>>1;
    ls[root] = build(l, mid);
    rs[root] = build(mid + 1, r);
    return root;
}
int update(int k, int l, int r, int u) {
    int now = ++cnt;
    ls[now] = ls[u]; rs[now] = rs[u]; sum[now] = sum[u] + 1;
    if(l == r) return now;
    int mid = (l + r)>>1;
    if(k <= mid) ls[now] = update(k, l, mid, ls[now]);
    else rs[now] = update(k, mid+1, r, rs[now]);
    return now;
}
int query(int u, int v, int l, int r, int k) {  // 查询操作
  int mid = l + r >> 1,x = sum[ls[v]] - sum[ls[u]];  // 通过区间减法得到左儿子的信息
  if (l == r) return l;
  if (k <= x)  // 说明在左儿子中
    return query(ls[u], ls[v], l, mid, k);
  else  // 说明在右儿子中
    return query(rs[u], rs[v], mid + 1, r, k - x);
}

void init()
{
    scanf("%d%d",&n,&m);
    for (int i = 1;i <= n;i++) scanf("%d",&a[i]),id[i] = a[i];
    sort(id + 1, id + 1 + n);
    len = unique(id + 1, id + 1 + n) - id - 1;
    rt[0] = build(1, len);
    for (int i = 1;i <= n;i++) rt[i] = update(Getid(a[i]), 1, len, rt[i-1]);
}
void work()
{
    for (int i = 1;i <= m;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        printf("%d\n",id[query(rt[x-1], rt[y], 1, len, z)]);
      // 查找在[x,y]区间上的第z个
    }
}
int main()
{
    init();
    work();
    return 0;
}