[BZOJ 3343]教主的魔法-CSDN博客

本文链接：https://blog.csdn.net/Horizon_SMZ/article/details/47255649

3343: 教主的魔法

Time Limit: 10 Sec Memory Limit: 256 MB
Submit: 570 Solved: 247
[ Submit][ Status][ Discuss]

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。

Source

引用黄学长的话---第一次正经的写分块

结果是：错误百出

让我来总结一下出现的错误、先贴代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 1000010
using namespace std;
bool read(){
	char ch=getchar();
	while(ch<'!')ch=getchar();
	return ch=='M';
}
int n,Q,t;
struct P{
	int h,pos;
	bool operator<(const P& k)const{return h>k.h;}
}p[maxn];
int fa[maxn],set[1005];
void update(int l,int r,int w){
	if(fa[l]==fa[r]){
		int L=(fa[l]-1)*t+1,R=fa[l]*t;
		for(int i=L;i<=R;i++)
			if(p[i].pos>=l&&p[i].pos<=r)
				p[i].h+=w;
		sort(p+L,p+R+1);
		return;
	}
	for(int i=fa[l]+1;i<=fa[r]-1;i++)set[i]+=w;
	int L=(fa[l]-1)*t+1,R=min(fa[l]*t,n);
	for(int i=L;i<=R;i++)
		if(p[i].pos>=l&&p[i].pos<=r)p[i].h+=w;
	sort(p+L,p+R+1);
	L=(fa[r]-1)*t+1;R=min(fa[r]*t,n);
	for(int i=L;i<=R;i++)
		if(p[i].pos>=l&&p[i].pos<=r)p[i].h+=w;
	sort(p+L,p+R+1);
}
int Binary_Search(int l,int r,int c){
	while(l<=r){	
		int mid=(l+r)>>1;
		if(p[mid].h>=c)l=mid+1;
		else r=mid-1;
	}
	while(p[l].h>=c&&l<=r)l++;
	return l;
}
int ask(int l,int r,int c){
	int ans=0,L,R;
	if(fa[l]==fa[r]){
		L=(fa[l]-1)*t+1,R=min(fa[l]*t,n);
		for(int i=L;i<=R;i++)
			if(p[i].pos>=l&&p[i].pos<=r&&p[i].h+set[fa[l]]>=c)ans++;
		return ans;
	}
	for(int i=fa[l]+1;i<=fa[r]-1;i++){
		L=(i-1)*t+1,R=i*t;
		int pos=Binary_Search(L,R,c-set[i]);
		ans+=pos-L;
	}
	L=(fa[l]-1)*t+1;R=min(fa[l]*t,n);
	for(int i=L;i<=R;i++)
		if(p[i].pos>=l&&p[i].pos<=r&&p[i].h+set[fa[i]]>=c)ans++;
	L=(fa[r]-1)*t+1,R=min(fa[r]*t,n);
	for(int i=L;i<=R;i++)
		if(p[i].pos>=l&&p[i].pos<=r&&p[i].h+set[fa[i]]>=c)ans++;
	return ans;
}
int main(){
	scanf("%d%d",&n,&Q);
	t=(int)sqrt(n);
	for(int i=1;i<=n;i++){scanf("%d",&p[i].h);p[i].pos=i;fa[i]=(i-1)/t+1;}
	for(int i=1;i!=n;i=min(i+t,n))sort(p+i,p+min(i+t,n));
	int l,r,w;
	for(int i=1;i<=Q;i++){
		if(read()){
			scanf("%d%d%d",&l,&r,&w);
			update(l,r,w);
		}
		else{
			scanf("%d%d%d",&l,&r,&w);
			printf("%d\n",ask(l,r,w));
		}
	}
	return 0;
}

1.二分是蒙出来的，那个while为了保险。。p[mid].h写成了p[l].h

2.fa[i]=(i-1)/t+1 居然把+1忘掉了233

3.边界L要加1(上一个块的末尾)，R不变

4.sort：初始化sort：注意就是了

for(int i=1;i!=n;i=min(i+t,n))sort(p+i,p+min(i+t,n));

2016-3-12

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define maxn 1000010
using namespace std;

void read(int& num){
	char ch = getchar();num = 0;
	for(; ch < '!'; ch = getchar());
	for(; ch > '!'; ch = getchar())
	    num = num * 10 + ch - 48;
}

int n, test;

int h[maxn], pos[maxn];

struct Block{
	int a[1010], l, r, cnt, add;
	void init(int Q){
		for(int i = l; i <= r; i ++){
			a[++ cnt] = h[i];
			pos[i] = Q;
		}sort(a + 1, a + 1 + cnt);
	}
	
	void rebuild(){
		cnt = 0;
        for(int i = l; i <= r; i ++)
			a[++ cnt] = h[i];
        sort(a + 1, a + 1 + cnt);
	}
	
	int solve(int p){
		p -= add;
		int ret = lower_bound(a + 1, a + 1 + cnt, p) - a;
		if(ret == cnt + 1)return 0;
		while(ret > 0 && a[ret] >= p)ret --;
		return cnt - ret;
	}
	
	void print(){
		for(int i = 1; i <= cnt; i ++)
		    printf("%d ", a[i]);
		printf("\n");
	}
}b[1010];

int blo;

void build(){
	blo = sqrt(n);
	if(blo * blo != n)blo ++;
	for(int i = 1; i <= blo; i ++){
		b[i].l = (i - 1) * blo + 1;
		b[i].r = min(i * blo, n);
		b[i].init(i);
	}
}

void change(int l, int r, int w){
	if(pos[l] == pos[r]){
		for(int i = l; i <= r; i ++)
			h[i] += w;
		b[pos[l]].rebuild();
		return;
	}
	if(b[pos[l]].l != l){
		for(int i = l; i <= b[pos[l]].r; i ++)
			h[i] += w;
		b[pos[l]].rebuild();
		l = b[pos[l] + 1].l;
	}
	if(b[pos[r]].r != r){
        for(int i = b[pos[r]].l; i <= r; i ++)
			h[i] += w;
		b[pos[r]].rebuild();
		r = b[pos[r] - 1].r;
	}
	l = pos[l], r = pos[r];
	for(int i = l; i <= r; i ++)
	    b[i].add += w;
}

int Solve(int l, int r, int p){
    int ret = 0;
	if(pos[l] == pos[r]){
		for(int i = l; i <= r; i ++)
		    if(h[i] >= p)ret ++;
		return ret;
	}
	if(b[pos[l]].l != l){
        int val = b[pos[l]].add;
		for(int i = l; i <= b[pos[l]].r; i ++)
			ret += (h[i] + val) >= p;
		l = b[pos[l] + 1].l;
	}
	if(b[pos[r]].r != r){
		int val = b[pos[r]].add;
        for(int i = b[pos[r]].l; i <= r; i ++)
			ret += (h[i] + val) >= p;
		b[pos[r]].rebuild();
		r = b[pos[r] - 1].r;
 	}
	
	l = pos[l], r = pos[r];
	for(int i = l; i <= r; i ++)
		ret += b[i].solve(p);
	return ret;
}

int main(){
	int test;
	read(n), read(test);
	for(int i = 1; i <= n; i ++)
		read(h[i]);
	build();
	int u, v, d;
	while(test --){
		char ch = getchar();
		for(; ch < '!'; ch = getchar());
		read(u), read(v), read(d);
		if(ch == 'M')change(u, v, d);
		else printf("%d\n", Solve(u, v, d));
	}
	return 0;
}