2019.01.21 bzoj2989: 数列(二进制分组+主席树)

时间:2023-03-09 18:00:02
2019.01.21 bzoj2989: 数列(二进制分组+主席树)

传送门

二进制分组入门题。

主席树写错调题2h+2h+2h+体验极差。

题意简述:给一堆点,支持加入一个点,询问有多少个点跟(x,y)(x,y)(x,y)曼哈顿距离不超过kkk。


思路:题目要求的是对于一个斜着的正方形的查询。

我们考虑转切比雪夫距离转成正常的正方形。

然后就变成了一个动态的二维数点问题。

这个时候已经可以上cdqcdqcdq分治+扫描线或者树套树切题啦。

然而还有一种叫做二进制分组的方法可以支持强制在线的操作。

我们考虑将修改分组,例如对于前19=16+2+119=16+2+119=16+2+1个修改操作可以把它拆成第111 ~ 161616个修改操作,第171717~181818个修改操作,第191919个修改操作这三组,每个组分别维护自己的答案,询问就从各个组分别询问之后把答案累加起来。

然后如果现在又来了第202020个操作,就需要把最后两个合并成一组,变成20=16+2+220=16+2+220=16+2+2,接着发现最后两个组都是222,又需要把最后两个合并成一组,变成20=16+420=16+420=16+4。

即我们模拟二进制数的进位来对修改操作进行组与组之间的合并

可以发现这样的时间复杂度是O(nlogn2)O(nlog_n^2)O(nlogn2​)的,足以通过所有测试点。

最后提一下合并时候的处理方法:我们暴力删除最后一个组,然后重构倒数第二个组

代码:

#include<bits/stdc++.h>
#define ri register int
#define fi first
#define se second
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
const int N=2e5+5,K=1e7+5,lim=2e5,Maxn=100000;
int n,m,a[N],rt[20][N],tp=0;
struct segement_tree{
	#define lc (son[p][0])
	#define rc (son[p][1])
	#define mid (l+r>>1)
	int siz[K],son[K][2],stk[K],top;
	bool vis[K];
	segement_tree(){memset(siz,0,sizeof(siz)),memset(son,0,sizeof(son)),top=0;for(ri i=1;i<K;++i)stk[++top]=i,vis[i]=0;}
	inline int newnode(){return vis[stk[top]]=1,stk[top--];}
	inline void insert(int&p,int o,int l,int r,int k){
		siz[p=newnode()]=siz[o]+1,lc=son[o][0],rc=son[o][1];
		if(l==r)return;
		k<=mid?insert(lc,son[o][0],l,mid,k):insert(rc,son[o][1],mid+1,r,k);
	}
	inline int query(int pl,int pr,int l,int r,int ql,int qr){
		if(ql<=l&&r<=qr)return siz[pr]-siz[pl];
		if(qr<=mid)return query(son[pl][0],son[pr][0],l,mid,ql,qr);
		if(ql>mid)return query(son[pl][1],son[pr][1],mid+1,r,ql,qr);
		return query(son[pl][0],son[pr][0],l,mid,ql,qr)+query(son[pl][1],son[pr][1],mid+1,r,ql,qr);
	}
	inline void delet(int&p){
		if(!vis[p])return;
		stk[++top]=p,vis[p]=0,delet(lc),delet(rc),siz[p]=0,p=0;
	}
}T;
typedef pair<int,int> pii;
vector<pii>v[20];
inline void update(int x,int y){
	v[++tp].push_back(pii(x,y));
	T.insert(rt[tp][1],rt[tp][0],1,lim,y);
	while(tp>1&&v[tp].size()==v[tp-1].size()){
		int p1=0,p2=0;
		vector<pii>tmp;
		for(ri i=1;i<=v[tp].size();++i)T.delet(rt[tp][i]);
		for(ri i=1;i<=v[tp-1].size();++i)T.delet(rt[tp-1][i]);
		while(p1<v[tp-1].size()||p2<v[tp].size()){
			if(p1<v[tp-1].size()&&(p2==v[tp].size()||v[tp-1][p1]<v[tp][p2]))tmp.push_back(v[tp-1][p1++]);
			else tmp.push_back(v[tp][p2++]);
			T.insert(rt[tp-1][p1+p2],rt[tp-1][p1+p2-1],1,lim,tmp[p1+p2-1].se);
		}
		v[tp-1]=tmp,v[tp].clear(),--tp;
	}
}
inline int query(int x,int y,int k){
	int ret=0;
	for(ri p1,p2,i=1;i<=tp;++i){
		p1=lower_bound(v[i].begin(),v[i].end(),pii(x-k,0))-v[i].begin();
		p2=lower_bound(v[i].begin(),v[i].end(),pii(x+k,1e9))-v[i].begin();
		ret+=T.query(rt[i][p1],rt[i][p2],1,lim,max(1,y-k),min(lim,y+k));
	}
	return ret;
}
int main(){
	n=read(),m=read();
	for(ri i=1;i<=n;++i)a[i]=read(),update(a[i]+i,a[i]-i+Maxn);
	for(ri i=1,x,y;i<=m;++i){
		char s[2];
		scanf("%s",s);
		x=read(),y=read();
		if(s[0]=='M')a[x]=y,update(a[x]+x,a[x]-x+Maxn);
		else cout<<query(a[x]+x,a[x]-x+Maxn,y)<<'\n';
	}
	return 0;
}