HDU1166 线段树 或 树状数组

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1166

题意：给定一个区间长度，三种操作，ADD i, j是将id为i的位置的值增加j；SUB i，j是将id为i 的位置的值减j；QUERY i，j为查询闭区间i，j的和。很经典的线段树单点修改和区间和查询。

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <algorithm>
#include <string>
#include <cstring>
using namespace std;
const int maxn = 55555;
int sum[maxn << 2];//按位左偏一位相当于乘以2，偏移2位相当于乘以2的平方
void PushUp(int rt)
{
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];//这个按位或就是+1的意思
}

void build(int l, int r, int rt)//建立线段树
{
    if (l == r)
    {
        scanf("%d", &sum[rt]);
        return;
    }
    int m = (l + r) >> 1;//按位右偏1位相当于除以2
    build(l, m, rt << 1);
    build(m + 1, r, rt << 1 | 1);
    PushUp(rt);
}

void update(int p, int add, int l, int r, int rt)
{
    if (l == r)
    {
        sum[rt] += add;
        return;
    }
    int m = (l + r) >> 1;
    if (p <= m) update(p, add, l, m, rt << 1);
    else update(p, add, m + 1, r, rt << 1 | 1);
    PushUp(rt);
}

int query(int ll, int rr, int l, int r, int rt)//查询线段树
{
    if (ll <= l && rr >= r) return sum[rt];
    int m = (l + r) >> 1;
    int ret = 0;
    if (ll <= m) ret += query(ll, rr, l, m, rt << 1);
    if (rr > m) ret += query(ll, rr, m + 1, r, rt << 1 | 1);
    return ret;
}


int main()
{
    int t;
    scanf("%d",&t);
    int kase=1;
    while(t--){
        printf("Case %d:\n",kase++);
        int n;
        scanf("%d",&n);
        build(1,n,1);
        char op[20];
        int l,r;
        while(~scanf("%s",op)&&op[0]!='E'){
            scanf("%d%d",&l,&r);
            if(op[0]=='Q'){
                printf("%d\n",query(l,r,1,n,1));
            }
            else if(op[0]=='A')
                update(l,r,1,n,1);
            else if(op[0]=='S')
                update(l,-r,1,n,1);
        }
    }
    return 0;
}

---

其实在做树状数组的时候觉得也可以用树状数组来做，思路都差不多。

/*树状数组模板题*/
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 500005;
int que[maxn],c[maxn];
int lowbit(int x)
{
    return x&(-x);//用来计算a^k中的k
}
//实现每个树状数组元素所分管的元素之和的计算
void create(int n)
{
    int i,s,j;
    for(i=1;i<=n;i++)
    {
        s=lowbit(i);
        for(j=0;j<s;j++)//含自身向前推s个
        {
            c[i]=c[i]+que[i-j];
        }
    }
}
//求数组的和的算法如下
int sum(int n)
{
    int s=0;
    while(n>0)
    {
        s+=c[n];
        n=n-lowbit(n);
    }
    return s;
}
//修改操作，往上修改
void change(int i,int n,int x)
{
    while(i<=n)
    {
        c[i]=c[i]+x;
        i=i+lowbit(i);
    }
}
int main()
{
    int t,kase=1,i,j,n,x;
    scanf("%d",&t);
    while(t--)
    {
        memset(c,0,sizeof(c));
        scanf("%d",&n);
        for(i=1;i<=n;i++)
            scanf("%d",&que[i]);
        create(n);
        printf("Case %d:\n",kase++);
        char op[20];
        while(~scanf("%s",op)&&op[0]!='E'){
            scanf("%d%d",&j,&x);
            if(op[0]=='A')  change(j,n,x);
            else if(op[0]=='S') change(j,n,-x);
            else if(op[0]=='Q') printf("%d\n",sum(x)-sum(j-1));
        }
    }
    return 0;
}

板子都是比较常见的板子，另外这两种做法的效率都差不多,但是貌似线段树的内存还要少一点。