线段树求助
  • 板块学术版
  • 楼主Binary_Lee
  • 当前回复9
  • 已保存回复9
  • 发布时间2022/8/15 19:45
  • 上次更新2023/10/27 15:15:57
查看原帖
线段树求助
241485
Binary_Lee楼主2022/8/15 19:45

重新写了遍区间修改、区间查询,不知哪里挂了qwq

#include<bits/stdc++.h>
using namespace std;
int n,m,a[400005],b[400005],x,l,r,k;
struct node{
	int l,r,sum,tag;
}tree[400005];
void build(int i,int l,int r){
	tree[i].l=l,tree[i].r=r;
	if(l==r){
		tree[i].sum=a[l];
		return ;
	}
	int md=(l+r)>>1;
	build(i*2,l,md);
	build(i*2+1,md+1,r);
	tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;
	return ;
}
void ad(int i,int l,int r,int k){
    tree[i].tag+=k;
    tree[i].sum+=k*(r-l+1);
    return ;
}
void pushdown(int i,int l,int r,int md){
    if(!tree[i].tag) return;
    ad(i*2,l,md,tree[i].tag);
    ad(i*2+1,md+1,r,tree[i].tag);
    tree[i].tag=0;
    return ;
} 
void add(int i,int l,int r,int x,int y,int k){
    if(l>=x&&r<=y){
        ad(i,l,r,k);
        return;
    }
    int md=(l+r)>>1;
    pushdown(i,l,r,md);
    if(x<=md) add(tree[i].l,l,md,x,y,k);
    if(md<y) add(tree[i].r,md+1,r,x,y,k);
    tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;
    return ;
}
int search(int i,int l,int r,int x,int y){
    if(l>=x&&r<=y) return tree[i].sum;
    int md=(l+r)>>1,ans=0;
    pushdown(i,l,r,md);
    if(x<=md) ans=search(i*2,l,md,x,y);
    if(md<y) ans+=search(i*2+1,md+1,r,x,y);
    return ans;
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	build(1,1,n);
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&x,&l,&r);
		if(x==1){
			scanf("%d",&k);
			add(1,1,n,l,r,k);
		}
		else printf("%d\n",search(1,1,n,l,r));
	}
	return 0;
}
2022/8/15 19:45
加载中...