【线段树区间求和裸题】poj 3468 A Simple Problem with Integers

最新推荐文章于 2022-12-09 12:45:57 发布

原创最新推荐文章于 2022-12-09 12:45:57 发布 · 168 阅读

1 ·

CC 4.0 BY-SA版权

ACM 专栏收录该内容

57 篇文章

订阅专栏

本文介绍了解决POJ3468问题的一种方法，该问题涉及线段树的数据结构和lazy标记技巧来实现区间更新与查询操作。通过使用线段树进行区间求和和修改，可以高效地解决这类问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【线段树区间求和裸题】poj 3468 A Simple Problem with Integers

【poj链接】

【vj链接】

题目大意

第一行输入n和q
第二行n个位置的初始值
接下来q 行
首字母为‘Q’ ，跟两个数x,y 输出[x,y]区间数的和
首字母为‘C’ ，跟三个数x,y,z 把[x,y]区间里每个数加上z

Sample Input

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

Sample Output

解题思路

线段树求和，lazy标记

AC代码

#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
#define LL long long
const LL maxn=100000+100;

struct p
{
    LL l,r,sum;
}tree[maxn<<2];
LL lazy[maxn<<2];
LL cnt;
LL a[maxn];
void push_down(LL i)
{
    lazy[i*2]+=lazy[i];
    lazy[i*2+1]+=lazy[i];
    tree[i].sum+=(tree[i].r-tree[i].l+1)*lazy[i];
    lazy[i]=0;
}
void build(LL i,LL l,LL r)
{
    tree[i].l=l;
    tree[i].r=r;
    tree[i].sum=0;
    if(l==r)
    {
        tree[i].sum=a[++cnt];
        return;
    }
    LL mid=(l+r)>>1;
    build(i*2,l,mid);
    build(i*2+1,mid+1,r);
    tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;
}
void update(LL i,LL l,LL r,LL v)
{
    if(tree[i].l==l&&tree[i].r==r)
    {
        lazy[i]+=v;
        return;
    }
    tree[i].sum+=(r-l+1)*v;
    if(lazy[i]!=0)
    {//printf("YYYYYYYYYYYYYY");
        push_down(i);
    }
    LL mid=(tree[i].l+tree[i].r)>>1;
    if(r<=mid) update(i*2,l,r,v);
    else if(l>mid) update(i*2+1,l,r,v);
    else
    {
        update(i*2,l,mid,v);
        update(i*2+1,mid+1,r,v);
    }
    //tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;
}
LL q(LL i,LL l,LL r)
{
    if(tree[i].l==l&&tree[i].r==r)
    {
        return tree[i].sum+(r-l+1)*lazy[i];
    }
    if(lazy[i]!=0)
    {
        push_down(i);
    }
    LL mid=(tree[i].l+tree[i].r)>>1;
    if(r<=mid) {return q(i*2,l,r);}
    else if(l>mid) { return q(i*2+1,l,r);}
    else
    {
        return q(i*2,l,mid)+q(i*2+1,mid+1,r);
    }
}
int main()
{
    char s[5];
    LL n,Q,i,x,y,z;
    scanf("%lld%lld",&n,&Q);
    for(i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    cnt=0;
    build(1,1,n);
    while(Q--)
    {
        scanf("%s",s);
        if(s[0]=='C')
        {
            scanf("%lld%lld%lld",&x,&y,&z);
            update(1,x,y,z);
        }
        else if(s[0]=='Q')
        {
            scanf("%lld%lld",&x,&y);
            printf("%lld\n",q(1,x,y));
        }
    }
    return 0;
}