zkw线段树特点:
非递归,代码简短,结合位运算速度快
结构:
我们可以用一个一维数组c[]来储存数据信息
接下来详细介绍单点更新和区间和求法:
假设需要处理的数组为a[n],则上图叶子数至少为n + 2(其中第一片叶子跟最后一片叶子是不能储存数据的,后面解释),比如说如果n = 2,那只需要把c[]数组开到8;但3 <= n <= 6都要把c[]数组开到16,依次类推。所以叶子数M可以这样求:for(M = 1; M < n + 2; M <<= 1); 则整个c[]就需要2*M个空间。
然后是把a[n](这里设n = 6)里的元素存到c[]里面,准确地说是存到叶子里面,以上图举例,从第二片叶子9开始;依次把a[0] ~ a[n]存入c[9] ~ c[14];(即是a[]下标后移一位 + M)当然还要往上更新,相当于update,后面讲。
假设把所有信息都存进去了,那每个结点的值表示以该结点为根的所有叶子之和。比如结点3的值c[3] = c[12] + c[13] + c[14] + c[15]。即c[3] = a[3] + a[4] + a[5] + 0(空叶子的值定义为0)
对于区间求和,比如我们要求a[1] ~ a[5]的和,那就把下标后移一位,即求[2, 6]的和。那我们可以通过left + M - 1(left在这里表示2,M为叶子数),right + M +1直接定位[2, 6]对应于c[]的开区间(9, 15),这里注意a[]下标不能从0开始,因为定不了左开区间(0 + M - 1 = 7 不是叶子),所以a[]数组下标要后移一位;上面讲的要从第二片叶子开始储存的原因也是这样;右边同理,因此叶子最少要n + 2;也即是说如果n = 1023,那M就要不小于n + 2,就只能是M = 2^11了,基本属于最坏情况。
我们的目标是求c[10] ~ c[14]之和,先放代码:
int Query(int l, int r) //[l, r]
{
int ans = 0;
for(l = l + M - 1, r = r + M + 1; l^r^1; l >>= 1, r >>= 1){
if(~l&1) ans += c[l^1]; //l为偶数
if(r&1) ans += c[r^1]; //r为奇数
}
return ans;
}
即是:
一)ans = 0;
二)如果left为偶数,ans就加上left的右结点的值;如果right为奇数,ans就加上right的左结点的值;
三)left 和 right 分别除以2后,如果left 与 right相邻,就退出循环;否则回到第二步。
这样做可以求区间和的原理手动算算都能明白~
对于更新操作,比如要把a[2]的值修改为v,同样下标右移一位,即把c[3 + M] = c[11] 的值修改为v;接下来往上更新。这步不难想到:
void Update(int p, int v)
{
p += M;
val[p] = v;
while(p > 1){
p >>= 1;
val[p] = val[p<<1] + val[p<<1|1];
}
}
用一道题练手,HDU 1754 :点击打开链接
这是求区间最值的,相同的道理
#include
#include
#include
#include
#define N 200005
#define INF (1<<30)
#define MAX(x, y) ((x) > (y) ? (x) : (y))
int *val;
int M;
void Update(int p, int v)
{
p += M;
val[p] = v;
while(p > 1){
p >>= 1;
val[p] = MAX(val[p<<1], val[p<<1|1]);
}
}
int Query(int l, int r) //[l, r]
{
int ans = -INF;
for(l = l + M - 1, r = r + M + 1; l^r^1; l >>= 1, r >>= 1){
if(~l&1) ans = MAX(val[l^1], ans); //l为偶数
if(r&1) ans = MAX(val[r^1], ans); //r为奇数
}
return ans;
}
int main()
{
//freopen("in.txt", "r", stdin);
int n, m, i, a, p;
char str[3];
while(scanf("%d%d", &n, &m) != EOF)
{
for(M = 1; M < n + 2; M <<= 1);
val = (int *)malloc(sizeof(int) * M * 2);
memset(val, 0, sizeof(val));
for(i = 0; i < n; i++){
scanf("%d", &a);
Update(i + 1, a);
}
scanf("%d", &m);
while(m--)
{
scanf("%s", str);
scanf("%d%d", &p, &a);
if('U' == str[0]) Update(p, a);
else printf("%d\n", Query(p, a));
}
free(val);
}
return 0;
}
用这样的方法做区间更新,暂时没有悟透PPT的思想,弄懂了再上,以下是原文链接:
点击打开链接