树状数组实现区间修改+区间查询

事实上,这只是我弱弱的luogu博客的存档……

线段树模板(1)

题意要求:给定一个序列,支持区间修改和区间查询。

智障数据结构模板题……

当然,题目名字告诉我们要用线段树。但是线段树很长,容易出现问题,而且跑得稍慢,所以就有dalao开始yy:可不可以让树状数组支持区间修改和查询呢?

于是伟大的“超级树状数组”横空出世了。

首先,我们看树状数组是如何支持区间修改的:

设 tree[i]=a[i]-a[i-1] (差分),那么容易得到:

tree[1]+tree[2]+……+tree[i]=a[i] 这个公式

所以,只需要维护tree数组就可以实现区间修改了。

那么问题来了,如果这样,那么如何实现区间查询呢?

我们已经推出了一个公式:

tree[1]+tree[2]+……tree[i]=a[i]

那么,对于1到r的区间和,即为:

1  a[1]+a[2]+……+a[r-1]+a[r]
2 //用上方公式推导得出
3 =tree[1]+(tree[1]+tree[2])+……+(tree[1]+……+tree[r])
4 //根据加法交换律与结合律:
5 =(tree[1]*(r))+(tree[2]*(r-1))+……(tree[r]*1)
6 //那么:
7 =r*(tree[1]+tree[2]+……+tree[r])-(tree[1]*0+tree[2]*1+……+tree[r]*(r-1))

看到这里,是不是已经很清晰了呢?

对于a的树状数组(差分)tree,建立一个新的树状数组tree1使得:

tree1[i]=tree[i]*(i-1)

之后,x到y的区间和即为:

(y*getsum(tree,y)-(x-1)*getsum(tree,x-1))-(getsum(tree1,y)-getsum(tree1,x-1))

当然,对于更新操作也需要进行一些细微调整,详细的就看代码吧……

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 long long n,m,tree[100005],tree1[100005];//题目要求longlong
 4 inline void add(long long*z,long long x,long long num)
 5 {
 6     while(x<=n)
 7     {
 8         z[x]+=num;
 9         x+=x&(-x);
10     }
11 }
12 inline long long getsum(long long*z,long long x)
13 {
14     long long sum=0;
15     while(x>0)
16     {
17         sum+=z[x];
18         x-=x&(-x);
19     }
20     return sum;
21 }
22 int main()
23 {
24     cin.sync_with_stdio(false);
25     cin>>n>>m;
26     long long a,b=0;
27     for(long long i=1;i<=n;i++)
28     {
29         cin>>a;
30         b=a-b;
31         add(tree,i,b);
32         add(tree1,i,(i-1)*b);
33         b=a;
34     }
35     for(long long i=1;i<=m;i++)
36     {
37         int t,x,y,z;
38         cin>>t;
39         if (t==1)
40         {
41             cin>>x>>y>>z;
42             add(tree,x,z);
43             add(tree,y+1,-z);
44             add(tree1,x,z*(x-1));
45             add(tree1,y+1,-z*y);//此处为核心,联系上方的公式,想一想为什么这么修改。
46         }
47         else
48         {
49             cin>>x>>y;
50             cout<<(y*getsum(tree,y)-(x-1)*getsum(tree,x-1))-(getsum(tree1,y)-getsum(tree1,x-1))<<endl;
51         }
52     }
53     return 0;
54 } 

原文地址:https://www.cnblogs.com/mengmengdezhongyezijiang/p/8148841.html

时间: 2024-10-20 16:07:44

树状数组实现区间修改+区间查询的相关文章

二维树状数组模板(区间修改+区间查询)

二维树状数组模板(区间修改+区间查询) 例题:JOIOI上帝造题的七分钟 一共两种操作: \(L\ x_1\ y_1\ x_2\ y_2\ d\):把\((x_1,y_1)\),\((x_2,y_2)\)这个矩形内所有元素加\(d\). \(k\ x_1\ y_1\ x_2\ y_2\):查询\((x_1,y_1)\),\((x_2,y_2)\)这个矩形内所有元素的和. 代码如下: #include<bits/stdc++.h> #define RG register #define IL i

【LuoguP3038/[USACO11DEC]牧草种植Grass Planting】树链剖分+树状数组【树状数组的区间修改与区间查询】

模拟题,可以用树链剖分+线段树维护. 但是学了一个厉害的..树状数组的区间修改与区间查询.. 分割线里面的是转载的: -------------------------------------------------------------------------------- [ 3 ]  上面都不是重点--重点是树状数组的区间修改+区间查询 这个很好玩 其实也挺简单 首先依旧是引入delta数组 delta[i]表示区间 [i, n] 的共同增量 于是修改区间 [l, r] 时修改 delt

Luogu P3368 【模板】树状数组 2 [区间修改-单点查询]

P3368 [模板]树状数组 2 题目描述 如题,已知一个数列,你需要进行下面两种操作: 1.将某区间每一个数数加上x 2.求出某一个数的和 输入输出格式 输入格式: 第一行包含两个整数N.M,分别表示该数列数字的个数和操作的总个数. 第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值. 接下来M行每行包含2或4个整数,表示一个操作,具体如下: 操作1: 格式:1 x y k 含义:将区间[x,y]内每个数加上k 操作2: 格式:2 x 含义:输出第x个数的值 输出格式: 输出

树状数组的区间修改与单点查询与区间查询

如何将普通树状数组升级 普通的单点修改单点查询就不讲了,从区间修改和单点查询讲起. 原来的值存在a[]里面,多建立个数组c1[],注意:c1[i]=a[i]-a[i-1]. 那么求a[i]的值的时候a[i]=a[i-1]+c1[i]=a[i-2]+c1[i]+c1[i-1]=-..=c1[1]+c1[2]+-+c1[i]. 所以就用c1[]建立树状数组,便可以很快查询a[i]的值.不多说,见代码. #include<iostream> #include<cstdio> #defin

Luogu P3374 【模板】树状数组 1[单点修改-区间查询]

P3374 [模板]树状数组 1 题目描述 如题,已知一个数列,你需要进行下面两种操作: 1.将某一个数加上x 2.求出某区间每一个数的和 输入输出格式 输入格式: 第一行包含两个整数N.M,分别表示该数列数字的个数和操作的总个数. 第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值. 接下来M行每行包含3或4个整数,表示一个操作,具体如下: 操作1: 格式:1 x k 含义:将第x个数加上k 操作2: 格式:2 x y 含义:输出区间[x,y]内每个数的和 输出格式: 输出包

关于树状数组的区间修改和单点查询

写在前面 之前一直不知道树状数组可以支持区间修改,所以写一篇博客记录一下. 首先给个小栗子: 如下图: 利用差分的思路,就得到下图: 那么如果我们要求将2~4的所有元素+2呢?我们就可以得到下图: 可以发现,差分的第二项和第五项一个加了2,一个减了2,所以对于每次区间[l,r]操作,我们只需要在l和r+1的位置加.减操作值即可,证明也很简单,首先操作区间内的数的差肯定不会变,所以区间内的一段相同,因为第一项增加了一个值k,所以他与前面一项的差就增加了k,最后一项增加了k,最后一项的后一项与最后一

Color the ball 树状数组的区间修改

Color the ball Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others) Total Submission(s): 9712    Accepted Submission(s): 4998 Problem Description N个气球排成一排,从左到右依次编号为1,2,3....N.每次给定2个整数a b(a <= b),lele便为骑上他的"小飞鸽"牌

树状数组(区间修改,区间查询 )

1 #include<iostream> 2 #include<cstdio> 3 using namespace std; 4 5 long long n,m; 6 long long a[500009]; 7 long long c[500009],d[500009]; 8 long long lowbit(long long x) 9 { 10 return (x&(-x)); 11 } 12 void change(long long *e,long long x,

A Simple Problem with Integers poj 3468 多树状数组解决区间修改问题。

A Simple Problem with Integers Time Limit: 5000MS   Memory Limit: 131072K Total Submissions: 69589   Accepted: 21437 Case Time Limit: 2000MS Description You have N integers, A1, A2, ... , AN. You need to deal with two kinds of operations. One type of