838CDOJ 母仪天下

最新推荐文章于 2015-08-17 22:22:07 发布

原创最新推荐文章于 2015-08-17 22:22:07 发布 · 504 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一个简单的线段树模板实现方法，包括构建、更新和查询操作。适用于解决区间求和等问题，通过递归方式实现了高效的区间操作。

一个简单的线段树，直接套模板就OK。

#include<stdio.h>
#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

struct Tree
{
    int left,right;
    int sum;
}tree[400009];

int s[200001];

void build(int id, int l, int r)
{
    tree[id].left=l; tree[id].right=r;
    if(l==r)
    {
        tree[id].sum=s[l];
        return ;
    }
    int mid=(l+r)/2;
    build(id*2,l,mid);
    build(id*2+1,mid+1,r);
    tree[id].sum=tree[id*2].sum+tree[id*2+1].sum;
}

void update(int id,int pos,int val)
{
    if(tree[id].left==tree[id].right)
    {
        tree[id].sum+=val;
        return;
    }
    if(tree[id*2].right>=pos)
        update(id*2,pos,val);
    else
        update(id*2+1,pos,val);
    tree[id].sum=tree[id*2].sum+tree[id*2+1].sum;
}

int query(int id,int l, int r)
{
    if(tree[id].left==l&&tree[id].right==r)
    return tree[id].sum;
    else
    {
        int mid=(tree[id].left+tree[id].right)/2;
        if(r<=mid)
        return query(id*2,l,r);
        else if(l>mid)
        return query(id*2+1,l,r);
        else
        return query(id*2,l,mid)+query(id*2+1,mid+1,r);
    }
}

int main()
{
    int n,m,i,a,b,c;
    while(scanf("%d%d",&n,&m)==2)
{
    for(i=1;i<=n;i++)
    scanf("%d",&s[i]);
    build(1,1,n);
    while(m--)
    {
        scanf("%d%d%d",&a,&b,&c);
        if(a==1)
        {
            update(1,b,c);
        }
        else
        {
            printf("%d\n",query(1,b,c));
        }
    }
    }
    return 0;
}