首页 > 代码库 > 【poj2104-求区间第k大数(不修改)】主席树/可持续化线段树

【poj2104-求区间第k大数(不修改)】主席树/可持续化线段树

第一道主席树~然而是道比较水的。。。因为它不用修改。。。

转载一个让我看懂的主席树的讲解吧:http://blog.csdn.net/regina8023/article/details/41910615 (未授权,侵权删)

---------------------------------------------------------------------------------------------------------------------

技术分享

 

那么如果要询问i-j之间数字出现的次数怎么办呢?

因为每一棵线段树的区间都是相同的,所以要求l-r之间的数字的出现次数只要用前r位出现的次数减去前l-1位出现的次数,就是ans

 

但是如果有修改操作怎么办?

如果沿用上面的做法,那么修改操作是O(nlogn)的,查询是O(1)的,修改要花好长时间。。。

前缀和联想到了树状数组,那么将前缀和用树状数组维护的话修改是O(logn*logn),查询时O(logn),查询的时间虽然变长,但是修改的时间缩短许多!!

 

注意:

函数式线段树的数组要开大一点!!

---------------------------------------------------------------------------------------------------------------------

 

这题就是模版题啦,先离散,求区间第k大的时候lx=root[l-1],rx=root[r],两边同时走不断作差,看看左孩子的数量,如果k更大就减掉左孩子的到有右孩子中找。

代码:

  1 #include<cstdio>
  2 #include<cstdlib>
  3 #include<cstring>
  4 #include<cmath>
  5 #include<iostream>
  6 #include<algorithm>
  7 using namespace std;
  8 
  9 const int N=100010,INF=(int)1e9+100;
 10 struct trnode{
 11     int lc,rc,cnt;
 12 }t[30*N];
 13 struct node{
 14     int d,id;
 15 }p[N];
 16 int n,m,tl,mx;
 17 int a[N],num[N],root[N];
 18 
 19 bool cmp(node x,node y){return x.d<y.d;}
 20 
 21 int bt(int l,int r)
 22 {
 23     int x=++tl;
 24     // a[x].l=l;a[x].r=r;
 25     t[x].lc=t[x].rc=0;
 26     t[x].cnt=0;
 27     if(l<r)
 28     {
 29         int mid=(l+r)/2;
 30         t[x].lc=bt(l,mid);
 31         t[x].rc=bt(mid+1,r);
 32     }
 33     return x;
 34 }
 35 
 36 int add(int rt,int x)
 37 {
 38     int now=++tl,tmp=now;
 39     t[now].cnt=t[rt].cnt+1;
 40     int l=1,r=mx,mid;
 41     while(l<r)
 42     {
 43         mid=(l+r)/2;
 44         if(x<=mid)
 45         {
 46             t[now].lc=++tl;
 47             t[now].rc=t[rt].rc;
 48             rt=t[rt].lc;
 49             now=tl;
 50             r=mid;
 51         }
 52         else
 53         {
 54             t[now].lc=t[rt].lc;
 55             t[now].rc=++tl;
 56             rt=t[rt].rc;
 57             now=tl;
 58             l=mid+1;
 59         }
 60         t[now].cnt=t[rt].cnt+1;
 61     }
 62     return tmp;
 63 }
 64 
 65 int query(int lx,int rx,int k)
 66 {
 67     int l=1,r=mx,mid;
 68     while(l<r)
 69     {
 70         mid=(l+r)/2;
 71         if(t[t[rx].lc].cnt-t[t[lx].lc].cnt>=k)
 72         {
 73             r=mid;
 74             lx=t[lx].lc;
 75             rx=t[rx].lc;
 76         }
 77         else
 78         {
 79             l=mid+1;
 80             k-=t[t[rx].lc].cnt-t[t[lx].lc].cnt;
 81             lx=t[lx].rc;
 82             rx=t[rx].rc;
 83         }
 84     }
 85     return l;
 86 }
 87 
 88 void output(int x)
 89 {
 90     printf("x = %d  lc = %d  rc = %d  cnt = %d\n",x,t[x].lc,t[x].rc,t[x].cnt);
 91     if(t[x].lc) output(t[x].lc);
 92     if(t[x].rc) output(t[x].rc);
 93 }
 94 
 95 int main()
 96 {
 97     freopen("a.in","r",stdin);
 98     while(scanf("%d%d",&n,&m)!=EOF)
 99     {
100         tl=0;mx=0;
101         for(int i=1;i<=n;i++)
102         {
103             scanf("%d",&a[i]);
104             p[i].d=a[i];p[i].id=i;
105         }
106         sort(p+1,p+1+n,cmp);
107         p[0].d=INF;
108         for(int i=1;i<=n;i++)
109         {
110             if(p[i].d!=p[i-1].d) mx++,num[mx]=p[i].d;
111             a[p[i].id]=mx;
112         }
113         root[0]=bt(1,mx);
114         for(int i=1;i<=n;i++)
115             root[i]=add(root[i-1],a[i]);
116         // output(root[5]);
117         for(int i=1;i<=m;i++)
118         {
119             int l,r,k;
120             scanf("%d%d%d",&l,&r,&k);
121             printf("%d\n",num[query(root[l-1],root[r],k)]);
122         }
123     }
124     return 0;
125 }

 

【poj2104-求区间第k大数(不修改)】主席树/可持续化线段树