【ybt】【数据结构 线段树 课过 例2】区间查改

区间查改

题目链接:YbtOJ


在这里插入图片描述

解题思路

线段树模板题。

code

#include<iostream>
#include<cstdio>
#define int long long
using namespace std;

int n,q;
int a[1000010];

struct abc{
	int x,y;
	int s,lzy;
}tree[4000010];

int build(int now,int x,int y)
{
	tree[now].x=x;
	tree[now].y=y;
	if(x==y)
	{
		tree[now].s=a[x];
		return tree[now].s;
	}
	int mid=(x+y)/2;
	int l=build(now*2,x,mid);
	int r=build(now*2+1,mid+1,y);
	return tree[now].s=l+r;
}

void down(int now)
{
	if(!tree[now].lzy)
		return;
	tree[now*2].lzy+=tree[now].lzy;
	tree[now*2+1].lzy+=tree[now].lzy;
	tree[now*2].s+=(tree[now*2].y-tree[now*2].x+1)*tree[now].lzy;
	tree[now*2+1].s+=(tree[now*2+1].y-tree[now*2+1].x+1)*tree[now].lzy;
	tree[now].lzy=0;
}

void add(int now,int l,int r,int g)
{
	if(tree[now].x>=l&&tree[now].y<=r)
	{
		tree[now].s+=(tree[now].y-tree[now].x+1)*g;
		tree[now].lzy+=g;
		return;
	}
	down(now);
	int mid=(tree[now].x+tree[now].y)/2;
	if(r<=mid)
		add(now*2,l,r,g);
	else if(l>mid)
		add(now*2+1,l,r,g);
	else
		add(now*2,l,r,g),add(now*2+1,l,r,g);
	tree[now].s=tree[now*2].s+tree[now*2+1].s;
}

int fd(int now,int l,int r)
{
	if(tree[now].x>=l&&tree[now].y<=r)
		return tree[now].s;
	int mid=(tree[now].x+tree[now].y)/2;
	down(now);
	if(r<=mid)
		return fd(now*2,l,r);
	else if(l>mid)
		return fd(now*2+1,l,r);
	else
		return fd(now*2,l,r)+fd(now*2+1,l,r);
}

signed main()
{
	cin>>n>>q;
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	build(1,1,n);
	while(q--)
	{
		int t,l,r;
		scanf("%lld%lld%lld",&t,&l,&r);
		if(t==1)
		{
			int g;
			scanf("%lld",&g);
			add(1,l,r,g);
		}
		else
			cout<<fd(1,l,r)<<endl;
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值