ACM新手DAY 16 树状数组

题解

A - 敌兵布阵

树状数组样题：第一行一个整数T，表示有T组数据。
每组数据第一行一个正整数N（N<=50000）,表示敌人有N个工兵营地，接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人（1<=ai<=50）。
接下来每行有一条命令，命令有4种形式：
(1) Add i j,i和j为正整数,表示第i个营地增加j个人（j不超过30）
(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人（j不超过30）;
(3)Query i j ,i和j为正整数,i<=j，表示询问第i到第j个营地的总人数;
(4)End 表示结束，这条命令在每组数据最后出现;
每组数据最多有40000条命令
对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问，输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
///lowbit(x)是x这个数字对应的a[x]记录的原单元格的长度
#define lowbit(x) (x&(-x))//树状数组最巧妙之处：i&(-i)
int n, a[50005];
void update(int i, int val) //更新函数，即add与sub操作
{
    while(i <= n)
    {
        a[i] += val;
        i += lowbit(i);//lowbit更新部分
    }
}
///这个部分是得到了每个输入的i的前i个数的和，所以最后是两个相减的结果
int query(int i)      //求和函数
{
    int sum = 0;
    while(i > 0)
    {
        sum += a[i];
        i -= lowbit(i);//更新的时候是加，现在变成了减
    }
    return sum;
}
int main()
{
    int val, t, x, y, cou = 1;
    scanf("%d", &t);
    while(t--)
    {
        memset(a, 0, sizeof(a));
        scanf("%d", &n);
        ///这次读入之后，a[2]记录的是前两个，a[4]记录的是前四个。。。
        for(int i = 1; i <= n; i++)//营地人数读入
        {
            scanf("%d", &val);
            update(i, val);
        }
        printf("Case %d:\n", cou++);
        char ch[15];
        while(scanf("%s", ch))
        {
            if(ch[0] == 'E') break;//END截止
            scanf("%d %d", &x, &y);
            if(ch[0] == 'A') update(x, y);//Add操作
            else if(ch[0] == 'S') update(x, -y);//Sub操作
            else printf("%d\n", query(y)-query(x-1));//Query操作 两段区间和相减
        }
    }
    return 0;
}

B - Color the ball

题目：N个气球排成一排，从左到右依次编号为1,2,3…N.每次给定2个整数a b(a <= b),lele便为骑上他的“小飞鸽"牌电动车从气球a开始到气球b依次给每个气球涂一次颜色。但是N次以后lele已经忘记了第I个气球已经涂过几次颜色了，你能帮他算出每个气球被涂过几次颜色吗？

初看起来把上一题简单改一下就行了，然而时间会花费很久吧。。 ，还是要抓住每次每个气球涂一次这个点。

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
#define lowbit(x) (x&(-x))//树状数组最巧妙之处：i&(-i)
const int maxn = 100010;
int n, a[maxn];
void update(int i, int val) //更新函数，即add与sub操作
{
    while(i <= n)
    {
        a[i] += val;
        i += lowbit(i);//lowbit更新部分
    }
}
int query(int i)      //求和函数
{
    int sum = 0;
    while(i > 0)
    {
        sum += a[i];
        i -= lowbit(i);//更新的时候是加，现在变成了减
    }
    return sum;
}
int main()
{
    int x, y;
    while(~scanf("%d", &n) && n)
    {
        memset(a, 0, sizeof(a));

        int tt = n;
        while(tt--)
        {
            scanf("%d %d", &x, &y);
            ///由于每次区间修改每个气球只是加1，而query函数得到的是每个点前面所有的，所以
            ///这里只把修改区间的左端点加1，然后右端点向右移1格减1，query得到的就只是单点的了
            update(x, 1);
            update(y+1, -1);//Add操作
        }
        printf("%d", query(1));
        for(int i=2; i<=n; i++)
            printf(" %d", query(i));
        printf("\n");
    }
    return 0;
}