【线段树基础】NKOJ 1321 数列操作

时间限制 : 10000 MS   空间限制 : 165536 KB

问题描述

假设有一列数{Ai}(1≤i≤n)，支持如下两种操作：
将Ak的值加D。（k, D是输入的数）
输出As+As+1+…+At。（s, t都是输入的数，S≤T）

输入格式

第一行一个整数n，
第二行为n个整数，表示{Ai}的初始值≤10000。
第三行为一个整数m，表示操作数
下接m行，每行描述一个操作，有如下两种情况：
ADD k d (表示将Ak加d，1<=k<=n，d为数,d的绝对值不超过10000)
SUM s t （表示输出As+…+At）

输出格式

对于每一个SUM提问，输出结果

样例输入 1

5 
1 2 3 2 4 
5 
SUM 1 2
SUM 1 5
ADD 1 2
SUM 1 2
SUM 1 5

样例输出 1

3 
12 
5 
14

样例输入 2

10
44 37 20 29 13 8 32 14 46 29 
8
ADD 5 3
SUM 1 8
SUM 4 6
ADD 3 18
SUM 2 5
ADD 4 15
SUM 1 7
SUM 5 10

样例输出 2

200
53
120
219
145

提示

M,N<=100000

【提示】

线段树板题

【参考代码】

 #include<cstdio>
 #define maxn 100003
 int n, m;
 int A[maxn];
 char op[];
 struct node {
     int a, b, v;
 }Tree[maxn << ];
 namespace Ironclad_Programming {
     #define R register int
     #define For(i, s, n) for (R i = s; i <= n; ++ i)
     namespace ini {
         void MakeTree(int p, int x, int y) {
             Tree[p].a = x;
             Tree[p].b = y;
             if (x < y) {
                 MakeTree(p * , x, ((x + y) >> ));
                 MakeTree(p *  + , ((x + y) >> ) + , y);
                 Tree[p].v = Tree[p * ].v + Tree[p *  + ].v;
             } else Tree[p].v = A[x];
         }
         void executive() {
             scanf("%d", &n);
             For (i, , n)scanf("%d", &A[i]);
             MakeTree(, , n);
         }
     }
     namespace solve {
         void Add(int p, int k, int d) {
             Tree[p].v += d;
             if (Tree[p].a == Tree[p].b)return;
             if (Tree[p * ].a <= k && Tree[p * ].b >= k)Add(p * , k, d);
             if (Tree[p *  + ].a <= k && Tree[p *  + ].b >= k)Add(p *  + , k, d);
         }
         int GetSum(int p, int s, int t) {
             if (t < Tree[p].a || s > Tree[p].b)return ;
             if (s <= Tree[p].a && Tree[p].b <= t)return Tree[p].v;
             else {
                 int Total = ;
                 Total += GetSum(p * , s, t);
                 Total += GetSum(p *  + , s, t);
                 return Total;
             }
         }
         void executive() {
             scanf("%d", &m);
             For (i, , m) {
                 scanf("%s", op);
                 if (op[] == 'A') {
                     int k, d;
                     scanf("%d%d", &k, &d);
                     Add(, k, d);
                 } else {
                     int s, t;
                     scanf("%d%d", &s, &t);
                     printf("%d\n", GetSum(, s, t));
                 }
             }
         }
     }
     void Main() {
         ini::executive();
         solve::executive();
     }
     #undef R
     #undef For
 }
 int main() {
     Ironclad_Programming::Main();
     return ;
 }