【bzoj4627】[BeiJing2016]回转寿司 离散化+树状数组

题目描述

给出一个长度为n的序列,求所有元素的和在[L,R]范围内的连续子序列的个数。

输入

第一行包含三个整数N,L和R,分别表示寿司盘数,满意度的下限和上限。
第二行包含N个整数Ai,表示小Z对寿司的满意度。
N≤100000,|Ai|≤100000,0≤L, R≤10^9

输出

仅一行,包含一个整数,表示共有多少种选择可以使得小Z的满意度之和
不低于L且不高于R。

样例输入

5 5 9
1 2 3 4 5

样例输出

6


题解

离散化+树状数组

把序列和转化为前缀相减,即选出满足$L\le sum[x]-sum[y]\le R$的$x>y$的数对个数。

那么我们枚举$x$,即可得到$y$的范围,要求的是以前的满足条件的$y$的个数。可以维护1到当前位置树状数组,在树状数组中查询个数,最后再把该数加入到树状数组中。由于数据范围大,因此需要离散化。

时间复杂度$O(n\log n)$

#include <cstdio>
#include <algorithm>
#define N 100010
#define now v + 1 , v + n + 2
using namespace std;
typedef long long ll;
ll sum[N] , v[N];
int f[N] , n;
inline void add(int x)
{
    int i;
    for(i = x ; i <= n + 1 ; i += i & -i) f[i] ++ ;
}
inline int query(int x)
{
    int i , ans = 0;
    for(i = x ; i ; i -= i & -i) ans += f[i];
    return ans;
}
int main()
{
    int i;
    ll l , r , ans = 0;
    scanf("%d%lld%lld" , &n , &l , &r);
    for(i = 1 ; i <= n ; i ++ ) scanf("%lld" , &sum[i]) , sum[i] += sum[i - 1] , v[i] = sum[i];
    sort(now);
    for(i = 0 ; i <= n ; i ++ ) ans += query(upper_bound(now , sum[i] - l) - v - 1) - query(lower_bound(now , sum[i] - r) - v - 1) , add(lower_bound(now , sum[i]) - v);
    printf("%lld\n" , ans);
    return 0;
}

 

 

|转载请注明 [原文链接][作者] ,谢谢!

posted @ 2017-10-13 18:50  GXZlegend  阅读(362)  评论(0编辑  收藏  举报