学习了一下动态开树的技巧&&值域线段树……
PRoblem Description 酷爱日料的小Z经常光顾学校东门外的回转寿司店。在这里,一盘盘寿司 通过传送带依次呈现在小Z眼前。 不同的寿司带给小Z的味觉感受是不一样的,我们定义小Z对每盘寿司都 有一个满意度,例如小Z酷爱三文鱼,他对一盘三文鱼寿司的满意度为10; 小Z觉得金枪鱼没有什么味道,他对一盘金枪鱼寿司的满意度只有5;小Z最近 看了电影“美人鱼”,被里面的八爪鱼恶心到了,所以他对一盘八爪鱼刺身的 满意度是-100。 特别地,小Z是个著名的吃货,他吃回转寿司有一个习惯,我们称之 为“狂吃不止”。具体地讲,当他吃掉传送带上的一盘寿司后,他会毫不犹豫 地吃掉它后面的寿司,直到他不想再吃寿司了为止。 今天,小Z再次来到了这家回转寿司店,N盘寿司将依次经过他的面前, 其中,小Z对第i盘寿司的满意度为Ai。小Z可以选择从哪盘寿司开始吃,也可 以选择吃到哪盘寿司为止,他想知道共有多少种不同的选择,使得他的满意度 之和不低于L,且不高于R。注意,虽然这是回转寿司,但是我们不认为这是 一个环上的问题,而是一条线上的问题。即,小Z能吃到的是输入序列的一个 连续子序列;最后一盘转走之后,第一盘并不会再出现一次。 Input 第一行包含三个整数N,L和R,分别表示寿司盘数,满意度的下限和上 限。 第二行包含N个整数Ai,表示小Z对寿司的满意度。 对20%的数据,1≤N≤2000. 另有30%的数据,Ai≥0. 对100%的数据,N≤100000,|Ai|≤100000,0≤L, R≤109. Output 仅一行,包含一个整数,表示共有多少种选择可以使得小Z的满意度之和 不低于L且不高于R。 Sample Input 【输入样例1】 5 5 9 1 2 3 4 5 【输入样例2】 4 2 4 2 -1 2 3 Sample Output 【输出样例1】 6 【输出样例2】 5 首先预处理出前缀和s[i],先在值域线段树中插入值0,每次对于s[i],查询值域属于[s[i]-R,s[j]-L]的前缀个数,累加答案,在将当前前缀插入线段树。注意longlong的使用,被算法库的find坑了QAQ……以后起变量名最好起好一点……不然就会造成编译错误调了大半天……
#include <cstdio>#include <algorithm>#define LL long long#define maxn 2000000#define find Findusing namespace std;LL INF=99999999999999;struct xx{ int l,r; int g;}a[maxn];int rt,nd,x,n,m,l,r;LL v,tmp,ans,L,R;int ins(int &t,LL l,LL r){ if (!t) t=++nd; a[t].g++; if (l==r) return 0; LL mid=l+((r-l)>>1); if (v<=mid) ins(a[t].l,l,mid); if (v>mid) ins(a[t].r,mid+1,r);}LL find(int t,LL l,LL r){ if (!t) return 0; if (L<=l && r<=R) return a[t].g; int midd=l+r>>1; LL tmp=0; if (L<=midd) tmp+=find(a[t].l,l,midd); if (R>midd) tmp+=find(a[t].r,midd+1,r); return tmp;}int main(){ scanf("%d%d%d",&n,&l,&r); v=0;ins(rt,-INF,INF); for (int i=1;i<=n;i++) { scanf("%d",&x);tmp+=x; L=tmp-r;R=tmp-l;v=tmp; int hh=1; ans+=find(hh,-INF,INF); ins(rt,-INF,INF); } printf("%lld",ans);}新闻热点
疑难解答