【BZOJ】2989: 数列-二进制分组&主席树

传送门:bzoj2989


题解

二进制分组-CA

欧几里得距离转成曼哈顿距离 ( x , y ) , ( x + y , x − y ) (x,y),(x+y,x-y) (x,y),(x+y,xy)之后问题就变成了每次给一个点+1,询问某个矩形范围内的总值。

通常方法是 c d q cdq cdq,二进制分组可以解决强制在线的限制。

具体来说,设现有总操作数为 n n n n n n转成2进制后有 n u m num num位,则将操作分成按 n n n的二进制来划分:对于任意 0 ≤ i &lt; n u m 0\leq i&lt;num 0i<num n &amp; 2 i = 2 i n\&amp;{2^i}=2^i n&2i=2i,则划分出一个大小为 2 i 2^i 2i的组(实际操作很简单,没有我说的复杂。。。)。

对于每个询问就在这 log ⁡ n \log n logn个组中依次查询。主席树替代了二维线段树的功能, r t rt rt这一维维护递增的 x x x坐标,树内维护递增的值域范围内的 y y y坐标。所以每次二分找出 x x x坐标对应的树范围,查询区间 [ y − k , y + k ] [y-k,y+k] [yk,y+k]的总值即可。

对于每个新增的操作,强行划分出一个大小为1的组后不断向前合并即可。

复杂度 O ( n log ⁡ 2 n ) O(n\log^2n) O(nlog2n)。需要建内存池动态维护线段树的点标号。


代码

#include<bits/stdc++.h>
#define mid ((l+r)>>1)
#define lc ls[k],l,mid
#define rc rs[k],mid+1,r
#define pb push_back
using namespace std;
const int N=6e4+10,M=1e5+10,MX=16e4,NX=1e7+5;

int n,m,cnt,a[N],rt[20][M],num,top;
int ss[NX],ls[NX],rs[NX];
char op[10];bool vs[NX];

struct Pl{
	int que[NX],top;
	inline int pop(){int re;if(!top) re=++cnt;else re=que[top--];vs[re]=true;return re;}
	inline void psh(int x){que[++top]=x;vs[x]=false;ls[x]=rs[x]=ss[x]=0;}
}pl;

struct Pr{
    int x,y;
    Pr(int x_=0,int y_=0):x(x_),y(y_){};
    bool operator <(const Pr&ky) const{return x==ky.x?y<ky.y:x<ky.x;}
    bool operator ==(const Pr&ky) const{return (x==ky.x && y==ky.y);}
}rep[M];
vector<Pr>v[20];

void ins(int pre,int &k,int l,int r,int pos)
{
	k=pl.pop();ss[k]=ss[pre]+1;ls[k]=ls[pre],rs[k]=rs[pre];
	if(l^r){if(pos<=mid) ins(ls[pre],lc,pos);else ins(rs[pre],rc,pos);}
}

void dl(int k,int l,int r)
{if((!vs[k])) return;if(l^r) dl(lc),dl(rc);pl.psh(k);}

inline void ins(int x,int y)
{
	int i,j,len;
	v[++num].pb(Pr(x,y));
	ins(rt[num][0],rt[num][1],1,MX,y);
	for(;num>1 && v[num].size()==v[num-1].size();){
		len=v[num].size();top=0;
		for(i=1;i<=len;++i) dl(rt[num][i],1,MX),dl(rt[num-1][i],1,MX);
		i=j=0;
		for(;i<len || j<len;){
			if((j>=len) || (i<len && v[num-1][i]<v[num][j])){
				ins(rt[num-1][i+j],rt[num-1][i+j+1],1,MX,v[num-1][i].y);
				rep[top++]=v[num-1][i++];
			}else{
				ins(rt[num-1][i+j],rt[num-1][i+j+1],1,MX,v[num][j].y);
				rep[top++]=v[num][j++];
			}
		}
		v[num--].clear();v[num].clear();
		for(i=0;i<top;++i) v[num].pb(rep[i]);
	}
}

int ask(int pre,int k,int l,int r,int L,int R)
{
	if(ss[k]==ss[pre]) return 0;
	if(L<=l && r<=R) return ss[k]-ss[pre];
	if(R<=mid) return ask(ls[pre],lc,L,R);
	if(L>mid) return ask(rs[pre],rc,L,R);
	return ask(ls[pre],lc,L,R)+ask(rs[pre],rc,L,R);
}

inline int query(int x,int y,int vv)
{
	int i,re=0,a,b,l=max(1,y-vv),r=min(MX,y+vv);
	for(i=1;i<=num;++i){
		a=lower_bound(v[i].begin(),v[i].end(),Pr(x-vv,0))-v[i].begin();
		b=lower_bound(v[i].begin(),v[i].end(),Pr(x+vv,2e9))-v[i].begin();
		re+=ask(rt[i][a],rt[i][b],1,MX,l,r);
	}
	printf("%d\n",re);
}

int main(){
	int i,x,y;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;++i) {scanf("%d",&a[i]);ins(a[i]+i,a[i]-i+N);}
	for(;m;--m){
		scanf("%s%d%d",op,&x,&y);
		if(op[0]=='Q') query(a[x]+x,a[x]-x+N,y);
		else a[x]=y,ins(y+x,y-x+N);
	}
	return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值