常用代码模板1 ----- 基础算法

一维前缀和

求一维前缀和数组:
从下标1开始记录
for(i = 1 ... n)
	s[i] = s[i - 1] + a[i]
从下标0开始记录
s[0] = a[0]
for(i = 1 ... n)
	s[i] = s[i - 1] + a[i]

根据前缀和数组求a数组[l, r]区间和:
sum = s[r] - s[l - 1]

二维前缀和

求二维前缀和数组:
行列均从下标1开始记录
for(i = 1 ... n)
	for(j = 1 ... m)
  		S[i][j] = S[i - 1][j] + S[i][j - 1] - S[i - 1][j - 1] + a[i][j]

根据二维前缀和数组求主对角线端点为(x1, y1) (x2, y2)的子矩阵和
sum = S[x2][y2] - S[x2][y1 - 1] - S[x1 - 1][y2] + S[x1 - 1][y1 - 1]
©️2020 CSDN 皮肤主题: 游动-白 设计师:上身试试 返回首页