今天给大家分享一下树状数组。
题目描述:
已知一个数列,你需要进行下面两种操作:
1.将某区间每一个数加上x
2.求出某个点的值
输入格式:
第一行包含两个整数N、M,分别表示该数列数字的个数和操作的总个数。
第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。
接下来M行每行包含2或4个整数,表示一个操作,具体如下:
操作1: 格式:1 x y k 含义:将区间[x,y]内每个数加上k
操作2: 格式:2 x 含义:输出第x个点的值
输出格式:
输出包含若干行整数,即为所有操作2的结果。
输入样例:
5 5 1 5 4 2 3 2 2 1 2 3 2 2 3 1 1 5 1 2 1
输出样例:
5 6 2
样例代码:
#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define RI(n) scanf("%d",&(n))
#define RII(n,m) scanf("%d%d",&n,&m);
#define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define RS(s) scanf("%s",s);
#define ll long long
#define REP(i,N) for(int i=0;i<(N);i++)
#define CLR(A,v) memset(A,v,sizeof A)
#define N 100010
using namespace std;
#pragma GCC optimize (3)
ll n,m,a[N],ans[N];
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
ans[i]=a[i];
}
int x;
for(int i=1;i<=m;i++)
{
scanf("%d",&x);
if(x==1)
{
int s1,s2,o;
scanf("%d%d%d",&s1,&s2,&o);
for(int j=s1;j<=s2;j++)
{
ans[j]+=o;
}
}
if(x==2)
{
int y;
scanf("%d",&y);
printf("%d\n",ans[y]);
}
}
return 0;
}
©著作权归作者所有:来自51CTO博客作者XKLcxywangmy的原创作品,谢绝转载,否则将追究法律责任 树状数组模板2 https://blog.51cto.com/u_15722423/5727484