害怕 第二篇博客竟然是线段树2333
来一道简单的裸·线段树模板:
codevs 1080 线段树练习
题目描述 Description:
一行N个方格,开始每个格子里都有一个整数。现在动态地提出一些问题和修改:提问的形式是求某一个特定的子区间[a,b]中所有元素的和;修改的规则是指定某一个格子x,加上或者减去一个特定的值A。现在要求你能对每个提问作出正确的回答。1小于等于N大于100000,,提问和修改的总数m小于10000条。
输入描述 Input Description:
输入文件第一行为一个整数N,接下来是n行n个整数,表示格子中原来的整数。接下一个正整数m,再接下来有m行,表示m个询问,第一个整数表示询问代号,询问代号1表示增加,后面的两个数x和A表示给位置X上的数值增加A,询问代号2表示区间求和,后面两个整数表示a和b,表示要求a到b之间的区间和。
输出描述 Output Description:
共m行,每行一个整数
样例输入 Sample Input:
6
4
5
6
2
1
3
4
1 3 5
2 1 4
1 1 9
2 2 6
样例输出 Sample Output:
22
22
数据范围及提示 Data Size & Hint:
1小于等于N小于等于100000, m小于等于10000 。
一句话:单点修改,区间查询
QAQ废话不多说 很简单的线段树模板 直接上代码吧~
#include<iostream>
#include<cstdio>
#define ll long long
using namespace std;
const int maxn=2e5+5;
struct hh
{
int l,r;
ll summ,add;
}tree[maxn<<2];
int a[maxn];
void Updata(int n,int x)
{
tree[n].summ+=x*(tree[n].r-tree[n].l+1);
}
void Spread(int n)
{
ll &ax=tree[n].add;
if(ax)
{
tree[n<<1].add+=ax;
tree[n<<1|1].add+=ax;
Updata(n<<1,ax);
Updata(n<<1|1,ax);
ax=0;
}
}
ll Ask(int l,int r,int n)
{
int lx=tree[n].l;
int rx=tree[n].r;
if(l<=lx&&r>=rx)
return tree[n].summ;
int mid=(lx+rx)>>1;
Spread(n);//查询时不要忘记下放标记
ll ans=0;
if(l<=mid) ans+=Ask(l,r,n<<1);
if(r>mid) ans+=Ask(l,r,n<<1|1);
return ans;
}
void Add(int l,int r,int n,ll x)
{
if(tree[n].l>=l&&tree[n].r<=r)
{
tree[n].add+=x;
Updata(n,x);
return;
}
Spread(n);
int mid=(tree[n].l+tree[n].r)>>1;
if(l<=mid) Add(l,r,n<<1,x);
if(r>mid) Add(l,r,n<<1|1,x);
tree[n].summ=tree[n<<1].summ+tree[n<<1|1].summ;
}
ll Build(int l,int r,int n)
{
tree[n].l=l;
tree[n].r=r;
if(l==r)
{
tree[n].summ=a[l];
return a[l];
}
int mid=(l+r)>>1;
return tree[n].summ=Build(l,mid,n<<1)+Build(mid+1,r,n<<1|1);
}
int main()
{
int n,m;
scanf("%d",&n);
for(int i=1;i<=n;++i) scanf("%d",a+i);
Build(1,n,1);
scanf("%d",&m);
while(m--)
{
int x,y,z,w;
scanf("%d%d%d",&x,&y,&z);
if(x==1)
{
scanf("%d",&w);
Add(y,z,1,w);
}
else cout<<Ask(y,z,1)<<"\n";
}
return 0;
}