其他分享
首页 > 其他分享> > 树状数组

树状数组

作者:互联网

简单论一下区间修改,单点查询的树状数组。

我们是考虑维护一个对于原数组的差分数组,然后取答案时用原数组加上查分数组得到。

资料

二维树状数组

参考资料:资料

单点修改

void modify(int i,int j,int k)
{
	a[i][j]+=k;
	for(int x=i;x<=n;x+=(x&-x))
	{
		for(int y=i;y<=m;y+=(y&-y))
		{
			c[x][y]+=k;
		}
	}
}

区间查询

int query(int i,int j)
{
	int res=0;
	for(int x=i;x;x-=(x&-x))
	{
		for(int y=j;y;y-=(y&-y))
		{
			res+=c[x][y];
		}
	}
	return res;
}

区间修改

如果用了区间修改,区间查询会降级为单点查询
参考资料: 资料

void range_add(int xa,int ya,int xb,int yb,int k)
{
	modify(xa,ya,k);
	modify(xa,yb+1,-k);
	modify(xb+1,ya,-k);
	modify(xb+1,yb+1,k);
}

标签:树状,int,res,xa,modify,ya,数组
来源: https://www.cnblogs.com/cbdsopa/p/15930240.html