[bzoj3343]&&[洛谷2810] [分块][块内排序][二分查找]教主的魔法

来源:互联网 发布:vb打开文件夹 编辑:程序博客网 时间:2024/06/06 20:43

bzoj3343&&洛谷2810 教主的魔法

Description
教主最近学会了一种神奇的魔法,能够使人长高。于是他准备演示给XMYZ信息组每个英雄看。于是N个英雄们又一次聚集在了一起,这次他们排成了一列,被编号为1、2、……、N。
每个人的身高一开始都是不超过1000的正整数。教主的魔法每次可以把闭区间[L, R](1≤L≤R≤N)内的英雄的身高全部加上一个整数W。(虽然L=R时并不符合区间的书写规范,但我们可以认为是单独增加第L(R)个英雄的身高)
CYZ、光哥和ZJQ等人不信教主的邪,于是他们有时候会问WD闭区间 [L, R] 内有多少英雄身高大于等于C,以验证教主的魔法是否真的有效。
WD巨懒,于是他把这个回答的任务交给了你。
Input
第1行为两个整数N、Q。Q为问题数与教主的施法数总和。
第2行有N个正整数,第i个数代表第i个英雄的身高。
第3到第Q+2行每行有一个操作:
(1) 若第一个字母为“M”,则紧接着有三个数字L、R、W。表示对闭区间 [L, R] 内所有英雄的身高加上W。
(2) 若第一个字母为“A”,则紧接着有三个数字L、R、C。询问闭区间 [L, R] 内有多少英雄的身高大于等于C。
Output
对每个“A”询问输出一行,仅含一个整数,表示闭区间 [L, R] 内身高大于等于C的英雄数。
Sample Input
5 3
1 2 3 4 5
A 1 5 4
M 3 5 1
A 1 5 4
Sample Output
2
3
HINT
【输入输出样例说明】
原先5个英雄身高为1、2、3、4、5,此时[1, 5]间有2个英雄的身高大于等于4。教主施法后变为1、2、4、5、6,此时[1, 5]间有3个英雄的身高大于等于4。
【数据范围】
对30%的数据,N≤1000,Q≤1000。
对100%的数据,N≤1000000,Q≤3000,1≤W≤1000,1≤C≤1,000,000,000。

【题解】
第一次写分块,,,,感觉真是不错,,比线段树随便150行的感觉真的好太多了,,,。而且这道题线段树好像只能修改,并不能查询比c大的个数,,,。。然而查询最快的当然就是二分。
所有这道题的做法就是分块加二分,分块包括块内排序。
【代码】

#include<iostream>#include<cstdio>#include<cstring>#include<cmath>#include<algorithm>#define LL long long#ifdef WIN32#define AUTO "%I64d"#else #define AUTO "%lld"#endifusing namespace std;const int N = 1000050;int n,q;int block,m;//logn的个数,共有mint pos[N];//在第几块int a[N],b[N];int add[1005];//块中加法标记void reset(int x){    int l=(x-1)*block+1,r=min(x*block,n);//防止最后一块,,起终    for(int i = l ; i <= r; i++)        b[i]=a[i];//重置块内,updata后    sort(b+l,b+r+1);}int find(int x,int v){    int l=(x-1)*block+1,r=min(x*block,n);    int last=r;    while(l<=r){        int mid=(l+r)>>1;        if(b[mid]<v)l=mid+1;        else r=mid-1;//大于等于    }    return last-l+1;//是减l!!!!!!!!!,最后l==r,}void updata(int x,int y,int v){    if(pos[x]==pos[y])for(int i = x; i <= y; i++)a[i]+=v;//整块不完整    else {        for(int i = x; i <=pos[x]*block; i++)a[i]+=v;        for(int i = (pos[y]-1)*block+1; i <= y; i++)a[i]+=v;    }//暴力两边区间    reset(pos[x]);reset(pos[y]);//重置块内    for(int i = pos[x]+1; i <pos[y]; i++)        add[i]+=v;//标记整块}int query(int x,int y,int v){    int sum=0;    if(pos[x]==pos[y])    {        for(int i = x; i<=y; i++)            if(a[i]+add[pos[i]] >= v)sum++;    }    else     {        for(int i = x; i <= pos[x]*block; i++)            if(a[i] + add[pos[i]] >= v)sum++;        for(int i = (pos[y]-1)*block+1; i<=y;i++)            if(a[i] +add[pos[i]]>=v)sum++;    }    for(int i = pos[x]+1; i<pos[y];i++)        sum+=find(i,v-add[i]);//除去标记的答案    return sum;}int main(){    freopen("cod.in","r",stdin);    freopen("cod.out","w",stdout);    scanf("%d%d",&n,&q);    block=int(sqrt(n));    if(n%block)m = n/block + 1;    else m = n / block;    for(int i = 1; i <= n; i++){        scanf(AUTO,&a[i]);        pos[i]=(i-1)/block+1;        b[i]=a[i];    }    for(int i = 1; i<= m; i++)reset(i);//块内排序    for(int i = 1; i <= q; i++){        char ch[5];        int x,y,c;        scanf("%s%d%d%d",ch,&x,&y,&c);        if(ch[0]=='M')updata(x,y,c);        else printf(AUTO "\n",query(x,y,c));    }    return 0;}

写分块还是挺舒服,。。。。。比线段树好,可以求很多东东,,,好吧虽然我也是第一次写。分块的复杂度也是O(nlogn)

hzwer

0 0