POJ---3468-A Simple Problem with Integers(线段树)

最新推荐文章于 2022-08-28 01:34:07 发布

EIKY

最新推荐文章于 2022-08-28 01:34:07 发布

阅读量300

点赞数

CC 4.0 BY-SA版权

分类专栏：树 POJ

本文链接：https://blog.youkuaiyun.com/qq978874169/article/details/49803255

POJ 同时被 2 个专栏收录

64 篇文章

订阅专栏

树

42 篇文章

订阅专栏

本文介绍了一种使用线段树解决区间更新和查询问题的方法。通过实例演示了如何实现区间加法操作及区间求和查询，适用于处理大规模数据集中的区间修改与查询任务。

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

B - A Simple Problem with Integers

Crawling in process... Crawling failed Time Limit:5000MS Memory Limit:131072KB 64bit IO Format:%I64d & %I64u

Submit Status

Description

给出了一个序列，你需要处理如下两种询问。

"C a b c"表示给[a, b]区间中的值全部增加c (-10000≤ c ≤ 10000)。

"Q a b" 询问[a, b]区间中所有值的和。

Input

第一行包含两个整数N,Q。1≤ N,Q ≤ 100000.

第二行包含n个整数，表示初始的序列A (-1000000000≤ Ai ≤1000000000)。

接下来Q行询问，格式如题目描述。

Output

对于每一个Q开头的询问，你需要输出相应的答案，每个答案一行。

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

线段树：区间更新 = =。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
#define SIZE 100050
int n,m;
int num[SIZE];
struct Node
{
    int a,b;
    long long sum;
    long long poi;
} node[SIZE<<2];
void build_tree(int a,int b,int i)
{
    node[i].a=a;
    node[i].b=b;
    if(a==b)
    {
        node[i].sum=num[a];
        node[i].poi=0;
        return;
    }
    int mid=(a+b)/2;
    build_tree(a,mid,i<<1);
    build_tree(mid+1,b,i<<1|1);
    node[i].sum=node[i<<1].sum+node[i<<1|1].sum;
    node[i].poi=0;
}
void add_num(int a,int b,long long num,int i)
{
    if(node[i].a==a && node[i].b==b)
    {
        node[i].poi+=num;
        return;
    }
    node[i].sum+=num*(b-a+1);
    int mid=(node[i].a+node[i].b)>>1;
    if(b<=mid)add_num(a,b,num,i<<1);
    else if(a>mid)add_num(a,b,num,i<<1|1);
    else
    {
        add_num(a,mid,num,i<<1);
        add_num(mid+1,b,num,i<<1|1);
    }
}
long long query(int a,int b,int i)
{
    if(node[i].a==a && node[i].b==b)
        return node[i].sum+node[i].poi*(b-a+1);
    if(node[i].poi)
    {
        node[i].sum+=node[i].poi*(node[i].b-node[i].a+1);
        node[i<<1].poi+=node[i].poi;
        node[i<<1|1].poi+=node[i].poi;
        node[i].poi=0;
    }
    int mid=(node[i].a+node[i].b)>>1;
    if(b<=mid)return query(a,b,i<<1);
    else if(a>mid)return query(a,b,i<<1|1);
    else
        return query(a,mid,i<<1) + query(mid+1,b,i<<1|1);
}
int main()
{
    int i,j,k,x,y,z;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(i=1; i<=n; ++i)
            scanf("%d",&num[i]);
        build_tree(1,n,1);
        char str;
        for(i=0; i<m; ++i)
        {
            cin>>str;
            if(str=='Q')
            {
                scanf("%d%d",&x,&y);
                cout<<query(x,y,1)<<endl;
            }
            else
            {
                scanf("%d%d%d",&x,&y,&z);
                add_num(x,y,z,1);
            }
        }
    }
    return 0;
}