A Simple Problem with Integers
X M
输入一个长度为X的数组,然后接下来有M个测试案例
当为Q A B 的时候就是查询从A-B这个区间的sum值
当为CA B C 的时候就是从A-B这个区间所有的节点都加上C,更新整个数组
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
</pre><pre class="sio" name="code" style="white-space: pre-wrap; word-wrap: break-word; color: rgb(51, 51, 51); font-size: 14px; line-height: 26px; background-color: rgb(255, 255, 255); font-family: 'Courier New', Courier, monospace;">
#include<stdio.h> #include<string.h> const int N=100000; long long add[N<<2]; long long sum[N<<2]; int a[N]; void build(int root,int l,int r){ <span style="white-space: pre;"> </span>add[root]=0; <span style="white-space: pre;"> </span>if(l==r){ <span style="white-space: pre;"> </span>sum[root]=a[l]; <span style="white-space: pre;"> </span>return; <span style="white-space: pre;"> </span>} <span style="white-space: pre;"> </span>int mid=(l+r)>>1; <span style="white-space: pre;"> </span>build(root<<1,l,mid); <span style="white-space: pre;"> </span>build(root<<1|1,mid+1,r); <span style="white-space: pre;"> </span>sum[root]=sum[root<<1]+sum[root<<1|1]; } void push(int root,int m) { <span style="white-space: pre;"> </span>if(add[root]) <span style="white-space: pre;"> </span>{ <span style="white-space: pre;"> </span>add[root<<1]+=add[root]; <span style="white-space: pre;"> </span>add[root<<1|1]+=add[root]; <span style="white-space: pre;"> </span>sum[root<<1]+=(m-(m>>1))*add[root]; <span style="white-space: pre;"> </span>sum[root<<1|1]+=(m>>1)*add[root]; <span style="white-space: pre;"> </span>add[root]=0; <span style="white-space: pre;"> </span>} } void update(int root,int L,int R,int l,int r,int c) { <span style="white-space: pre;"> </span>if(l<=L&&r>=R) <span style="white-space: pre;"> </span>{ <span style="white-space: pre;"> </span> add[root]+=c; <span style="white-space: pre;"> </span>sum[root]+=(long long)c*(R-L+1); <span style="white-space: pre;"> </span>return; <span style="white-space: pre;"> </span>} <span style="white-space: pre;"> </span>push(root,R-L+1); <span style="white-space: pre;"> </span>int mid=(R+L)>>1; <span style="white-space: pre;"> </span>if(l<=mid) <span style="white-space: pre;"> </span> update(root<<1,L,mid,l,r,c); <span style="white-space: pre;"> </span>if(r>mid) <span style="white-space: pre;"> </span> update(root<<1|1,mid+1,R,l,r,c); <span style="white-space: pre;"> </span>sum[root]=sum[root<<1]+sum[root<<1|1]; } long long querysum(int root,int L,int R,int l,int r) { <span style="white-space: pre;"> </span>if(l<=L&&r>=R){ <span style="white-space: pre;"> </span>return sum[root]; <span style="white-space: pre;"> </span>} <span style="white-space: pre;"> </span>push(root,R-L+1); <span style="white-space: pre;"> </span>int mid=(R+L)>>1; <span style="white-space: pre;"> </span>long long sum1=0; <span style="white-space: pre;"> </span>if(l<=mid) <span style="white-space: pre;"> </span> sum1+=querysum(root<<1,L,mid,l,r); <span style="white-space: pre;"> </span>if(r>mid) <span style="white-space: pre;"> </span> sum1+=querysum(root<<1|1,mid+1,R,l,r); <span style="white-space: pre;"> </span> sum[root]=sum[root<<1]+sum[root<<1|1]; return sum1; } int main() { <span style="white-space: pre;"> </span>int m,x,h,l,r; <span style="white-space: pre;"> </span>scanf("%d%d",&x,&m); <span style="white-space: pre;"> </span>for(int i=1;i<=x;i++) <span style="white-space: pre;"> </span> scanf("%d",&a[i]); <span style="white-space: pre;"> </span>char b[2]; <span style="white-space: pre;"> </span> build(1,1,x); while(m--) { <span style="white-space: pre;"> </span> scanf("%s",b); <span style="white-space: pre;"> </span> if(b[0]=='Q') <span style="white-space: pre;"> </span> { <span style="white-space: pre;"> </span> <span style="white-space: pre;"> </span>scanf("%d%d",&l,&r); <span style="white-space: pre;"> </span> <span style="white-space: pre;"> </span>printf("%lld\n",querysum(1,1,x,l,r)); <span style="white-space: pre;"> </span> } <span style="white-space: pre;"> </span> else <span style="white-space: pre;"> </span> { <span style="white-space: pre;"> </span> <span style="white-space: pre;"> </span>scanf("%d%d%d",&l,&r,&h); <span style="white-space: pre;"> </span> <span style="white-space: pre;"> </span>update(1,1,x,l,r,h); <span style="white-space: pre;"> </span> } } <span style="white-space: pre;"> </span>return 0; }
这是一道使用线段树动态维护区间和的题目
当字母输入为Q的时候 然后给定一个区间,求出这个区间的和
当字母输入为C的时候,给定一个区间和某个值 这个区间上的每个点都要加上这个值
刚开始的建树和去求某个区间最大值或者最小的时候差不多,无非就是把比较最大值最小值的语句换成去保留两个节点的和
我觉得这道题最难得部分在于去动态维护区间的和
当去要求更新某个区间的和的时候,我选择的是我使用到哪个区间就去更新哪个区间及它下一级的子树,再往下的时候就不更新,什么时候用到了就更新,用一个数组来保留节点所更新的那个数值是多少,当某个节点的左右孩子已经更新过,就把该节点的add值置成0,然后把他孩子节点的add的值加上。
值得注意的在查询的函数里,也要把更新节点的语句加上,因为有可能需要用到的区间的和还没有更新,这就是什么时候用到某个节点的值什么时候去更新的它的和
这道题唯一的一个难点也就在于去动态更新add数组和sum数组、