区间操作+区间查询(求和)——树状数组解法 & 线段树解法

区间操作+区间查询(求和)

给定一个长度为 $N$ 的数列 $A$，以及 $M$ 条指令，每条指令可能是以下两种之一：

1. C l r d，表示把 $A[l],A[l+1],\dots ,A[r]$ 都加上 $d$。
2. Q l r，表示询问数列中第 $l\sim r$ 个数的和。

对于每个询问，输出一个整数表示答案。

输入格式

第一行两个整数 $N,M$。

第二行 $N$ 个整数 $A[i]$。

接下来 $M$ 行表示 $M$ 条指令，每条指令的格式如题目描述所示。

输出格式

对于每个询问，输出一个整数表示答案。

每个答案占一行。

数据范围

$1\le N,M\le 10^5$,
$|d|\le 10000$,
$|A[i]|\le 10^9$

输入样例：

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

输出样例：

4
55
9
15

1. 树状数组+差分 解法:

思路：构造差分数组

$a[i]$为原数组,

$C[i]$为的差分数组,对$C[i]$求前缀和就是$a[i]$的值

那么要求原数组$a[]$的区$x$点的值,可以计算$C[1]+C[2]+...+C[x-1]+C[x]$

要求$a[]$的前$x$项的和,计算${\sum }_i^x{\sum }_j^{i}C[i]$

相当于

int s = 0;
for(int i = 1;i <= x;++ i)
    for(int j = 1;j <= i;++ j)
        s += C[i];

c1	c2	c3	…	…	$c_{n-1}$	$c_x$
c1	c2	c3	…	…	$c_{n-1}$	$c_x$
c1	c2	c3	…	…	$c_{n-1}$	$c_x$
c1	c2	c3	…	…	$c_{n-1}$	$c_x$
c1	c2	c3	…	…	$c_{n-1}$	$c_x$
c1	c2	c3	…	…	$c_{n-1}$	$c_x$
…	…	…	…	…	$c_{x-1}$	$c_x$
c1	c2	c3	…	…	$c_{x-1}$	$c_x$

以上红色的就是$a[]$的前缀和所加上的差分数组的值(一共$x+1$行)

可以知道整个矩阵的和为 矩阵的和减去黑色的值的和

$(C_1+C_2+...+C_x)\ast (x+1)-(C_1\ast 1+C_2\ast 2+C_3\ast 3+...+C_x\ast x)$

$(C_1+C_2+...+C_x)\ast (x+1)$ ,要快速计算$C[]$的前缀和 ,那么需要一个树状数组来维护$C[i]$

$(C_1\ast 1+C_2\ast 2+C_3\ast 3+...+C_x\ast x)$,要快速计算该算式 ,那么需要一个树状数组来维护$i\ast C[i]$

#include <iostream>
#include <algorithm>
using namespace std;
const int maxn =1e5+5; 
typedef long long ll;
int q[maxn];
ll C[maxn]; //维护 q[i] + ... + q[j]
ll D[maxn]; //维护 i*q[i] +...+j*q[j]
int n,m;
void update(int i,ll v,int c) //
{
    while(i <= n) {
        if(c == 0) C[i] += v;
        else D[i] += v;
        i += i & -i;
    }
}
ll sum(int i,int c)
{
    ll res = 0;
    while(i > 0){
        if(c == 0) res += C[i];
        else res += D[i];
        i -= i & -i;
    }
    return res;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i) scanf("%d",&q[i]);
    for(int i=1;i<=n;++i) 
        update(i,q[i]-q[i-1],0) , update(i,(q[i]-q[i-1])*(ll)i,1);
    char op[2];
    ll l,r,d;
    while(m--)
    {
        scanf("%s",op);
        if(*op == 'C'){
            scanf("%d%d%lld",&l,&r,&d);
            update(l,d,0);
            update(l,d*l,1);
            
            update(r+1,-d,0);
            update(r+1,-d*(r+1),1);
        }
        else 
        {
            scanf("%d%d",&l,&r);
            ll xx = sum(l-1,0)*(l) - sum(l-1,1);
            ll yy = sum(r+1,0)*(r+1) - sum(r+1,1);
            printf("%lld\n",yy-xx);
        }
    }
    return 0;
}

2. 线段树 解法:

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn = 1e5+5;
typedef long long ll;
int a[maxn];
int n,m;
struct Node{
	int l ,r;
	ll sum;
	ll add; //懒标记
}tr[maxn*4];

void pushup(int u)
{
	tr[u].sum = tr[u<<1].sum + tr[u<<1|1].sum;
}

void pushdown(int u)
{
	if(tr[u].add){
		Node& f = tr[u] ,&l = tr[u<<1], &r = tr[u<<1|1];
		l.add += f.add;
		l.sum += (ll)(l.r-l.l+1)*f.add;
		r.add += f.add;
		r.sum += (ll)(r.r-r.l+1)*f.add;
		f.add=0;
	}
}

void build(int u,int l,int r)
{
	if(l == r){
		tr[u] = {l, r, a[r], 0};
		return ;
	}
	tr[u] = {l, r};
	int mid = l + r >> 1;
	build(u << 1, l ,mid), build(u << 1 | 1, mid+1, r);
	pushup(u);
}

void modify(int u, int l, int r, int v)
{
	if(l <= tr[u].l && tr[u].r <= r) {
		tr[u].sum += (ll)(tr[u].r - tr[u].l + 1)*v;
		tr[u].add += v;
		return ;
	}
	int mid = tr[u].l + tr[u].r >> 1;
	pushdown(u);
	if(l <= mid) modify(u<<1,l,r,v);
	if(r > mid) modify(u<<1|1,l,r,v);
	pushup(u);
}

ll query(int u,int l,int r)
{
	if(l <= tr[u].l && tr[u].r <= r) return tr[u].sum;
	int mid = tr[u].l + tr[u].r >> 1;
	pushdown(u);
	if(r <= mid) return query(u<<1,l,r);
	if(l > mid) return query(u<<1|1,l,r);
	ll res = 0;
	if(l <= mid) res = query(u<<1,l,r);
	if(r > mid) res += query(u<<1|1,l,r);
	return res;
}

int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i) scanf("%d",&a[i]);
	build(1,1,n);
	char op[2];
	int l,r,d;
	while(m--)
	{
		scanf("%s%d%d",op,&l,&r);
		if(*op == 'Q')
		{
			printf("%lld\n",query(1,l,r));
		} 
		else 
		{
			scanf("%d",&d);
			modify(1,l,r,d);
		}
	}
	return 0;
}