[模板]-树状数组(一维)

一维树状数组

1.1 单点修改 + 区间和

【任务】修改数组中某个元素，求 $[l,r]$ 区间和。
【说明】原始数组 $a[i]$, 树状数组 $c[i]$ 存储 $a[i]$.
【接口】
           \;\;\;\;\; void update(int i, int x);
           \;\;\;\;\; 复杂度：$O(logn)$
           \;\;\;\;\; 功能：单点修改 $a[i]+x$.
           \;\;\;\;\; 调用：update(i, x).

           \;\;\;\;\; int Query_sum(int i);
           \;\;\;\;\; 复杂度：$O(logn)$
           \;\;\;\;\; 功能：返回前缀和 $sum[i]={\sum }_{k=1}^{i}a[k]$.
           \;\;\;\;\; 调用：sum[ l, r ] = Query_sum( r ) - Query_sum( l - 1 ).
【代码】

void update(int i, int x) {  
    for(int k = i; k <= n; k += lowbit(k)) {      // 单点修改 a[i] + x  
        c[k] += x;  
    }     
}  
int Query_sum(int i) {  
    int ans = 0;  
    for(int k = i; k >= 1; k -= lowbit(k)) {      // 前缀和 sum[i]  
        ans += c[k];  
    }  
    return ans;  
}  

1.2 区间修改 + 单点查询

【任务】将区间 $[l,r]$ 中每个数 $+x$ , 查询数组中某个元素的值。
【说明】原始数组 $a[i]$, 树状数组 $c[i]$ 存储 $d[i]$, $d[i]=a[i]-a[i-1]$.
           \;\;\;\;\; 区间修改可以转化为单点修改：对区间 $[l,r]$ 中每个数 $+x$ 之后，对于 $d[i]$ 数组，区间 $[l+1,r]$ 保持不变，只有两个地方需要改 $d[l]+x,d[r+1]-x$.
           \;\;\;\;\; 单点查询可以转化为求前缀和：$a[i]=(d[i]-d[i-1])+(d[i-1]-d[i-2])+\cdot \cdot \cdot +(d[2]-d[1])$.
【接口】
           \;\;\;\;\; void update(int k, int x);
           \;\;\;\;\; 复杂度：$O(logn)$
           \;\;\;\;\; 功能：单点修改 $d[k]+x$.
           \;\;\;\;\; 调用：update(l, x), update(r+1, -x).

           \;\;\;\;\; ll Query_sum(int i);
           \;\;\;\;\; 复杂度：$O(logn)$
           \;\;\;\;\; 功能：返回前缀和 $sum[i]={\sum }_{k=1}^{i}d[k]$.
           \;\;\;\;\; 调用：a[i] = Query_sum(i).
【代码】

void update(int k, int x) {
	for(int i = k; i <= n; i += lowbit(i)) {
		c[i] += x;
	}
	return;
}
ll Query_sum(int k) {
	ll sum = 0;
	for(int i = k; i >= 1; i -= lowbit(i)) {
		sum += c[i];
	}
	return sum;
}

1.3 区间修改 + 区间和

【任务】将区间 $\left[l_1,r_1\right]$ 中每个数 $+x$ , 查询区间和 $\left[l_2,r_2\right]$ 。
【说明】原始数组 $a[i]$, 树状数组 $sum1[i]$ 存储 $d[i]$, $d[i]=a[i]-a[i-1]$.
           \;\;\;\;\; 区间修改：可以转化为单点修改：对区间 $[l,r]$ 中每个数 $+x$ 之后，对于 $d[i]$ 数组，区间 $[l+1,r]$ 保持不变，只有两个地方需要改 $d[l]+x,d[r+1]-x$.
           \;\;\;\;\; 区间求和： 前缀和 ${\sum }_{i=1}^{r}a[i]=a_1+a_2+...+a_r$

           \;\;\;\;\; $=d_1+(d_1+d_2)+(d_1+d_2+d_3)+\cdot \cdot \cdot +(d_1+d_2+...+d_r)$

           \;\;\;\;\; $=r\cdot d_1+(r-1)\cdot d_2+(r-2)\cdot d_3+\cdot \cdot \cdot +2\cdot d_{r-1}+d_r$

           \;\;\;\;\; $={\sum }_{k=1}^r(r+1-k)\cdot d_k$

           \;\;\;\;\; $={\sum }_{k=1}^r(r+1)\cdot d_k-{\sum }_{k=1}^{r}k\cdot d_k$

           \;\;\;\;\; 需要第二个树状数组 $sum2[i]$ 存储 $i\cdot d[i]$.

【接口】
           \;\;\;\;\; void update(int i, int x);
           \;\;\;\;\; 复杂度：$O(logn)$
           \;\;\;\;\; 功能：单点修改 $d[i]+x,$.
           \;\;\;\;\; 调用：update(l, x), update(r+1, -x).

           \;\;\;\;\; ll Query_sum(int i);
           \;\;\;\;\; 复杂度：$O(logn)$
           \;\;\;\;\; 功能：返回前缀和 ${\sum }_{k=1}^{i}a[k]$.
           \;\;\;\;\; 调用：Query_sum( r ) - Query_sum( l - 1 ).
【代码】

void update(int k, ll x) {
	for(int i = k; i <= n; i += lowbit(i)) {
		sum1[i] += x;
		sum2[i] += x * k;
	}
	return;
}

ll Query_sum(int k) {
	ll sum = 0;
	for(int i = k; i >= 1; i -= lowbit(i)) {
		sum += sum1[i] * (k + 1) - sum2[i];
	}
	return sum;
}