POJ 3468 A simple problem with Integers (线段树区间更新,区间查询) 2/5

题目链接:http://poj.org/problem?id=3468

首先,感谢雨巨的线段树入门视频,可算让我整明白线段树的原理,还学到了一些编码的技巧。orz,再次感谢雨巨。

这道题目就是裸的线段树,区间更新,区间查询

代码:

#include<stdio.h>

#define LL long long
struct SeTreeNode
{
    int left,right;
    LL int lazy;
    LL int len;
    LL int sum;
};
SeTreeNode tree[400000];
LL int a[100010];
void build(int id,int l,int r);
void update(int id,int l,int r,LL int val);
void PushDown(int id);

LL query(int id,int l,int r);
int main()
{
    int  T,n,a1,a2,q;
    LL a3;
    char s[12];

    scanf("%d%d",&n,&q);
    for(int i=1 ; i<=n ; i++)
    {
        scanf("%lld",&a[i]);
    }
    build(1,1,n);

    while(q--)
    {
        scanf(" %s",s);

        scanf("%d%d",&a1,&a2);
        if(s[0]=='Q')
        {
            LL int ans=query(1,a1,a2);
            printf("%lld\n",ans);
        }
        if(s[0]=='C')
        {
            scanf("%lld",&a3);
            update(1,a1,a2,a3);

        }
    }

    return 0;
}

void build(int id,int l,int r)
{
    tree[id].left=l,tree[id].right=r;
    tree[id].len=tree[id].right-tree[id].left+1;
    tree[id].lazy=0;
    if(tree[id].left==tree[id].right)
    {
        tree[id].sum=a[l];
        //tree[id].lazy=0;
        //tree[id].len=tree[id].right-tree[id].left+1;
        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;
}
LL int query(int id,int l,int r)
{
    if(tree[id].left>=l&&tree[id].right<=r)
        return tree[id].sum;

    if(tree[id].left>r||tree[id].right<l)
        return  0;

    if(tree[id].lazy)
        PushDown(id);

    return query(id*2,l,r)+query(id*2+1,l,r);

}
void update(int id,int l,int r,LL int val)
{
    if(tree[id].left>=l&&tree[id].right<=r)
    {
        tree[id].sum+=val*tree[id].len;
        tree[id].lazy+=val;
        return ;
    }
    if(tree[id].left>r||tree[id].right<l)
        return ;
    if(tree[id].lazy)
        PushDown(id);

    update(id*2+1,l,r,val);

    update(id*2,l,r,val);

    tree[id].sum=tree[id*2].sum+tree[id*2+1].sum;

}
void PushDown(int id)
{
    if(tree[id].lazy)
    {
        tree[id*2].lazy+=tree[id].lazy;
        tree[id*2+1].lazy+=tree[id].lazy;
        tree[id*2].sum+=tree[id].lazy*tree[id*2].len;
        tree[id*2+1].sum+=tree[id].lazy*tree[id*2+1].len;
        tree[id].lazy=0;
    }
}

 

全部评论

相关推荐

面试摇了我吧:啊哈哈面试提前五个小时发,点击不能参加就是放弃
点赞 评论 收藏
分享
10-09 00:50
已编辑
长江大学 算法工程师
不期而遇的夏天:1.同学你面试评价不错,概率很大,请耐心等待;2.你的排名比较靠前,不要担心,耐心等待;3.问题不大,正在审批,不要着急签其他公司,等等我们!4.预计9月中下旬,安心过节;5.下周会有结果,请耐心等待下;6.可能国庆节前后,一有结果我马上通知你;7.预计10月中旬,再坚持一下;8.正在走流程,就这两天了;9.同学,结果我也不知道,你如果查到了也告诉我一声;10.同学你出线不明朗,建议签其他公司保底!11.同学你找了哪些公司,我也在找工作。
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务