1175 区间中第K大的数
题目链接:http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1175&judgeId=649231
主席树与普通线段树的建树方法有点不同,他只保存他的左儿子和右儿子,以及这条路径覆盖了多少次
比如要找
1 2 3 3 4 5
这6个数[2,4]中第3大的数
先离散化,就只有5个数了,然后建一颗空的线段树,每条路径都是0,然后依次把 1 2 3 3 4 5 加进去,每加一个数,就多加线段树上的一条路径
①这是空树:
②:然后把1加进去
③然后把2加进去
④然后把3加进去
⑤再加一个3进去
⑥把4加进去
⑦把5加进去
假如要求[2,4]这个区间的第3大的数,那么就是用加入了4个数的线段树来减去,加入了两个数的线段树的前面一个线段树(类似于前缀和那种sum[R]-sum[L-1]),就能得到这段区间有多少数了,并且大小顺序是有的,数量也是有的
第4个线段树最下面一层就是:1 1 2 0 0
第1个线段树最下面一层就是:1 0 0 0 0
于是减完就是:0 1 2 0 0
我们查询就是用这个东西来查询的,之前一直没理解好~
那么要求第k小就是求前缀和,看多少能达到k,而求第k大就是求后缀和,看多少能达到k
就像这样,从左边开始,当框起来的数的和达到了k,那就找到了
同理第k大,就从右边开始框,框起来的数达到了k,那么这个位置的数就是第 k大了
#include"bits/stdc++.h"
using namespace std;
const int maxn=5e4+5;
int a[maxn];
int data[maxn];//原始数据
int N,n,tot,Q;
vector<int>Root;//用来保存加进去的每个数的头节点是多少
int tree[maxn*30],Ls[maxn*30],Rs[maxn*30];
int Build(int L,int R)//对离散后的n个数先建一个空数
{
int id=++tot;
tree[id]=0;
if(L==R)return id;
int mid=L+R>>1;
Ls[id]=Build(L,mid);
Rs[id]=Build(mid+1,R);
return id;
}
void Update(int id1,int v)
{
v=lower_bound(a+1,a+1+n,v)-a; //找到这个数排在第几
int id2=++tot; //新建一个root
tree[id2]=tree[id1]+1;
Root.push_back(id2);
int L=1,R=n;
while(L<R)
{
int mid=L+R>>1;
if(v<=mid) //说明这个数是在左边
{
R=mid;
Ls[id2]=++tot;
Rs[id2]=Rs[id1];//继承原来的右儿子
id2=tot;
id1=Ls[id1];//向左走
}
else
{
L=mid+1;
Rs[id2]=++tot;
Ls[id2]=Ls[id1];
id2=tot;
id1=Rs[id1];//向右走
}
tree[id2]=tree[id1]+1;//这条路径上都加1
}
}
int query(int rt1,int rt2