POJ-3468-A Simple Problem with Integers

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

南宮逸辰

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

阅读量637

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM POJ 文章标签：线段树

本文链接：https://blog.youkuaiyun.com/z309241990/article/details/11832569

ACM 同时被 2 个专栏收录

716 篇文章

订阅专栏

POJ

378 篇文章

订阅专栏

本文介绍了一种基于线段树的数据结构实现方法，用于处理区间查询和区间更新操作。通过具体的代码示例，详细解释了如何构建线段树、进行区间加法更新以及区间求和查询的过程。

这个题就是一个线段树的区间操作，题意也比较简单那，只有2种操作，一种是查询[L,R]，另外一种是[L,R]区间每个数都加上val，不太难

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=110000;
struct Node
{
    int l;
    int r;
    int len;
    long long sum;
    long long pos;
}t[maxn*3];
int n,m;
long long a[maxn];
void Build(int l,int r,int index)
{
    if(l==r)
    {
	t[index].l=l;
	t[index].r=r;
	t[index].len=r-l+1;
	t[index].sum=a[l];
	t[index].pos=0;
	return ;
    }
    int mid=(l+r)>>1;
    Build(l,mid,index<<1);
    Build(mid+1,r,index<<1|1);
    t[index].sum=t[index<<1].sum+t[index<<1|1].sum;
    t[index].l=l;
    t[index].r=r;
    t[index].len=r-l+1;
    t[index].pos=0;
}   
void pushdown(int index)
{
    if(!t[index].pos||t[index].l==t[index].r)
	return;
    t[index<<1].pos+=t[index].pos;
    t[index<<1|1].pos+=t[index].pos;
    t[index<<1].sum+=t[index<<1].len*t[index].pos;
    t[index<<1|1].sum+=t[index<<1|1].len*t[index].pos;
    t[index].pos=0;
}
void Update(int l,int r,int index,int val)
{
    pushdown(index);
    if(l==t[index].l&&r==t[index].r)
    {
	t[index].sum+=val*t[index].len;
	t[index].pos+=val;
	return;
    }
    int mid=(t[index].l+t[index].r)>>1;
    if(r<=mid)
	Update(l,r,index<<1,val);
    else if(l>mid)
	Update(l,r,index<<1|1,val);
    else
    {
	Update(l,mid,index<<1,val);
	Update(mid+1,r,index<<1|1,val);
    }
    t[index].sum=t[index<<1].sum+t[index<<1|1].sum;
}
long long Query(int l,int r,int index)
{
    pushdown(index);
    if(l==t[index].l&&r==t[index].r)
	return t[index].sum;
    int mid=(t[index].l+t[index].r)>>1;
    long long ans;
    if(r<=mid)
	ans=Query(l,r,index<<1);
    else if(l>mid)
	ans=Query(l,r,index<<1|1);
    else
	ans=Query(l,mid,index<<1)+Query(mid+1,r,index<<1|1);
    t[index].sum=t[index<<1].sum+t[index<<1|1].sum;
    return ans;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
	for(int i=1;i<=n;i++)
	    scanf("%I64d",&a[i]);
	Build(1,n,1);
	while(m--)
	{
	    char op[5];
	    scanf("%s",op);
	    if(op[0]=='Q')
	    {
		int l,r;
		scanf("%d%d",&l,&r);
		printf("%I64d\n",Query(l,r,1));	
	    }
	    else
	    {
		int l,r,val;
		scanf("%d%d%d",&l,&r,&val);
		Update(l,r,1,val);
	    }
	}
    }
    return 0;
}