[BJOI2016] 回转寿司
题目描述
酷爱日料的小Z经常光顾学校东门外的回转寿司店。在这里,一盘盘寿司通过传送带依次呈现在小Z眼前。
不同的寿司带给小Z的味觉感受是不一样的,我们定义小Z对每盘寿司都有一个满意度。
例如小Z酷爱三文鱼,他对一盘三文鱼寿司的满意度为 10 10 10;小Z觉得金枪鱼没有什么味道,他对一盘金枪鱼寿司的满意度只有 5 5 5;小Z最近看了电影《美人鱼》,被里面的八爪鱼恶心到了,所以他对一盘八爪鱼刺身的满意度是 − 100 -100 −100。
特别地,小Z是个著名的吃货,他吃回转寿司有一个习惯,我们称之为“狂吃不止”。具体地讲,当他吃掉传送带上的一盘寿司后,他会毫不犹豫地吃掉它后面的寿司,直到他不想再吃寿司了为止。
今天,小Z再次来到了这家回转寿司店, N N N 盘寿司将依次经过他的面前。其中,小Z对第 i i i 盘寿司的满意度为 a i a_i ai。
小Z可以选择从哪盘寿司开始吃,也可以选择吃到哪盘寿司为止。他想知道共有多少种不同的选择,使得他的满意度之和不低于 L L L,且不高于 R R R。
注意,虽然这是回转寿司,但是我们不认为这是一个环上的问题,而是一条线上的问题。即,小Z能吃到的是输入序列的一个连续子序列;最后一盘转走之后,第一盘并不会再出现一次。
输入格式
第一行三个正整数
N
,
L
,
R
N,L,R
N,L,R,表示寿司盘数,满意度的下限和上限。
第二行包含
N
N
N 个整数
a
i
a_i
ai,表示小Z对寿司的满意度。
输出格式
一行一个整数,表示有多少种方案可以使得小Z的满意度之和不低于 L L L 且不高于 R R R。
样例 #1
样例输入 #1
5 5 9
1 2 3 4 5
样例输出 #1
6
提示
【数据范围】
1
≤
N
≤
1
0
5
1\le N \le 10^5
1≤N≤105
∣
a
i
∣
≤
1
0
5
|a_i| \le 10^5
∣ai∣≤105
0
≤
L
,
R
≤
1
0
9
0\le L,R \le 10^9
0≤L,R≤109
思路
这道题根据题目意思,我们可以很明显的知道一下三维关系:
l < r l< r l<r s [ r ] − s [ l ] ≤ R s[r]-s[l]\le R s[r]−s[l]≤R L ≤ s [ r ] − s [ l ] L\le s[r]-s[l] L≤s[r]−s[l]
如果正常做的话,我们肯定是把它变成:
l
<
r
l< r
l<r
−
s
[
l
]
≤
−
s
[
r
]
−
R
-s[l]\le -s[r]-R
−s[l]≤−s[r]−R
s
[
l
]
≤
s
[
r
]
−
L
s[l]\le s[r]-L
s[l]≤s[r]−L
但这个式子非同寻常,无法用我们的模板来解决,因此我们就得换一种思路:因此我们可以采用取尺法解决:
先对小区间排序,然后拿双指针(相邻的)去看是否满足
≥
L
\ge L
≥L 或者
≤
R
\le R
≤R,最后的答案就是双指针之间的答案了。
代码(时间复杂度 O ( n l o g 2 n ) O(nlog^2n) O(nlog2n))
//满足两个性质,L<=w[i],w[j]<=R 且 i<j => 此时满足CDQ分治
#include<iostream>
#include<algorithm>
#define int long long
using namespace std;
const int N = 1e6+10;
int n,L,R;
int s[N];
int ans;
void CDQ(int l,int r){
if(l==r)return;
int mid=l+r>>1ll;
CDQ(l,mid),CDQ(mid+1,r);
int i=l,j=l-1;
sort(s+l,s+mid+1),sort(s+mid+1,s+r+1);
for(int k=mid+1;k<=r;k++){
while(j+1<=mid&&s[k]-s[j+1]>=L)j++;
while(i<=mid&&s[k]-s[i]>R)i++;
ans+=j-i+1;
}
}
signed main(){
cin>>n>>L>>R;
for(int i=1;i<=n;i++){
cin>>s[i];
s[i]+=s[i-1];
}
CDQ(0,n);
cout<<ans;
return 0;
}