題目描述

qn姐姐最好了~

    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

輸入描述:

第一行兩個數n,m

接下來一行n個數表示初始序列

就下來m行每行第一個數為操作方法opt，

若opt=1或者opt=2，則之後跟著兩個數為l,r

若opt=3或者opt=4，則之後跟著三個數為l,r,x

操作意思為題目描述裡說的

輸出描述:

對於每一個操作1,2，輸出一行表示答案

示例1

輸入

複製

5 6
1 2 3 4 5
1 1 5
2 1 5
3 1 2 1
4 1 3 2
1 1 4
2 2 3

輸出

複製

15
55
16
41

備註:

對於100%的資料 n=10000,m=200000 (注意是等於號)

保證所有詢問的答案在long long 範圍內

PS：比較難的線段樹吧，但是這道題可以暴力過，線段樹注意資料更新和懶標記下放。

AC程式碼：

#include <iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<cmath>
#include<stack>
#include<string>
const int maxn=1e4+10;
const int mod=1e9+7;
const int inf=1e8;
#define me(a,b) memset(a,b,sizeof(a))
#define lowbit(x) x&(-x)
typedef long long ll;
using namespace std;
ll sum[maxn<<2],num[maxn<<2],add[maxn<<2],mul[maxn<<2];
void init(int n)
{
    for(int i=1;i<=n;i++)
        sum[i]=num[i]=add[i]=0,mul[i]=1;
}
void up_data(int rt)
{
    num[rt]=num[rt<<1]+num[rt<<1|1];
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        scanf("%lld",&num[rt]);
        sum[rt]=num[rt]*num[rt];
        return ;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    up_data(rt);
}
void push_down(int l,int r,int rt)
{
    int m=(l+r)>>1;
    if(add[rt])
    {
        sum[rt<<1]+=(m-l+1)*add[rt]*add[rt]+2*add[rt]*num[rt<<1];
        sum[rt<<1|1]+=(r-m)*add[rt]*add[rt]+2*add[rt]*num[rt<<1|1];
        add[rt<<1]+=add[rt],add[rt<<1|1]+=add[rt];
        num[rt<<1]+=add[rt]*(m-l+1),num[rt<<1|1]+=add[rt]*(r-m);
        add[rt]=0;
    }
    if(mul[rt]>1)
    {
        mul[rt<<1]*=mul[rt],mul[rt<<1|1]*=mul[rt];
        add[rt<<1]*=mul[rt],add[rt<<1|1]*=mul[rt];
        num[rt<<1]*=mul[rt],num[rt<<1|1]*=mul[rt];
        sum[rt<<1]*=mul[rt]*mul[rt],sum[rt<<1|1]*=mul[rt]*mul[rt];
        mul[rt]=1;
    }
}
void push_data(int flog,int x,int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
    {
        if(flog==3)
            add[rt]*=x,num[rt]*=x,mul[rt]*=x,sum[rt]*=x*x;
        else
        {
            sum[rt]+=(r-l+1)*x*x+2*num[rt]*x;
            add[rt]+=x,num[rt]+=(r-l+1)*x;
        }
        return ;
    }
    push_down(l,r,rt);
    int mid=(l+r)>>1;
    if(L<=mid)
        push_data(flog,x,L,R,l,mid,rt<<1);
    if(R>mid)
        push_data(flog,x,L,R,mid+1,r,rt<<1|1);
    up_data(rt);
}
ll get_sum(int flog,int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
    {
        if(flog==1)
            return num[rt];
        return sum[rt];
    }
    push_down(l,r,rt);
    int mid=(l+r)>>1;
    ll s=0;
    if(L<=mid)
        s+=get_sum(flog,L,R,l,mid,rt<<1);
    if(R>mid)
        s+=get_sum(flog,L,R,mid+1,r,rt<<1|1);
    return s;
}
int main()
{
    int n,m;
    cin>>n>>m;
    init(n);
    build(1,n,1);
    while(m--)
    {
        int opt,l,r,k;
        scanf("%d",&opt);
        if(opt==1||opt==2)
        {
            scanf("%d%d",&l,&r);
            printf("%lld\n",get_sum(opt,l,r,1,n,1));
        }
        else
        {
            scanf("%d%d%d",&l,&r,&k);
            push_data(opt,k,l,r,1,n,1);
        }
    }
    return 0;
}