鏈接：https://ac.nowcoder.com/acm/problem/21862
來源：牛客網

時間限制：C/C++ 1秒，其他語言2秒
空間限制：C/C++ 262144K，其他語言524288K
64bit IO Format: %lld
題目描述
有一個沿海地區，可以看作有n行m列的城市，第i行第j列的城市海拔為h[i][j]。
由于沿海，所以這個地區經常會發生海嘯。
海嘯發生時，部分城市會被淹沒，具體來說，海水高度會達到d，因此海拔低于d的城市都會被淹沒。
現在有q次詢問，每次問你一個矩形區域中，有多少城市不會被淹沒。
輸入描述:
第一行三個整數n,m,d，具體含義見題目描述。
接下來n行，每行m個整數，其中第i行第j列的整數為h[i][j]，具體含義見題目描述。
第n+2行一個整數q，表示詢問數。
接下來q行，每行四個整數a,b,x,y，
表示詢問從第a行第b列到第x行第y列的矩形地區中，有多少地區不會被淹沒。
即有多少個i,j，滿足a≤i≤x,b≤j≤y ，且h[i][j]≥d 。
輸出描述:
共q行，第i行一個整數，表示第i個詢問的答案。
示例1
輸入
復制

3 3 3 1 2 3 2 1 5 4 3 2 2 1 2 2 3 2 1 3 3

輸出
復制

2 3

備注:
1≤n×m≤1e6

1≤q≤1e5

0≤d,h[i][j]≤1e9

1≤a≤x≤n,1≤b≤y≤m

思路：

輸入的時候就可以判斷出會不會淹沒，
所以輸入的時候預處理出一個0,1矩陣，1代表不會淹沒（h[i][j] >= d）
詢問次數很多，每次我們都要輸出要求的那個子矩陣的和，
很明顯還需要預處理一下，我們需要維護一個二維前綴和：
s[i][j]為(0,0)到(i,j)的前綴和
假設現在我們要求如下圖中的子矩陣A的和：
那么：
A = D - B - C + (B∩C)

即 ans = s [x][y]- s[a-1][y] - s[x][b-1] + s[a-1][b-1]
預處理前綴和數組s，是ans的逆過程
s[i][j] = (val>=d)+s[i-1][j]+s[i][j-1]-s[i-1][j-1]
AC_code:

#include <bits/stdc++.h>using namespace std;int main() {int n,m,d,val;scanf("%d%d%d",&n,&m,&d);int dp[n+2][m+2];for(int i = 0; i <= n; i++){for(int j = 0; j <= m; j++){dp[i][j] = 0;}}for(int i = 1; i <= n; i++){for(int j = 1; j <= m; j++){scanf("%d",&val);dp[i][j]= (val>=d)+dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1];}}int t,a,b,x,y;scanf("%d",&t);while(t--){scanf("%d%d%d%d",&a,&b,&x,&y);int ans = dp[x][y]-dp[x][b-1]-dp[a-1][y]+dp[a-1][b-1];printf("%d\n",ans);}return 0; }

way2用二維數狀數組：
數狀數組在求前綴和方面比較強大，
本題也可以用二維樹狀數組來維護。
AC_code:

#include <bits/stdc++.h> using namespace std; const int MAXN = 1e6+2; vector<int> a[MAXN]; int n,m,d; void init() {for(int i = 1; i<= n; i++){a[i].push_back(0);for(int j = 1; j <= m; j++){a[i].push_back(0);}} } int lowbit(int x) {return x&-x; } void add(int x,int y,int v) {for(int i = x; i <= n; i+=lowbit(i)){for(int j = y; j <= m; j+=lowbit(j)){a[i][j] += v;}} } int sum(int x,int y) {int s = 0;for(int i = x; i > 0; i-=lowbit(i)){for(int j = y; j > 0; j-=lowbit(j)){s += a[i][j];}}return s; }int main() {std::ios::sync_with_stdio(false);cin.tie(0);cin>>n>>m>>d;init();int val;for(int i = 1; i <= n; i++){for(int j = 1; j <= m; j++){cin>>val;if(val >= d) add(i,j,1);}}int t,a,b,x,y;cin>>t;while(t--){cin>>a>>b>>x>>y;int ans = sum(x,y)-sum(x,b-1)-sum(a-1,y)+sum(a-1,b-1);cout<<ans<<endl;}return 0; }

總結

以上是生活随笔為你收集整理的海啸（二维前缀和/二维树状数组）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。