二維前綴和詳解


我們講一下什么是二維前綴和,建立在一維前綴和之上,我們要求一個矩陣內一個任意的子矩陣的數的和,我們就可以用二維前綴和,我們還是用DP來預處理,狀態和一維前綴和差不多,只不過我們多加了一維,DP[i][j]表示(1,1)這個點與(i,j)這個點兩個點分別為左上角和右下角所組成的矩陣內的數的和,好好想一下狀態轉移方程,DP[i][j]=DP[i-1][j]+DP[i][j-1]-DP[i-1][j-1]+map[i][j],怎么來的呢?我們畫一下圖就知道了。

404

這張圖就知道了(i,j)可以由(i-1,j)(i,j-1)兩塊構成,不過要注意兩個點

  1. 有一塊矩陣我們重復加了,也就是(i-1,j-1)這一塊,所以我們要減去它

  2. 我們這個矩陣是不完整的,由圖可知我們還有一塊深藍色的沒有加,也就是(i,j)這一點,所以我們要再加上map[i][j]也就是題目給出的矩陣中這一格的數。

這樣我們就預處理完了,現在講一下怎么通過我們的預處理從而快速地得出我們想要的任意子矩陣中的和,我們定義(x1,y1)為我們想要子矩陣的左上角,(x2,y2)為我們想要子矩陣的右下角,然后我們畫圖想一想。

404

我們可以通過DP[x2][y2]來計算,我們通過圖可以發現這個距離我們要的還差紅色的部分看看怎么表示紅色部分?我們可以分割成兩塊,分別是DP[x1][y2]DP[x2][y1]我們發現有一塊重復減了,所以我們再加上它即DP[x1][y1]

有一點注意,因為畫圖和定義原因我們發現邊界好像不對,我們來看看,我們定義的狀態是整個矩陣包括邊的和,而我們要求的也是要包括邊的,所以我們要再改一下,把DP[x1][y2]DP[x2][y1]DP[x1][y1]分別改成DP[x1-1][y2]DP[x2][y1-1]DP[x1-1][y1-1]

這樣一減我們就可以得到自己想要的答案,整理可得公式,DP[x2][y2]-DP[x1-1][y2]-DP[x2][y1-1]+DP[x1-1][y1-1]這樣我們就可以做到O(1)之內查詢,很奇妙吧,我們看一下實現代碼:

#include<iostream>
#include<cstring>
using namespace std;
int dp[2000][2000],map[2000][2000];
int main()
{
	int m,n,k;//所給的矩陣是n*m的,有k組查詢 
	cin >>n>>m>>k;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			cin >>map[i][j];
	memset(dp,0,sizeof(dp));
	for(int i=1;i<=n;i++)//預處理一波 
		for(int j=1;j<=m;j++)
			dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+map[i][j];
	for(int i=1;i<=k;i++)//接受查詢 
	{
		int x1,x2,y1,y2;
		cin >>x1>>y1>>x2>>y2;
		cout <<(dp[x2][y2]+dp[x1-1][y1-1]-dp[x1-1][y2]-dp[x2][y1-1])<<endl;//O(1)查詢 
	}
	return 0;
}

參考博文:https://blog.csdn.net/qq_34990731/article/details/82807870


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM