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;
    }
}

 

全部评论

相关推荐

01-02 00:50
三峡大学 Java
程序员牛肉:这简历一出手就离失业不远了。 作为一家公司来讲,我如果要招日常实习生,那我对实习生最基本的要求就是要能干活,毕竟你就待三四个月,谁会留心培养你? 那么除了院校之外,最重要的就是项目和实习了。没有实习的话项目就好好搞。 但是你说你这个项目吧:课程作业管理系统和TMS运输管理系统。这两个基本就和闹着玩差不多。 你作为一个想要应聘Java开发实习生的人,对后端的理解还仅仅停留在:“使用mapper和sql映射”,“使用SQL进行多表调用”,“基于MySQL简历表结构”,“基于Spring boot完成CURD操作”这种玩具上......... 找不到后端实习的
点赞 评论 收藏
分享
01-02 21:17
已编辑
西安理工大学 后端
程序员小白条:项目不太重要,你的优势的算法竞赛,然后多背相关的八股文,项目可以不作为重点考虑,面试可能就简单带过项目就行了,你可以直接写简历,背项目相关的八股文就行,也不用自己做,时间紧张的情况下,性价比最高
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务