[模板]二维前缀和
一维前缀和极易, 从前至尾依次加和即可.
二维如果仍按一维计和方式 i, j 遍历依次加和的话, emmm并没有什么卵用
查询为O( min( abs(y2 - y1), abs(x2 - x1)) * O(1) 妥妥超时
由此引出二维前缀和的标准统计方式
每单位存储的是到本单位左上矩形所有元素(包括本行本列)和
如图
由容斥证明得:
S(d) = S(a+b+c+d) - S(a+c) - S(a+b) + S(a)
则有
SUM(D) = arr[x2, y2] - arr[x2, y1 - 1] - arr[x1 - 1, y2] + arr[x1, y1]
如何建立二维前缀表
由容斥必定有
sum[x, y] = sum[x - 1, y] + sum[x, y - 1] - sum[x - 1, y - 1]
则该条件一定满足于前缀表得任意元素且可以递归定义
因此, i(from 0 to n), j(from 0 to m) sum[i, j] = sum[i - 1, j] + sum[i, j - 1] - sum[i - 1, j - 1]
例 牛客寒假集训6 E 海啸
代码
/*
Zeolim - An AC a day keeps the bug away
*/
const int MAXN = 1e6 + 10;
ll *arr[MAXN];
int main()
{
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
ll n, m, d, tmp;
cin>>n>>m>>d;
for(int i = 0; i <= n; i++)
arr[i] = new ll[m + 1];
for(int i = 0; i <= n; i++)
for(int j = 0; j <= m; j++)
memset(arr[i], 0, sizeof(arr[i]));
for(int i = 1; i <= n; i++)
{
for(int j = 1; j <= m; j++)
{
arr[i][j] = (arr[i - 1][j] + arr[i][j - 1] - arr[i - 1][j - 1]); //建立前缀表
cin>>tmp;
if(tmp >= d)
arr[i][j]++;
}
}
int q;
cin>>q;
while(q--)
{
int x1, y1, x2, y2;
cin>>x1>>y1>>x2>>y2;
ll ans = arr[x2][y2] - arr[x2][y1 - 1] - arr[x1 - 1][y2] + arr[x1 - 1][y1 - 1];//二维计算区域和值
cout<<ans<<"\n";
}
return 0;
}
结束..