线段树专题2-（加强版线段树-可持续化线段树）主席树 orz! ------用于解决区间第k大的问题----xdoj-1216

poj-2104(区间第K大问题）

 #include <iostream>
 #include <algorithm>
 #include <cstdio>
 #include <cstring>
 using namespace std;
 struct T {
     int ls;
     int rs;
     int sum;
 };
 const int N=1e5+;
 T tri[*N];
 int a[N],sort_a[N];
 int rt[N];
 int cnt;
 int n,m;
 void update (int l,int r,int rt1,int &rt2,int pos) {
     rt2=++cnt;
     tri[rt2]=tri[rt1];
     tri[rt2].sum++;// 在此处直接更新，就不用pushup了
     if (l==r) return ;
     int mid=(l+r)/;
     if (pos<=mid) update (l,mid,tri[rt1].ls,tri[rt2].ls,pos);
     else          update (mid+,r,tri[rt1].rs,tri[rt2].rs,pos);
     return ;
 }
 int  query (int l,int r,int rt1,int rt2,int k) {
     if (l==r)  return sort_a[l];
     int mid=(l+r)/;
     int sum=tri[tri[rt2].ls].sum-tri[tri[rt1].ls].sum;//我又出错了 sum应该是左子树的节点数 T_T !
     if (k<=sum)   return query (l,mid,tri[rt1].ls,tri[rt2].ls,k);
     else          return  query (mid+,r,tri[rt1].rs,tri[rt2].rs,k-sum);
 }
 int main ()
 {
     scanf ("%d %d",&n,&m);
     for (int i=;i<=n;i++) {
         scanf ("%d",&a[i]);
         sort_a[i]=a[i];
     }
     sort (sort_a+,sort_a++n);
     int num=;
     for (int i=;i<=n;i++) {
         if (sort_a[i]!=sort_a[i-])
             sort_a[++num]=sort_a[i];//离散化
     }
     rt[]=cnt=;
     tri[].sum=tri[].ls=tri[].rs=;
     for (int i=;i<=n;i++) {
         int pos=lower_bound (sort_a+,sort_a++num,a[i])-sort_a;
         update (,num,rt[i-],rt[i],pos);
     }
     for (int i=;i<=m;i++) {
         int x,y,k;
         scanf ("%d %d %d",&x,&y,&k);
         printf("%d\n",query (,num,rt[x-],rt[y],k));
     }
     return ;
 }

NO.2   HDOJ-4417(主席数的区间求和）

 #include<cstdio>
 #include<algorithm>
 const int N=1e5+;
 using namespace std;
 struct node {
     int ls;
     int rs;
     int sum;
 };
 node tri[*N];
 int rt[N];
 int a[N],sort_a[N+];
 int cnt,n,m,x,y,k;
 void updata (int l,int r,int rt1,int& rt2,int pos) {
     rt2=(++cnt);
     tri[rt2]=tri[rt1];
     tri[rt2].sum++;
     if (l==r) return ;
     int mid=(l+r)/;
     if (pos<=mid) updata(l,mid,tri[rt1].ls,tri[rt2].ls,pos);
     else          updata(mid+,r,tri[rt1].rs,tri[rt2].rs,pos);
     return ;
 }
 int query (int l,int r,int L,int R,int rt1,int rt2) {
     if (r<L||l>R) return ;
     if (l>=L&&r<=R) return tri[rt2].sum-tri[rt1].sum;
     int mid=(l+r)/;
     int t1=query(l,mid,L,R,tri[rt1].ls,tri[rt2].ls);
     int t2=query(mid+,r,L,R,tri[rt1].rs,tri[rt2].rs);
     return t1+t2;
 }
 int main ()
 {
     int T;
     int tt=;
     scanf ("%d",&T);
     while (T--) {
         printf("Case %d:\n",tt++);
         scanf ("%d %d",&n,&m);
         for (int i=;i<=n;i++) {
             scanf ("%d",&a[i]);
             sort_a[i]=a[i];
         }
         sort(sort_a+,sort_a++n);
         int num=;
         for (int i=;i<=n;i++) {
             if (sort_a[i]!=sort_a[num])
                 sort_a[++num]=sort_a[i];
         }
         cnt=;
         rt[]=;
         tri[].ls=tri[].rs=tri[].sum=;
         for (int i=;i<=n;i++) {
             int pos=lower_bound(sort_a+,sort_a++num,a[i])-sort_a;
             updata(,num,rt[i-],rt[i],pos);
         }
         for (int i=;i<=m;i++) {
             scanf ("%d %d %d",&x,&y,&k);
             int t=lower_bound(sort_a+,sort_a++num,k)-sort_a;
             if (t>num||sort_a[t]>k) t--;// 这个好  找最后一个小于等于k的数
             int ans=query (,num,,t,rt[x],rt[y+]);// 因为题目从0开始计数
             printf("%d\n",ans);
         }
     }
     return ;
 }

No.3 xdoj-1216(子树第k小）——dfs+主席树

 #include<cstdio>
 #include<algorithm>
 #include<vector>
 using namespace std;
 const int N=1e5+;
 struct T{
     int ls;
     int rs;
     int sum;
 };
 vector < vector <int> > g(N);
 T tri[N*];
 int rt[N],b[N],sort_b[N];
 int in[N],out[N];
 int val[N];
 int cnt,num;
 int n,q;
 int x,k;
 void dfs (int id,int pre) {
     num++;
     sort_b[num]=b[num]=val[id];
     in[id]=num;
     for (int i=;i<g[id].size();i++) {
         int  next=g[id][i];
         if (next==pre) continue;
         dfs (next,id);
     }
     out[id]=num;
 }
 void updata (int l,int r,int pos,int rt1,int& rt2) {
     cnt++;
     rt2=cnt;
     tri[cnt]=tri[rt1];
     tri[cnt].sum++;
     if (l==r) return ;
     int mid=(l+r)/;
     if (pos<=mid) updata (l,mid,pos,tri[rt1].ls,tri[rt2].ls);
     else          updata (mid+,r,pos,tri[rt1].rs,tri[rt2].rs);
     return ;
 }
 int query (int l,int r,int k,int rt1,int rt2) {
     if (l==r) return l;
     int mid=(l+r)/;
     int ans;
     int sum=tri[tri[rt2].ls].sum-tri[tri[rt1].ls].sum;
     if (k<=sum) ans=query (l,mid,k,tri[rt1].ls,tri[rt2].ls);
     else        ans=query (mid+,r,k-sum,tri[rt1].rs,tri[rt2].rs);
     return ans;
 }
 int main ()
 {
     while (scanf ("%d",&n)!=EOF) {
         for (int i=;i<=n;i++) g[i].clear();
         for (int i=;i<=n;i++)
             scanf ("%d",&val[i]);
         int u,v;
         for (int i=;i<=n-;i++) {
             scanf ("%d %d",&u,&v);
             g[u].push_back(v);
             g[v].push_back(u);
         }
         num=;
         dfs (,-);// 按理说有根树 直接dfs(1)就可以了可是这样答案是不对的 是我理解错了还是数据有问题呢
         sort(sort_b+,sort_b++n);
         int nn=;
         for (int i=;i<=n;i++)
             if (sort_b[i]!=sort_b[nn])
                 sort_b[++nn]=sort_b[i];
        rt[]=;
        tri[].rs=tri[].ls=tri[].sum=;
        cnt=;
        for (int i=;i<=n;i++) {
             int pos=lower_bound(sort_b+,sort_b++nn,b[i])-sort_b;
             updata(,nn,pos,rt[i-],rt[i]);
        }
        scanf ("%d",&q);
        for (int i=;i<=q;i++)  {
             scanf ("%d %d",&x,&k);
             int ans=query (,nn,k,rt[in[x]-],rt[out[x]]);
             printf ("%d\n",sort_b[ans]);
        }
     }
     return ;
 }