P5459 [BJOI2016]回转寿司 （动态开点的权值线段树）

题目链接：点击这里

题目大意：
给定一个长度为 $n$ 的序列 ，现在要从中选出一段连续子序列 $[l,r]$ ，使得 $L\le {\sum }_{i=l}^r{a}_i\le R$ ，求符合条件的子序列数量

题目分析：
我们可以在 $1-n$ 范围内对 $r$ 进行枚举， $logn$ 的复杂度求解此时符合条件的 $l$ 的数目，求和即可
记 $su{m}_x={\sum }_{i=1}^x{a}_i$ ，故 ${\sum }_{i=l}^r{a}_i=su{m}_r-su{m}_{l-1}$，故式子 $L\le {\sum }_{i=l}^r{a}_i\le R$ 可化为 $L\le su{m}_r-su{m}_{l-1}\le R$ 可继续化为 $su{m}_r-R\le su{m}_{l-1}\le su{m}_r-L$ ，此时问题就转换成了在 $[su{m}_r-R,su{m}_r-L]$ 中找符合条件的 $su{m}_{l-1}(l\in [1,r])$ 的数目
而此时还有一个问题就是这 $L,R$很大，并且没办法离散化(也可能是我太菜了)，所以还要用动态开点建立线段树

具体细节见代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define ll long long
#define inf 0x3f3f3f3f
#define Inf 0x3f3f3f3f3f3f3f3f
using namespace std;
int read()
{
	int res = 0,flag = 1;
	char ch = getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch == '-') flag = -1;
		ch = getchar();
	}
	while(ch>='0' && ch<='9')
	{
		res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
		ch = getchar();
	}
	return res*flag;
}
const int maxn = 1e5+5;
const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-8;
struct segmenttree{
	int root,cnt;
	int lson[maxn<<8],rson[maxn<<8],sum[maxn<<8];
	void pushup(int now)
	{
		sum[now] = sum[lson[now]]+sum[rson[now]]; 
	}
	void update(int &now,ll pos,ll val,ll l = -Inf,ll r = Inf)
	{
		if(!now) now = ++cnt;
		if(l == r)
		{
			sum[now] += val;
			return ;
		}
		ll mid = (l+r)>>1;
		if(mid >= pos) update(lson[now],pos,val,l,mid);
		else update(rson[now],pos,val,mid+1,r);
		pushup(now);
	}
	ll query(int &now,ll ql,ll qr,ll l = -Inf,ll r = Inf)
	{
		if(!now) now = ++cnt;
		if(ql > r || qr < l) return 0;
		if(ql <= l && qr >= r) return sum[now];
		ll mid = (l+r)>>1;
		return query(lson[now],ql,qr,l,mid)+query(rson[now],ql,qr,mid+1,r);
	}
}tree;
int n,l,r;
ll ans,sum[maxn];
int main()
{
	n = read(),l = read(),r = read();
	for(int i = 1;i <= n;i++)
	{
		int x = read();
		sum[i] = sum[i-1]+x;
	}
	tree.update(tree.root,sum[0],1); //插入sum[0] 
	for(int i = 1;i <= n;i++) //枚举右端点 
	{
		ans += tree.query(tree.root,sum[i]-r,sum[i]-l);
		tree.update(tree.root,sum[i],1); //插入sum[i] 
	}
	printf("%lld\n",ans);
	return 0;
}