线段树 (模板) 牛客 + Transformation HDU - 4578(加强版)

题目链接: https://ac.nowcoder.com/acm/contest/200/B
题意:
qn姐姐给你了一个长度为n的序列还有m次操作让你玩，
1 l r 询问区间[l,r]内的元素和
2 l r 询问区间[l,r]内的元素的平方 和
3 l r x 将区间[l,r]内的每一个元素都乘上x
4 l r x 将区间[l,r]内的每一个元素都加上x
思路:
这道题有许多需要注意的细节

1. 懒标记，注意初始化加法为0，乘法为1
2. 懒标记在下传的时候，是乘法先还是加法先，因为会出现我们在标记的时候加法标记还没下传，故在进行乘法标记时需要加法标记要同时*x
3. 加法下传时，求元素平方的和 $$(x+c{)}^2-x^2==2cx+c^2$$
   代码:

#include<bits/stdc++.h>
#define lson l,mid,root<<1
#define rson mid+1,r,root<<1|1
using namespace std;
typedef long long ll;

const int maxn=1e4+5;
ll n,m,tree[maxn<<2][2],add1[maxn<<2],add2[maxn<<2];

void pushup(ll root)
{
	tree[root][0]=tree[root<<1][0]+tree[root<<1|1][0];
	tree[root][1]=tree[root<<1][1]+tree[root<<1|1][1];

}

void pushdown1(ll l,ll r,ll root)
{
	if(add1[root])
	{
		add1[root<<1]+=add1[root];
		add1[root<<1|1]+=add1[root];
		ll x=tree[root<<1][0];
		ll y=tree[root<<1|1][0];
		tree[root<<1][0]+=l*add1[root];
		tree[root<<1|1][0]+=r*add1[root];
		tree[root<<1][1]+=2*x*add1[root]+l*add1[root]*add1[root];
		tree[root<<1|1][1]+=2*y*add1[root]+r*add1[root]*add1[root];
		add1[root]=0;
	}
}

void pushdown2(ll l,ll r,ll root)
{
	if(add2[root]!=1)
	{
		add2[root<<1]*=add2[root];
		add2[root<<1|1]*=add2[root];
		tree[root<<1][0]*=add2[root];
		tree[root<<1|1][0]*=add2[root];
		tree[root<<1][1]*=add2[root]*add2[root];
		tree[root<<1|1][1]*=add2[root]*add2[root];
		add2[root]=1;
	}
}

void build(ll l,ll r, ll root)
{
	add1[root]=0;
	add2[root]=1;
	if(l==r)
	{
		scanf("%lld",&tree[root][0]);
		tree[root][1]=tree[root][0]*tree[root][0];
		return ;
	}
	ll mid=(l+r)/2;
	build(lson);
	build(rson);
	pushup(root);
}

void update1(ll L,ll R,ll C,ll l,ll r,ll root)
{
	if(L<=l&&r<=R)
	{
		ll x=tree[root][0];
		tree[root][0]+=(r-l+1)*C;
		tree[root][1]+=2*x*C+(r-l+1)*C*C;
		add1[root]+=C;
		return ;
	}
	ll mid=(l+r)/2;
	pushdown2(mid-l+1,r-mid,root);
	pushdown1(mid-l+1,r-mid,root);
	if(L<=mid)
		update1(L,R,C,lson);
	if(R>mid)
		update1(L,R,C,rson);
	pushup(root);
}

void update2(ll L,ll R,ll C,ll l,ll r,ll root)
{
	if(L<=l&&r<=R)
	{
		tree[root][0]*=C;
		tree[root][1]*=C*C;
		if(add1[root])
			add1[root]*=C;
		add2[root]*=C;
		return ;
	}
	ll mid=(l+r)/2;
	pushdown2(mid-l+1,r-mid,root);
	pushdown1(mid-l+1,r-mid,root);
	if(L<=mid)
		update2(L,R,C,lson);
	if(R>mid)
		update2(L,R,C,rson);
	pushup(root);
}

ll query(ll L,ll R,ll c,ll l,ll r,ll root)
{
	if(L<=l&&R>=r)
	{
		if(c==1)
		    return tree[root][0];
		if(c==2)
		    return tree[root][1];
	}
	long long ans=0,mid=(l+r)>>1;
	pushdown2(mid-l+1,r-mid,root);
	pushdown1(mid-l+1,r-mid,root);
	if(L<=mid)
		ans+=query(L,R,c,lson);
	if(R>mid)
		ans+=query(L,R,c,rson);
	return ans;
}

int main()
{
	scanf("%lld%lld",&n,&m);
	build(1,n,1);
	while(m--)
	{
		ll l,r,x;
		ll op;
		scanf("%d",&op);
		if(op==1||op==2)
		{
			scanf("%lld%lld",&l,&r);
			printf("%lld\n",query(l,r,op,1,n,1));

		}
		else
		{
			scanf("%lld%lld%lld",&l,&r,&x);
			if(op==3)
				update2(l,r,x,1,n,1);
			else
				update1(l,r,x,1,n,1);
		}
	}
	return 0;
}

题目链接: https://vjudge.net/contest/66989#problem/K
题面: 三种操作+三种询问
思路: 这道题有两种做法，第一种类似上面，第二种就是因为本题初始化叶节点都是0，故较大概率存在一段区间内的元素都相等，故可以求到区间内元素均相等即可
若是一开始初始化随机，则用第一种办法较好，因为第二种每次极大可能要求到叶子节点这样容易TLE
搬运工
第一种

//代码来自https://blog.csdn.net/wl16wzl/article/details/80779406
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
 
using namespace std;
typedef long long ll;
const int maxn=100000+100; //数组大小
const ll mod=10007;
 
ll add[maxn<<2],mul[maxn<<2],cha[maxn<<2]; //懒惰标记
ll sum[maxn<<2],sum2[maxn<<2],sum3[maxn<<2];
int n,m,a,b,c,d;
void PushDown(int rt,ll m);
 
void change(int rt,ll m,ll c,int f)
{
    ll c2=c*c%mod;
    ll c3=c2*c%mod;
    if(f==1)
    {
        sum3[rt]=(sum3[rt]+3*sum2[rt]%mod*c%mod+3*sum[rt]%mod*c2%mod+c3*m%mod)%mod;
        sum2[rt]=(sum2[rt]+2*sum[rt]*c%mod+c2*m%mod)%mod;
        sum[rt]=(sum[rt]+c*m%mod)%mod;
        add[rt]=(add[rt]+c)%mod;
    }
    else if(f==2)
    {
        sum3[rt]=sum3[rt]*c3%mod;
        sum2[rt]=sum2[rt]*c2%mod;
        sum[rt]=sum[rt]*c%mod;
        mul[rt]=mul[rt]*c%mod;
        add[rt]=add[rt]*c%mod;
    }
    else if(f==3)
    {
        sum[rt]=c*m%mod;
        sum2[rt]=c2*m%mod;
        sum3[rt]=c3*m%mod;
        cha[rt]=c;
        add[rt]=0;
        mul[rt]=1;
    }
}
 
void PushUp(int rt)
{
    sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%mod;
    sum2[rt]=(sum2[rt<<1]+sum2[rt<<1|1])%mod;
    sum3[rt]=(sum3[rt<<1]+sum3[rt<<1|1])%mod;
}
 
void PushDown(int rt,ll m) //向下更新
{
    if(cha[rt]!=0)
    {
        ll ml=m%2==0?(m>>1):(m>>1)+1;
        change(rt<<1,ml,cha[rt],3);
        change(rt<<1|1,m-ml,cha[rt],3);
        cha[rt]=0;
    }
    if(mul[rt]!=1)
    {
        ll ml=m%2==0?(m>>1):(m>>1)+1;
        change(rt<<1,ml,mul[rt],2);
        change(rt<<1|1,m-ml,mul[rt],2);
        mul[rt]=1;
    }
    if(add[rt]!=0)
    {
        ll ml=m%2==0?(m>>1):(m>>1)+1;
        change(rt<<1,ml,add[rt],1);
        change(rt<<1|1,m-ml,add[rt],1);
        add[rt]=0;
    }
}
 
void build(int l,int r,int rt) //建树
{
    add[rt]=cha[rt]=0;
    mul[rt]=1;
    if(l==r)
    {
        sum[rt]=sum2[rt]=sum3[rt]=0;
        return;
    }
    ll m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    PushUp(rt);
}
 
void update(int L,int R,ll c,int l,int r,int rt,int f)
{
    if(L<=l&&R>=r)
    {
        ll m=r-l+1;
        change(rt,m,c,f);
        return;
    }
    if(l==r) return;
    PushDown(rt,r-l+1);
    ll m=(l+r)>>1;
    if(L<=m) update(L,R,c,l,m,rt<<1,f);
    if(m<R) update(L,R,c,m+1,r,rt<<1|1,f);
    PushUp(rt);
}
 
ll query(int L,int R,int l,int r,int rt,int f)
{
    if(L<=l&&R>=r)
    {
        if(f==1) return sum[rt];
        if(f==2) return sum2[rt];
        if(f==3) return sum3[rt];
    }
    PushDown(rt,r-l+1);
    ll m=(l+r)>>1;
    ll ret=0;
    if(L<=m) ret+=query(L,R,l,m,rt<<1,f);
    if(m<R) ret+=query(L,R,m+1,r,rt<<1|1,f);
    ret%=mod;
    return ret;
}
 
int main()
{
    //freopen("/home/zlwang/test.txt","r",stdin);
    while(~scanf("%d%d",&n,&m))
    {
        if(n==0&&m==0) break;
        build(1,n,1);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d%d",&a,&b,&c,&d);
            if(a==4)
            {
                printf("%lld\n",query(b,c,1,n,1,d));
            }
            else
            {
                update(b,c,d,1,n,1,a);
            }
        }
    }
    return 0;
}

第二种

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <vector>
#include <bitset>
 
using namespace std;
 
#define LL long long
const int INF = 0x3f3f3f3f;
 
const int mod=10007;
 
int n,q,flag[400009],x[400009];
 
void Union(int k,int l,int r)
{
    if(!flag[l]||!flag[r]) flag[k]=0;
    else if(x[l]!=x[r]) flag[k]=0;
    else flag[k]=1,x[k]=x[k<<1];
}
 
void update(int k,int l,int r,int ll,int rr,int val,int type)
{
    if(l>=ll&&r<=rr&&flag[k])
    {
        if(type==1) x[k]=(x[k]+val)%mod;
        else if(type==2) x[k]=(x[k]*val)%mod;
        else x[k]=val;
        return ;
    }
    if(flag[k])
    {
        flag[k<<1]=flag[k<<1|1]=1;
        x[k<<1]=x[k<<1|1]=x[k];
        flag[k]=0;
    }
    int mid=(l+r)>>1;
    if(ll<=mid) update(k<<1,l,mid,ll,rr,val,type);
    if(rr>mid) update(k<<1|1,mid+1,r,ll,rr,val,type);
    Union(k,k<<1,k<<1|1);
}
 
int query(int k,int l,int r,int ll,int rr,int val)
{
    if(flag[k]&&l>=ll&&r<=rr)
    {
        int ans=1;
        for(int i=1;i<=val;i++) ans=(ans*x[k])%mod;
        ans=(ans*(r-l+1))%mod;
        return ans;
    }
    if(flag[k])
    {
        flag[k<<1]=flag[k<<1|1]=1;
        flag[k]=0;
        x[k<<1]=x[k<<1|1]=x[k];
    }
    int mid=(l+r)>>1,ans=0;;
    if(mid>=ll) ans+=query(k<<1,l,mid,ll,rr,val);
    if(rr>mid) ans+=query(k<<1|1,mid+1,r,ll,rr,val);
    return ans%mod;
}
 
int main()
{
    while(~scanf("%d%d",&n,&q)&&(n+q))
    {
        memset(flag,1,sizeof flag);
        memset(x,0,sizeof x);
        int x,l,r,val;
        while(q--)
        {
            scanf("%d%d%d%d",&x,&l,&r,&val);
            if(x>=1&&x<=3) update(1,1,n,l,r,val,x);
            else printf("%d\n",query(1,1,n,l,r,val));
        }
    }
    return 0;
}