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

线段树 一个单独的数加上或减去某个数字 最后一段求和

时间:2016-07-13 19:29:10      阅读:109      评论:0      收藏:0      [点我收藏+]

标签:

#include<vector>
#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<math.h>
using namespace std;
#define INF 0x3f3f3f3f
#define N 60022
#define lson rood<<1
#define rson rood<<1|1
int w[N];
struct node
{
    int l,r,v,mid;
}a[N<<2];
void q(int rood,int l,int r)
{
    a[rood].l=l;a[rood].r=r;a[rood].mid=(r+l)/2;
    if(l==r)
    {
        a[rood].v=w[l];
        return ;
    }
    q(lson,l,a[rood].mid);
    q(rson,a[rood].mid+1,r);
    a[rood].v=a[lson].v+a[rson].v;
}   
void qq(int rood,int e,int f)
{
    a[rood].v+=f;
    if(a[rood].r==a[rood].l) return ;
    if(a[rood].mid>=e) qq(lson,e,f);
    else qq(rson,e,f);
}
int Q(int rood,int l,int r)
{
    if(a[rood].l==l&&a[rood].r==r)
        return a[rood].v;
    if(a[rood].mid>=r) return Q(lson,l,r);
    else if(a[rood].mid<l) return Q(rson,l,r);
    else
    {
        int x=Q(lson,l,a[rood].mid);
        int y=Q(rson,a[rood].mid+1,r);
        return x+y;
    }
}

int main()
{
    int T,i,e,f,n,t=1;
    char str[30];
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(i=1;i<=n;i++)
            scanf("%d",&w[i]);
        q(1,1,n);
        printf("Case %d:\n",t++);
        while(scanf("%s",str),strcmp(str,"End"))
        {
            scanf("%d%d",&e,&f);
            if(str[0]==Q)
            {
                printf("%d\n",Q(1,e,f));
                continue;
            }
            int ans=1;
            if(str[0]==S)
                ans*=-1;
            f*=ans;
            qq(1,e,f);
        }
    }
    return 0;
}

http://acm.hdu.edu.cn/showproblem.php?pid=1166

线段树 一个单独的数加上或减去某个数字 最后一段求和

标签:

原文地址:http://www.cnblogs.com/a719525932/p/5667177.html

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