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 100000
int 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;
}