快速了解二维线段树

学习二维线段树时想必肯定已经会了一维的了。那么二维线段树其实可以简单的理解为固定了一维然后在另一维上跑一维的线段树。

一维线段树上的数组的每个下标代表着一个区间,那么同样的二维线段树则是二维的数组,第一维下标表示第一维的区间,第二维代表第二位区间。

我们用(l,r)表示区间l到r的下标

如果我们要查询(x1,y1)为左下角,(x2,y2)为右上角的矩阵和,即为[x1,x2,y1,y2].

那么假设在横坐标的区间(lx,rx)被(x1,x2)所包含,那么[lx,rx,y1,y2]就是[x1,x2,y1,y2]的一部分。如果(lx,rx)有相应的二维线段树的数组下标。那么我们就可以在O(logn)的时间里在sum[(lx,rx)][]中求出sum[(lx,rx)][(y1,y2)],因为知道了一维,就可以把它看做一维的线段树来求了。

其实求解(lx,rx)也就是在x的那一维求解一维的线段树,这样有是O(logn).

总结就是在O(logn)的时间里面把一维分成不超过logn的固定区间,再在这些固定区间上用O(logn)求出另一维的值,

所以总的更新复杂度是O(log^2n),接下来附上更新和查询的代码以便更好的了解。

区间查询和

int query_y(int l,int r,int rt,int d)
{
	if(L2<=l&&R2>=r) return sum[d][rt];
	int mid = (l+r)>>1,ans = 0;
	if(L2<=mid) ans += query_y(lson,d);
	if(R2>mid) ans += query_y(rson,d);
	return ans;
}
int query_x(int l,int r,int rt)
{
	if(L1<=l&&R1>=r) return query_y(1,m,1,rt);
	int mid = (l+r)>>1,ans = 0;
	if(L1<=mid) ans += query_x(lson);
	if(R1>mid) ans += query_x(rson);
	return ans;
}

单点增加1更新

void update_y(int l,int r,int rt,int d)
{
	sum[d][rt]++;//将所有包含(L1,R1)的矩阵+1即可 
	if(l==r) return;
	int mid = (l+r)>>1;
	if(R1<=mid) update_y(lson,d);
	else update_y(rson,d);
}
void update_x(int l,int r,int rt)
{
	update_y(1,m,1,rt); 
	if(l==r) return ;
	int mid = (l+r)>>1;
	if(L1<=mid) update_x(lson);
	else update_x(rson);
	update_y(1,m,1,rt);
}

 

 

你可能感兴趣的:(线段树)