POJ 3468 A Simple Problem with Integers(线段树 成段增减,区间求和)


                                       A Simple Problem with Integers


Time Limit:5000MS     Memory Limit:131072KB     64bit IO Format:%I64d & %I64u



Description

给出了一个序列,你需要处理如下两种询问。

"C a b c"表示给[a, b]区间中的值全部增加c (-10000 ≤ c ≤ 10000)。

"Q a b" 询问[a, b]区间中所有值的和。

Input

第一行包含两个整数N, Q。1 ≤ N,Q ≤ 100000.

第二行包含n个整数,表示初始的序列A (-1000000000 ≤ Ai ≤ 1000000000)。

接下来Q行询问,格式如题目描述。

Output

对于每一个Q开头的询问,你需要输出相应的答案,每个答案一行。

Sample Input

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

Sample Output

   
   
4
55
9
15


求一段区间的和,一般是三种方法,一种就是暴力求,直接累加,在这道题中无疑是不适用的,所以不考虑,第二种是 树状数组,树状数组既可以完成点的修改,又可以完成区间求和,感觉不错,写写试试吧,结果会是一个TLE ,那么就只能用第三种的了,线段树,线段树种类有多种,如果用点修改,那么无疑的和树状数组一样的结果,所以要用 区间修改 。


线段树单点修改和区间修改的区别;

对于点修改和区间修改区别其实并不大,在点修改的时候,我们是找到该点,然后对该点进行加减操作,然后再一次递推修改其他的点,然后在区间修改中,有两种修改情况,一种就是和点修改一样的,找到点,修改点,递推其他点,但是,这只是其中的一点修改,这种修改方式只占用小部分。占大部分的还是对整个区间进行修改,如果该结点所表示的区间在要更改的区间内,那么就对该点直接修改,该节点加的值为 该区间每个结点要加的值得总和



代码的函数名称的功能

●  PushUP(int rt)是把当前结点的信息更新到父结点

●  PushDown(int rt)是把当前结点的信息更新给儿子结点

●  build(int l,int r,int rt) 建立线段树

●  rt表示当前子树的根(root),也就是当前所在的结点



附上代码

#include <cstdio>
#include <algorithm>
using namespace std;

#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
#define LL long long
const int maxn = 111111;
LL add[maxn<<2];
LL sum[maxn<<2];

void PushUp(int rt) {      //  根据孩子节点推父亲节点           
	sum[rt] = sum[rt<<1] + sum[rt<<1|1];      
}

void PushDown(int rt,int m) {    //   根据父亲推孩子      如果父亲的区间长度为奇数的时候,左孩子区间为 (m - (m >> 1)  较大的那一部分 
       if (add[rt]) {
              add[rt<<1] += add[rt];   //   父节点推子节点    左儿子节点   rt << 1 
              add[rt<<1|1] += add[rt]; //右儿子rt << 1 | 1  即为( rt << 1 + 1 )  
              sum[rt<<1] += add[rt] * (m - (m >> 1));    //   左孩子的区间长度为   (m - (m >> 1)  
              sum[rt<<1|1] += add[rt] * (m >> 1);
              add[rt] = 0;
       }
}


void build(int l,int r,int rt) {
       add[rt] = 0;
       if (l == r) {
              scanf("%lld",&sum[rt]);
              return ;
       }
       int m = (l + r) >> 1;
       build(lson);
       build(rson);
       PushUp(rt);
}

void update(int L,int R,int c,int l,int r,int rt) {
       if (L <= l && r <= R) {     //   如果该节点的区间在要修改的区间的范围内,那么对该点进行修改 
              add[rt] += c;
              sum[rt] += (LL)c * (r - l + 1);
              return ;
       }
       PushDown(rt , r - l + 1);       //   根据该点的信息  递推修改儿子节点的信息 
       int m = (l + r) >> 1;
       if (L <= m) update(L , R , c , lson);
       if (m < R) update(L , R , c , rson);
       PushUp(rt);          //  修改过下面的节点以后,在递归返回的过程中 利用儿子节点修改父亲节点 
}

LL query(int L,int R,int l,int r,int rt) {
       if (L <= l && r <= R) {  //如果该点的区间在所求区间内,直接返回该值 
              return sum[rt];
       }
       PushDown(rt , r - l + 1);  //根据该点的信息  递推修改儿子节点的信息 
       int m = (l + r) >> 1;
       LL ret = 0;
       if (L <= m) ret += query(L , R , lson);       
       if (m < R) ret += query(L , R , rson);
       return ret;
}

int main() {
       int N , Q;
       scanf("%d%d",&N,&Q);
       build(1 , N , 1);    //   建树 
       while (Q --) {
              char op[2];
              int a , b , c;
              scanf("%s",op);
              if (op[0] == 'Q') {
                     scanf("%d%d",&a,&b);  //   求 a 到 b 范围内的 累加和 
                     printf("%lld\n",query(a , b , 1 , N , 1));  // 所求范围的 a 到 b 当前节点的表示区间范围  1 - n ,当前节点 1 
              } else {
                     scanf("%d%d%d",&a,&b,&c);  //  a b 范围内都增加 c  
                     update(a , b , c , 1 , N , 1);     
              }
       }
       return 0;
}







评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值