技巧-二维前缀和(容斥)

问题描述:给定一个n行m列的矩阵a。

之后有许多查询,每个查询包括i,j,u,v四个数字,其中(i,j)表示一个矩阵的左上角,(u,v)表示一个矩阵的右下角。输出这次查询的矩阵内的和。

方法:先用n*m的时间求出二维前缀和,之后每次查询都是o(1)的时间复杂度。

首先我们定义一个sum[][]的二维数组。sum[i][j]表示左上角为(1,1)右下角为(i,j)的矩阵内所有元素的和。

根据容斥定理,可以递推求出所有sum[i][j]。

for(int i = 1 ; i<=n ; i++){
	for(int j = 1 ; j<=m; j++){
		sum[i][j] = sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j];
	}
} 
之后每次查询,输入i,j,u,v后,我们可以已经求得的sum数组,根据容斥定理得到这个矩阵内所有元素的和为:

sum[u][v] - sum[i-1][v] - sum[u][j-1] + sum[i-1][j-1];
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值