zoukankan      html  css  js  c++  java
  • 洛谷P3372线段树模板1——线段树

    题目:https://www.luogu.org/problemnew/show/P3372

    线段树模板。

    代码如下:

    #include<iostream>
    #include<cstdio>
    using namespace std;
    long long n,m,a[100005],ct;
    struct N{
    	long long lazy,sum;
    	long long ls,rs;
    }p[200005];
    void pushdown(long long cur,long long l,long long r)
    {
    	long long mid=(l+r)/2;
    	long long lson=p[cur].ls,rson=p[cur].rs;
    	p[lson].lazy+=p[cur].lazy;
    	p[rson].lazy+=p[cur].lazy;
    	p[lson].sum+=p[cur].lazy*(mid-l+1);
    	p[rson].sum+=p[cur].lazy*(r-mid);
    	p[cur].lazy=0;
    }
    void pushup(long long cur)
    {
    	long long lson=p[cur].ls,rson=p[cur].rs;
    	p[cur].sum=p[lson].sum+p[rson].sum;
    }
    void build(long long l,long long r,long long cur)
    {
    	if(l==r)
    	{
    		p[cur].sum=a[l];
    		p[cur].ls=-1;p[cur].rs=-1;
    		return;
    	}
    	long long mid=(l+r)/2;
    	p[cur].ls=++ct;p[cur].rs=++ct;
    	build(l,mid,p[cur].ls);build(mid+1,r,p[cur].rs);
    	pushup(cur);
    }
    void add(long long l,long long r,long long L,long long R,long long c,long long cur)
    {
    	if(l>=L&&r<=R)
    	{
    		p[cur].lazy+=c;
    		p[cur].sum+=c*(r-l+1);
    		return;
    	}
    	pushdown(cur,l,r);//
    	long long mid=(l+r)/2;
    	if(mid<R)add(mid+1,r,L,R,c,p[cur].rs);
    	if(mid>=L)add(l,mid,L,R,c,p[cur].ls);
    	pushup(cur);//
    }
    long long query(long long l,long long r,long long L,long long R,long long cur)
    {
    	long long tot=0;
    	if(l>=L&&r<=R)
    		return p[cur].sum;
    	long long mid=(l+r)/2;
    	pushdown(cur,l,r);
    	if(mid<R)tot+=query(mid+1,r,L,R,p[cur].rs);
    	if(mid>=L)tot+=query(l,mid,L,R,p[cur].ls);
    	return tot;
    }
    int main()
    {
    	scanf("%lld%lld",&n,&m);
    	for(long long i=1;i<=n;i++)
    		scanf("%lld",&a[i]);
    	long long rt=++ct;
    	build(1,n,rt);
    	long long d,x,y,k;
    	for(long long i=1;i<=m;i++)
    	{
    		scanf("%d",&d);
    		if(d==1)
    		{
    			scanf("%lld%lld%lld",&x,&y,&k);
    			add(1,n,x,y,k,rt);
    		}
    		if(d==2)
    		{
    			scanf("%lld%lld",&x,&y);
    			printf("%lld
    ",query(1,n,x,y,rt));
    		}
    	}
    	return 0;
    }
    

      

  • 相关阅读:
    【Android】GLSurfaceView
    【Android】手机分辨率
    【Mac】快捷键锁屏
    【Android】View
    【Android】状态保存
    【Android】Notification
    【Android】ContentProvider
    【Android】Bitmap的管理
    【Android】缩略图Thumbnails
    【Android】Activity生命周期
  • 原文地址:https://www.cnblogs.com/Zinn/p/8439986.html
Copyright © 2011-2022 走看看