值域线段树 (玲珑OJ 1117)

点击打开题目链接

题目意思很简单:

1、插入x
2、把小于x的数变成x
3、把大于x的数变成x
4、求集合中第x小数
5、求集合中小于x的数个数

思路: 线段树,节点是值的分数,你可以离散,也可以不离散,直接标记;我的写法是:   看代码注释>>>

据说数组改为指针会快点;代码比较挫.存一个;

 #include <cstdio>
 #include <cstring>
 #include <cstdlib>
 #include <vector>
 #include <iostream>
 using namespace std;

 typedef long long int LL;
 const LL INF=;
 const int maxn=6e6+;

 struct ACM
 {
     struct segment
     {
         int val,lson,rson;
         bool mark;
     } seg[maxn];
     int sz;
     /***
         开辟新的节点并初始化
     */
     int newnode()
     {
         sz++;
         if(sz>=maxn)
         {
             int lzq=,zhangpan=;
             lzq=lzq/zhangpan;

         }
         seg[sz].lson=seg[sz].rson=-;
         seg[sz].val=,seg[sz].mark=false;
         return sz;
     }
     /***
         初始化根节点
     */
     void init(int& root)
     {
         sz=-;
         root=newnode();
     }
     /***
         标记下放操作,如果没有节点开辟新的节点;
     */
     void pushdown(int i)
     {
         int l=seg[i].lson,r=seg[i].rson;
         if(l==-) seg[i].lson=newnode();
         if(r==-) seg[i].rson=newnode();
         if(seg[i].mark)
         {
             seg[i].mark=false;
             seg[l].mark=seg[r].mark=true;
             seg[l].val=seg[r].val=;
         }
     }
     void pushup(int i)
     {
         int l=seg[i].lson,r=seg[i].rson;
         seg[i].val=seg[l].val+seg[r].val;
     }
     /***
         插入函数 ; 在 pos 位置上面加上 val;
     */
     void ins(int i,int l,int r,int pos,int val)
     {
         if(l==r)
         {
             seg[i].val+=val;
             return ;
         }
         pushdown(i);
         int mid=(l+r)>>;
         if(pos<=mid) ins(seg[i].lson,l,mid,pos,val);
         else ins(seg[i].rson,mid+,r,pos,val);
         pushup(i);
     }
     /***
         将 L,R 这段区间上的值,更新为 0;
     */
     void del(int i,int l,int r,int L,int R)
     {
         if(l==L&&r==R)
         {
             seg[i].val=;
             seg[i].mark=true;
             return ;
         }
         pushdown(i);
         int mid=(l+r)>>;
         if(R<=mid) del(seg[i].lson,l,mid,L,R);
         else if(L>mid) del(seg[i].rson,mid+,r,L,R);
         else del(seg[i].lson,l,mid,L,mid),del(seg[i].rson,mid+,r,mid+,R);
         pushup(i);
     }
     int find(int i,int l,int r,int x) //第x小的数字;
     {
         if(l==r) return l;
         pushdown(i);
         int lval=;
         int mid=(l+r)>>;
         if(seg[i].lson!=-) lval=seg[seg[i].lson].val;
         if(x<=lval) return find(seg[i].lson,l,mid,x);
         else return find(seg[i].rson,mid+,r,x-lval);
     }
     int query(int i,int l,int r,int L,int R) //区间 数的个数;
     {
         if(l==L&&r==R) return seg[i].val;
         pushdown(i);
         int mid=(l+r)>>;
         if(R<=mid) return query(seg[i].lson,l,mid,L,R);
         else if(L>mid) return query(seg[i].rson,mid+,r,L,R);
         else return query(seg[i].lson,l,mid,L,mid)+query(seg[i].rson,mid+,r,mid+,R);
     }
 } AC;

 int main()
 {
     int m;
     scanf("%d",&m);
     int op,x,root,temp;
     AC.init(root);
     while(m--)
     {
         scanf("%d%d",&op,&x);
         switch(op)
         {
         case :
             AC.ins(root,,INF,x,);
             break;
         case :
             temp=AC.query(root,,INF,,x);
             AC.del(root,,INF,,x);
             AC.ins(root,,INF,x,temp);
             break;
         case :
             temp=AC.query(root,,INF,x,INF);
             AC.del(root,,INF,x,INF);
             AC.ins(root,,INF,x,temp);
             break;
         case :
             printf("%d\n",AC.find(root,,INF,x));
             break;
         case :
             printf("%d\n",AC.query(root,,INF,,x-));
             break;
         }
     }
     return ;
 }

代码二:

 #include <cstdio>
 #include <cstring>
 #include <cstdlib>
 #include <vector>
 #include <iostream>
 using namespace std;

 const int INF=;
 const int maxn=6e6+;
 struct ACM
 {
     /***
         对于 2,3 操作,先查询个数,再插入这个数字:
         意思就是更新到底部,加上查询出来的个数;
         并加上清零的标记;

     */
     struct Node
     {
         int val,lson,rson;
         bool mark;
     } seg[maxn];
     int sz;
     int newnode()
     {
         sz++;
         seg[sz].val=;
         seg[sz].lson=-;
         seg[sz].rson=-;
         seg[sz].mark=false;
         return sz;
     }
     void init(int& root)
     {
         sz=-;
         root=newnode();
     }
     void pushdown(int i)
     {
         int l=seg[i].lson,r=seg[i].rson;
         if(l==-) seg[i].lson=newnode();
         if(r==-) seg[i].rson=newnode();
         if(seg[i].mark)
         {
             seg[i].mark=false;
             seg[l].mark=seg[r].mark=true;
             seg[l].val=seg[r].val=;
         }
     }
     void pushup(int i)
     {
         int l=seg[i].lson,r=seg[i].rson;
         seg[i].val=seg[l].val+seg[r].val;
     }
     int query(int i,int l,int r,int L,int R) //区间 数的个数;
     {
         if(l==L&&r==R) return seg[i].val;
         pushdown(i);
         int mid=(l+r)>>;
         if(R<=mid) return query(seg[i].lson,l,mid,L,R);
         else if(L>mid) return query(seg[i].rson,mid+,r,L,R);
         else return query(seg[i].lson,l,mid,L,mid)+query(seg[i].rson,mid+,r,mid+,R);
     }
     void ins(int i,int l,int r,int pos,int val)
     {
         if(l==r)
         {
             seg[i].val+=val;
             return ;
         }
         pushdown(i);
         int mid=(l+r)>>;
         if(pos<=mid) ins(seg[i].lson,l,mid,pos,val);
         else ins(seg[i].rson,mid+,r,pos,val);
         pushup(i);
     }
     void operator2(int i,int l,int r,int x,int val)
     {
         if(l==r)
         {
             seg[i].val=val;
             return ;
         }
         int mid=(l+r)>>;
         pushdown(i);
         if(x<=mid) operator2(seg[i].lson ,l,mid,x,val);
         else
         {
             seg[seg[i].lson].mark=true;
             seg[seg[i].lson].val=;//如果，存在加到某一边，不存在也需要开新的节点！！
             operator2(seg[i].rson,mid+,r,x,val);
         }
         pushup(i);
     }
     void operator3(int i,int l,int r,int x,int val)
     {
         if(l==r)
         {
             seg[i].val=val;
             return ;
         }
         pushdown(i);
         int mid=(l+r)>>;
         if(x<=mid)
         {
             seg[seg[i].rson].mark=true;
             seg[seg[i].rson].val=;
             operator3(seg[i].lson,l,mid,x,val);
         }
         else operator3(seg[i].rson,mid+,r,x,val);
         pushup(i);
     }
     int operator4(int i,int l,int r,int x)
     {
         if(l==r) return l;
         int mid=(l+r)>>;
         pushdown(i);
         int val=seg[seg[i].lson].val;
         if(x<=val) return operator4(seg[i].lson,l,mid,x);
         else return operator4(seg[i].rson,mid+,r,x-val);
     }
 } AC;

 int main()
 {

     int m;
     while(scanf("%d",&m)!=EOF)
     {
         int op,x,root,temp;
         AC.init(root);
         while(m--)
         {
             scanf("%d%d",&op,&x);
             switch(op)
             {
             case :
                 AC.ins(root,,INF,x,);
                 break;
             case :
                 temp=AC.query(root,,INF,,x);
                 AC.operator2(root,,INF,x,temp);
                 break;
             case :
                 temp=AC.query(root,,INF,x,INF);
                 AC.operator3(root,,INF,x,temp);
                 break;
             case :
                 printf("%d\n",AC.operator4(root,,INF,x));
                 break;
             case :
                 printf("%d\n",AC.query(root,,INF,,x-));
                 break;
             }
         }
     }
     return ;
 }