【算法系列学习】线段树vs树状数组 单点修改，区间查询 [kuangbin带你飞]专题七 线段树 A

https://vjudge.net/contest/66989#problem/A

单点修改，区间查询

方法一：线段树

http://www.cnblogs.com/kuangbin/archive/2011/08/15/2139834.html

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>

using namespace std;
const int maxn=5e4+5;
const int inf=0x3f3f3f3f;
int num[maxn];
int n;
struct Node
{
    int l;
    int r;
    int nSum; 
}segTree[maxn<<2];

void Build(int i,int l,int r)
{
    segTree[i].l=l;
    segTree[i].r=r;
    if(l==r)
    {
        segTree[i].nSum=num[l];
        return;
    }
    int mid=(l+r)>>1;
    Build(i<<1,l,mid);
    Build((i<<1)|1,mid+1,r);
    segTree[i].nSum=segTree[i<<1].nSum+segTree[(i<<1)|1].nSum;
}

void Add(int i,int t,int b)
{
    segTree[i].nSum+=b;
    if(segTree[i].l==segTree[i].r)
    {
        return;    
    }    
    int mid=(segTree[i].l+segTree[i].r)>>1;
    if(t<=mid)
    {
        Add(i<<1,t,b);
    }
    else
    {
        Add((i<<1)|1,t,b);
    }
}

int Query(int i,int l,int r)
{
    if(segTree[i].l==l&&segTree[i].r==r)
    {
        return segTree[i].nSum;
    }
    int mid=(segTree[i].l+segTree[i].r)>>1;
    if(r<=mid)
    {
        return Query(i<<1,l,r);
    }
    else if(l>mid)
    {
        return Query((i<<1)|1,l,r);
    }
    else
    {
        return Query(i<<1,l,mid)+Query((i<<1)|1,mid+1,r);
    }
 } 
int main()
{
    int T;
    scanf("%d",&T);
    int kas=1;
    while(T--)
    {
        printf("Case %d:
",kas++);
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&num[i]);
        }    
        Build(1,1,n);
        char ch[20]; 
        int a,b;
        while(scanf("%s",ch)==1&&ch[0]!='E')
        {
            scanf("%d%d",&a,&b);
            if(ch[0]=='Q')
            {
                int ans=Query(1,a,b);
                printf("%d
",ans);
            }
            else if(ch[0]=='A')
            {
                Add(1,a,b);
            }
            else
            {
                Add(1,a,-b);
            }
        }
    }
    return 0;
 } 

方法二：树状数组

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>

using namespace std;
const int maxn=5e4+5;
const int inf=0x3f3f3f3f;
int tree[maxn];
int n;
void Init()
{
    memset(tree,0,sizeof(tree));
}
int lowbit(int x)
{
    return x&-x;
}

void Add(int i,int x)
{
    while(i<=n)
    {
        tree[i]+=x;
        i+=lowbit(i);
    }
}

int Query(int i)
{
    int res=0;
    while(i)
    {
        res+=tree[i];
        i-=lowbit(i);
    }
    return res;
 }
  
void Setval(int i,int x)
{
    int v=Query(i)-Query(i-1);
    Add(i,-v);
    Add(i,x); 
}

int Query(int l,int r)
{
    return Query(r)-Query(l-1);
 } 

int main()
{
    int T;
    scanf("%d",&T);
    for(int kas=1;kas<=T;kas++)
    {
        Init();
        scanf("%d",&n);
        int x;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x);
            Add(i,x);
        }
        printf("Case %d:
",kas);
        char ch[20];
        int a,b;
        while(scanf("%s",ch)==1&&ch[0]!='E')
        {
            scanf("%d%d",&a,&b);
            if(ch[0]=='A')
            {
                Add(a,b);
            }
            else if(ch[0]=='S')
            {
                Add(a,-b);
            }
            else
            {
                int ans=Query(a,b);
                printf("%d
",ans);
            }
        }
    }
    return 0;    
}