码迷,mamicode.com
首页 > 其他好文 > 详细

线段树

时间:2016-05-07 16:33:29      阅读:126      评论:0      收藏:0      [点我收藏+]

标签:

#include<iostream>
#include<cstdio>
using namespace std;
int a[100001],tot;
struct node
{
    int l,r;
    int ll,rr;
    int sum;
}tree[200001];
void build(int l,int r)
{
    int k=++tot;
    tree[k].l=l;
    tree[k].r=r;
    if(l==r-1)
    tree[k].sum=a[l];
    else
    {
        tree[k].ll=tot+1;
        build(l,(l+r)/2);
        tree[k].rr=tot+1;
        build((l+r)/2,r);
        tree[k].sum=tree[tree[k].ll].sum+tree[tree[k].rr].sum;
    }
}
int find(int k,int x,int y)
{
    if(x<=tree[k].l&&y>=tree[k].r)
    return tree[k].sum;
    int ans=0;
    if(x<(tree[k].l+tree[k].r)/2)ans+=find(tree[k].ll,x,y);
    if(y>(tree[k].l+tree[k].r)/2)ans+=find(tree[k].rr,x,y);
    return ans;
}
void change(int k,int x,int y)
{
    if(tree[k].l==tree[k].r-1)
    tree[k].sum+=y;
    else
    {
        if(x<(tree[k].l+tree[k].r)/2)
        change(tree[k].ll,x,y);
        else
        change(tree[k].rr,x,y);
        tree[k].sum=tree[tree[k].ll].sum+tree[tree[k].rr].sum;
    }
}
int main()
{
    int n,i,j,k;
    scanf("%d",&n);
    for(i=1;i<=n;i++)
    scanf("%d",&a[i]);
    build(1,n+1);
    int p;
    scanf("%d",&p);
    for(i=1;i<=p;i++)
    {
        int x,y,z,w;
        scanf("%d",&z);
        if(z==1)
        {
            scanf("%d%d",&x,&y);
            change(1,x,y);
        }
        else
        {
            scanf("%d%d",&x,&y);
            printf("%d\n",find(1,x,y+1));
        }
    }
    return 0;
}

 

线段树

标签:

原文地址:http://www.cnblogs.com/dingmenghao/p/5468411.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!