QDUOJ 39 - 签到题(线段树)

签到题
发布时间: 2015年11月8日 22:31 最后更新: 2016年6月30日 21:58 时间限制: 1000ms 内存限制: 128M

描述
QD大学创新实验室的学生最喜欢的就是A题,创新实验室一共有N个学生, 他们的学号分别为1-N,L老师想经常知道学号a到学号b的同学(包括a,b)一共A了多少题, 并且L老师问完之后同学们A题的数量可能会增加, 之后老师再询问的时候,也要考虑新增的A题数.

输入
第一行一个整数T, 表示有T(T <= 20)组测试数据 每组测试数据第一行有两个整数N(1 <= N <= 100000), M(1 <= M <= 100000), N表示有N个学生, 学号为(1-N), M表示有M次操作. 接下来一行有N(0 <= N <= 100)个整数, 第i个整数表示学号为i的同学A的题数. 随后的M行是M条指令, 这M条指令分为两种, 每种包含一个字符串和两个整数 1> ADD a b (1 <= a <= N, 0 <= b <= 100)表示学号为a的同学又A了b道题目 2> QUERY a b (1 <= a <= b <= N) 表示询问学号为a到b的学生一共A了多少题目

输出
对于每次查询，输出一个整数X表示学号为a的同学到学号为b的同学的A的题目总数，每组输出占一行

样例输入1 复制
1
5 6
1 2 3 4 5
QUERY 1 3
ADD 1 2
QUERY 1 3
ADD 2 3
QUERY 1 2
QUERY 1 5
样例输出1
6
8
8
20

AC代码

#include<stdio.h>#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1#define maxn 100000int SegTree[maxn<<2];void PushUp(int rt){    SegTree[rt] = SegTree[rt<<1] + SegTree[rt<<1|1];    return ;}void build(int l,int r,int rt){    if(l == r){        scanf("%d",&SegTree[rt]);        return ;    }    int m = (l + r) >> 1;    build(lson);    build(rson);    PushUp(rt);}void Update(int pos,int num,int l,int r,int rt){    if(l == r)    {        SegTree[rt] += num;        return ;    }    int m = (l + r) >> 1;    if(pos <= m)    Update(pos,num,lson);    else            Update(pos,num,rson);    PushUp(rt);}int Query(int L,int R,int l,int r,int rt){    if(L <= l && R >= r)    return SegTree[rt];    int m = (l+r) >> 1;    int res = 0;    if(L <= m) res += Query(L,R,lson);    if(R > m) res += Query(L,R,rson);    return res;}int main(){    int loop;    scanf("%d",&loop);    while(loop--)    {        int n;        int m;        scanf("%d%d",&n,&m);        build(1,n,1);        while(m--)        {            char tmp[8];            int a;            int b;            scanf("%s%d%d",tmp,&a,&b);            if(tmp[0] == 'Q')   printf("%d\n",Query(a,b,1,n,1));            else if(tmp[0] == 'A')  Update(a,b,1,n,1);        }    }    return 0;}