picks loves segment tree I

標籤：max   include   線段   hang   數列   整數   math   輸入輸出   lse   

picks loves segment tree I題目背景

來源：

\(\text {2018 WC Segment Tree Beats}\)

原

\(\text {C_SUNSHINE}\)

\(\text {jiry_2}\)

題目描述：

• 給定一個長度為\(n\)的數列\(A\)，接下來有\(m\)次操作：
• 區間\([l,r]\)中的所有數變成\(min(A_i,x)\)
• 詢問區間\([l,r]\)中所有數的和
• \(n,m \le 50000\)
• 我會分塊！
• \(n,m \le 500000\)
• 線段樹？

輸入輸出格式輸入格式

第一行兩個正整數表示\(N,M\)

第二行\(N\)個正整數，表示數列\(A_i\)

接下來\(M\)行每行包含\(3\)或\(4\)個整數，表示一個操作，具體如下：

操作1： 1 x y x 含義：將區間\([l,r]\)內每個數變成\(min(A_i,x)\)

操作2： 2 x y 含義：輸出區間\([l,r]\)內每個數的和

輸出格式

輸出包含若干行整數，即為所有操作\(2\)的結果。

說明：

對於所有的資料，有\(N \le 500000,M \le 500000,a_i \le 2 \times 10^9\)

保證所有出現的資料在\(int64/long \ long\)範圍內

本菜雞說不清楚，直接當板子了，看網上的dalao們都寫的Ⅴ，Ⅵ，Ⅶ，Ⅸ什麼的，我太菜，從基礎開始來吧。

Code:

#include <cstdio>#define ll long long#define ls id<<1#define rs id<<1|1const int N=5e5+10;ll mx[N<<2],se[N<<2],sum[N<<2],tag[N<<2],a[N];ll max(ll x,ll y){return x>y?x:y;}int cnt[N<<2],n,m;void updata(int id){    if(mx[ls]>mx[rs])    {        mx[id]=mx[ls];        cnt[id]=cnt[ls];        se[id]=max(se[ls],mx[rs]);    }    else if(mx[ls]<mx[rs])    {        mx[id]=mx[rs];        cnt[id]=cnt[rs];        se[id]=max(mx[ls],se[rs]);    }    else    {        mx[id]=mx[ls];        cnt[id]=cnt[ls]+cnt[rs];        se[id]=max(se[ls],se[rs]);    }    sum[id]=sum[ls]+sum[rs];}void build(int id,int l,int r){    if(l==r)    {        sum[id]=mx[id]=a[l];        cnt[id]=1;        se[id]=0;        return;    }    int mid=l+r>>1;    build(ls,l,mid),build(rs,mid+1,r);    updata(id);}void pushdown(int id){    if(tag[id])    {        if(mx[ls]>tag[id])        {            sum[ls]-=(mx[ls]-tag[id])*cnt[ls];            tag[ls]=mx[ls]=tag[id];        }        if(mx[rs]>tag[id])        {            sum[rs]-=(mx[rs]-tag[id])*cnt[rs];            tag[rs]=mx[rs]=tag[id];        }        tag[id]=0;    }}void change(int id,int L,int R,int l,int r,ll x){    if(mx[id]<=x) return;    if(L==l&&R==r&&se[id]<x)    {        sum[id]-=(mx[id]-x)*cnt[id];        tag[id]=mx[id]=x;        return;    }    pushdown(id);    int Mid=L+R>>1;    if(r<=Mid) change(ls,L,Mid,l,r,x);    else if(l>Mid) change(rs,Mid+1,R,l,r,x);    else change(ls,L,Mid,l,Mid,x),change(rs,Mid+1,R,Mid+1,r,x);    updata(id);}ll query(int id,int L,int R,int l,int r){    if(l==L&&r==R) return sum[id];    pushdown(id);    int Mid=L+R>>1;    if(r<=Mid) return query(ls,L,Mid,l,r);    else if(l>Mid) return query(rs,Mid+1,R,l,r);    else return query(ls,L,Mid,l,Mid)+query(rs,Mid+1,R,Mid+1,r);}int main(){    scanf("%d%d",&n,&m);    for(int i=1;i<=n;i++) scanf("%lld",a+i);    build(1,1,n);    ll x;    for(int op,l,r,i=1;i<=m;i++)    {        scanf("%d%d%d",&op,&l,&r);        if(op==1)        {            scanf("%lld",&x);            change(1,1,n,l,r,x);        }        else            printf("%lld\n",query(1,1,n,l,r));    }    return 0;}

2018.10.13

picks loves segment tree I