值域线段树 bzoj 4627

　　　　　这是题目链接4627: [BeiJing2016]回转寿司

题目大意:

给定n个数,求有多少个字段和在 满足 L<=sum<=R;

解题思路

需要解这个题目,需要有线段树加可持续化的思想,但是这个题目只需要上一棵线段树的信息,所以可以不用主席树,只要用到值域线段树。那么，这样就可以把问题转化为：

先处理处前缀和。 求 L<=sum[j]-sum[i]<=R (0 <=i < j<=n) 有多少个?

那么我们可以用值域线段树搞一下,线段树节点存值的个数,每次把前缀和这个值插入,每次查询,原问题就等同于 当前线段树中有多少个值是在 [ sum[j]-R , sum[j]-L ] 这个区间,

意思就是,每次把上一个前缀和插入需要计算一下上一棵线段树值在[ sum[j]-R , sum[j]-L ] 个数。

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

 typedef long long int LL;
 const LL INF=10000000000ll;
 const int maxn=*;
 struct ACM
 {
     struct Node
     {
         LL val;
         int lson,rson;
     }seg[maxn];
     int sz;
     int newnode()
     {
         sz++;
         seg[sz].val=;
         seg[sz].rson=-;
         seg[sz].lson=-;
         return sz;
     }
     void init()
     {
         sz=-;
     }
     void update(int &root,LL l,LL r,LL val)
     {
         if(root==-) root=newnode();
         seg[root].val++;
         if(l==r) return ;
         LL mid=(l+r)>>;
         if(val<=mid) update(seg[root].lson,l,mid,val);
         else update(seg[root].rson,mid+,r,val);
     }
     LL query(int root,LL l,LL r,LL ql,LL qr)
     {
         if(root==-) return ;
         if(l==ql&&r==qr) return seg[root].val;
         LL mid=(l+r)>>;
         if(qr<=mid) return query(seg[root].lson,l,mid,ql,qr);
         else if(ql>mid) return query(seg[root].rson,mid+,r,ql,qr);
         else return query(seg[root].lson,l,mid,ql,mid)+query(seg[root].rson,mid+,r,mid+,qr);
     }
     /**
     void debug(int i)
     {
         if(i==-1) return ;
         printf("node=%d val=%lld lson=%d rson=%d\n",i,seg[i].val,seg[i].lson,seg[i].rson);
         debug(seg[i].lson),debug(seg[i].rson);
     }
     */
 }AC;

 int main()
 {
     int n,L,R,root=-;
     scanf("%d%d%d",&n,&L,&R);
     LL sum=,ans=;
     AC.init();
     for(int i=; i<=n; i++)
     {
         int num;
         scanf("%d",&num);
         AC.update(root,-INF,INF,sum);
         sum+=num;
         ans+=AC.query(root,-INF,INF,sum-R,sum-L);
     }
     printf("%lld\n",ans);
     return ;
 }