1. 程式人生 > >技巧-二維字首和(容斥)

技巧-二維字首和(容斥)

問題描述:給定一個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];