hdu2665 主席树模板题

来源:互联网 发布:淘宝店铺音乐怎么修改 编辑:程序博客网 时间:2024/06/10 07:30

题目
http://acm.hdu.edu.cn/showproblem.php?pid=2665

区间k大值,区间极值很容易想到线段树,如果k是个位数的话,可以考虑开k个域的线段树= =,,,,滚~
又称可持久化线段树,函数式线段树
也许是上面两个字看的太长,同时主席两字给人一种不明觉厉的感觉,,,so,嘿嘿嘿

关于主席树的讲解可以看这
http://wenku.baidu.com/link?url=6sjYgDa2UEZujFwLE4KDqgJlOs0tio-hwKMoDBexBLKT6DSgSiq4Ed-w8cHF4L3JraAflBURPdzeHeOft-f02RT1xfneYIfM5pTVs3t1cBi

先离散化,对每个点i,建一个1~i的线段树(大小是数字的个数)
记录该前缀序列里出现的值的次数;记离散后的标记为1~n;
对于区间[x,y]的第k大的值,那么从root[x-1],root[y]开始,
t=root[y].[1,mid]-root[x-1].[1,mid] ,t表示区间[x,y]内值在[1,mid]的个数
先判断t是否大于K,如果t大于k,那么说明在区间[x,y]内存在[1,mid]的数的个数大于k,
也就是第k大的值在[1,mid]中,否则在[mid+1,r]中;

这样我们依次同时从root[x-1],root[y]往下走
当l==r时 第k大的值就是离散后标记为l的值;

如果每棵线段都建完整的化,n^2肯定会mle,,,BANG!!!
我们发现对于前缀[1,i]和前缀[1,i+1]的线段树,如果b[i+1]<=mid (b[i+1]表示a[i+1]离散后的标记)
那么线段树i和线段树i+1的左边是完全相同的,根本不需要在建,直接链过去就好;
那么对于一棵新的线段树其实我们最多要建的节点数为log(n);nlog(n)的节点数,duangduangduang

存个代码吧,,,,以后如果写熟练了有空写个去重的离散

//嘿嘿嘿# include <cstdio># include <iostream># include <algorithm># define N 100100using namespace std;int i,n,m,root[N],a[N],p[N],b[N],cnt;struct node{int lc,rc,w;}T[N*30];bool cmp(int i,int j){return a[i]<a[j];}void build(int &rot,int l,int r,int x){    T[++cnt]=T[rot];rot=cnt;    T[cnt].w++;    if (l==r)return ;    int mid=(l+r) >> 1;    if (x<=mid)build(T[cnt].lc,l,mid,x);    else build(T[cnt].rc,mid+1,r,x);}int query(int x,int y,int l,int r,int k){    if (l==r)return l;    int t=T[T[y].lc].w-T[T[x].lc].w;    int mid=(l+r) >> 1;    if (t>=k)return query(T[x].lc,T[y].lc,l,mid,k);    else return query(T[x].rc,T[y].rc,mid+1,r,k-t);}int main(){    //freopen("fuck.in","r",stdin);    int cas;scanf("%d",&cas);    for (;cas--;){        root[0]=0;cnt=0;        scanf("%d%d",&n,&m);        for (int i=1;i<=n;i++)          {scanf("%d",&a[i]);p[i]=i;}        sort(p+1,p+n+1,cmp);        for (int i=1;i<=n;i++)b[p[i]]=i;        for (int i=1;i<=n;i++){            root[i]=root[i-1];            build(root[i],1,n,b[i]);        }        for (int i=1;i<=m;i++){            int x,y,k; scanf("%d%d%d",&x,&y,&k);            int t=query(root[x-1],root[y],1,n,k);            printf("%d\n",a[p[t]]);        }    }    return 0;} 
0 0
原创粉丝点击