【WC2016模拟】最假女选手

Description

维护一个序列，资瓷：
1：区间加
2：区间取max
3：区间取min
4：区间求和
5：区间求max
6：区间求min

n<=5*1e5

Solution

最近正在搞数据结构专题，于是找了一道吉丽线段树练手题
维护区间和，最大值，次大值，最大值出现次数，最小值，次小值，次小值出现次数即可
细节很多，需要很细心的实现

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;

typedef long long ll;

int read() {
    char ch;int sig=1;
    for(ch=getchar();ch<'0'||ch>'9';ch=getchar()) if (ch=='-') sig=-1;
    int x=ch-'0';
    for(ch=getchar();ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
    return x*sig;
}

void write(ll x) {
    if (!x) {puts("0");return;}
    if (x<0) {x=-x;putchar('-');}
    char ch[20];int tot=0;
    for(;x;x/=10) ch[++tot]=x%10+'0';
    fd(i,tot,1) putchar(ch[i]);
    puts("");
}

const int N=5*1e5+5,inf=0x7fffffff;

struct Segment_Tree{
    int mx,smx,tmx,mn,smn,tmn;
    int cmx,cmn,lazy;
    ll sum;
}tr[N<<2];

int n,m,a[N],opt,ql,qr,x;

#define max(a,b) ((a>b)?(a):(b))
#define min(a,b) ((a<b)?(a):(b))
#define ls (v<<1)
#define rs (v<<1|1)
#define mid (l+r>>1)

void updata(int v) {
    tr[v].sum=tr[ls].sum+tr[rs].sum;

    if (tr[ls].mx==tr[rs].mx) {
        tr[v].tmx=tr[ls].tmx+tr[rs].tmx;
        tr[v].smx=max(tr[ls].smx,tr[rs].smx);
        tr[v].mx=tr[ls].mx;
    } else if (tr[ls].mx>tr[rs].mx) {
        tr[v].tmx=tr[ls].tmx;
        tr[v].smx=max(tr[ls].smx,tr[rs].mx);
        tr[v].mx=tr[ls].mx;
    } else {
        tr[v].tmx=tr[rs].tmx;
        tr[v].smx=max(tr[ls].mx,tr[rs].smx);
        tr[v].mx=tr[rs].mx;
    }

    if (tr[ls].mn==tr[rs].mn) {
        tr[v].tmn=tr[ls].tmn+tr[rs].tmn;
        tr[v].smn=min(tr[ls].smn,tr[rs].smn);
        tr[v].mn=tr[ls].mn;
    } else if (tr[ls].mn<tr[rs].mn) {
        tr[v].tmn=tr[ls].tmn;
        tr[v].smn=min(tr[ls].smn,tr[rs].mn);
        tr[v].mn=tr[ls].mn;
    } else {
        tr[v].tmn=tr[rs].tmn;
        tr[v].smn=min(tr[ls].mn,tr[rs].smn);
        tr[v].mn=tr[rs].mn;
    }
}

void build(int v,int l,int r) {
    tr[v].cmn=inf;tr[v].cmx=-inf;
    if (l==r) {
        tr[v].mx=tr[v].mn=a[l];
        tr[v].smx=-inf;tr[v].smn=inf;
        tr[v].tmx=tr[v].tmn=1;
        tr[v].sum=a[l];
        return;
    }
    build(ls,l,mid);build(rs,mid+1,r);
    updata(v);
}

#define len (r-l+1)

void Add(int &x,int y) {
    if (x==inf||x==-inf) return;
    x+=y;
}

void get_Add(int v,int l,int r,int x) {
    tr[v].sum+=(ll)len*x;
    tr[v].mx+=x;tr[v].mn+=x;
    Add(tr[v].smx,x);Add(tr[v].smn,x);
    Add(tr[v].cmx,x);Add(tr[v].cmn,x);
    tr[v].lazy+=x;
}

bool get_Min(int v,int l,int r,int x) {
    if (tr[v].mx<=x) return 1;
    if (tr[v].smx<x) {
        tr[v].sum-=(ll)tr[v].tmx*(tr[v].mx-x);tr[v].mx=x;
        tr[v].mn=min(tr[v].mn,x);tr[v].smn=min(tr[v].smn,x);
        if (tr[v].mn==x) {
            tr[v].smx=-inf;tr[v].smn=inf;
            tr[v].tmx=tr[v].tmn=len;
        }
        tr[v].cmn=min(tr[v].cmn,x);
        tr[v].cmx=min(tr[v].cmn,tr[v].cmx);
        return 1;
    }
    return 0;
}   

bool get_Max(int v,int l,int r,int x) {
    if (tr[v].mn>=x) return 1;
    if (tr[v].smn>x) {
        tr[v].sum+=(ll)tr[v].tmn*(x-tr[v].mn);tr[v].mn=x;
        tr[v].mx=max(tr[v].mx,x);tr[v].smx=max(tr[v].smx,x);
        if (tr[v].mx==x) {
            tr[v].smx=-inf;tr[v].smn=inf;
            tr[v].tmx=tr[v].tmn=len;
        }
        tr[v].cmx=max(tr[v].cmx,x);
        tr[v].cmn=max(tr[v].cmx,tr[v].cmn);
        return 1;
    }
    return 0;
}

void down(int v,int l,int r) {
    if (l==r) return;

    if (tr[v].lazy) {
        get_Add(ls,l,mid,tr[v].lazy);
        get_Add(rs,mid+1,r,tr[v].lazy);
        tr[v].lazy=0;
    }

    if (tr[v].cmn!=inf) {
        get_Min(ls,l,mid,tr[v].cmn);
        get_Min(rs,mid+1,r,tr[v].cmn);
        tr[v].cmn=inf;
    }

    if (tr[v].cmx!=-inf) {
        get_Max(ls,l,mid,tr[v].cmx);
        get_Max(rs,mid+1,r,tr[v].cmx);
        tr[v].cmx=-inf;
    }
}

void modify1(int v,int l,int r) {
    if (ql<=l&&r<=qr) {get_Add(v,l,r,x);return;}
    down(v,l,r);
    if (ql<=mid) modify1(ls,l,mid);
    if (qr>mid) modify1(rs,mid+1,r);
    updata(v);
}

void modify2(int v,int l,int r) {
    if (ql<=l&&r<=qr&&get_Max(v,l,r,x)) return;
    down(v,l,r);
    if (ql<=mid) modify2(ls,l,mid);
    if (qr>mid) modify2(rs,mid+1,r);
    updata(v);
}

void modify3(int v,int l,int r) {
    if (ql<=l&&r<=qr&&get_Min(v,l,r,x)) return;
    down(v,l,r);
    if (ql<=mid) modify3(ls,l,mid);
    if (qr>mid) modify3(rs,mid+1,r);
    updata(v);
}

ll query1(int v,int l,int r) {
    if (ql<=l&&r<=qr) return tr[v].sum;
    down(v,l,r);ll sum=0;
    if (ql<=mid) sum+=query1(ls,l,mid);
    if (qr>mid) sum+=query1(rs,mid+1,r);
    return sum;
}

int query2(int v,int l,int r) {
    if (ql<=l&&r<=qr) return tr[v].mx;
    down(v,l,r);int Mx=-inf;
    if (ql<=mid) {int now=query2(ls,l,mid);Mx=max(Mx,now);}
    if (qr>mid) {int now=query2(rs,mid+1,r);Mx=max(Mx,now);}
    return Mx;
}

int query3(int v,int l,int r) {
    if (ql<=l&&r<=qr) return tr[v].mn;
    down(v,l,r);int Mn=inf;
    if (ql<=mid) {int now=query3(ls,l,mid);Mn=min(Mn,now);}
    if (qr>mid) {int now=query3(rs,mid+1,r);Mn=min(Mn,now);}
    return Mn;
}

int main() {
    freopen("wyywyy.in","r",stdin);
    freopen("wyywyy.out","w",stdout);
    n=read();
    fo(i,1,n) a[i]=read();
    build(1,1,n);
    for(m=read();m;m--) {
        opt=read();ql=read();qr=read();
        if (opt<=3) x=read();
        if (opt==1) modify1(1,1,n);
        if (opt==2) modify2(1,1,n);
        if (opt==3) modify3(1,1,n);
        if (opt==4) write(query1(1,1,n));
        if (opt==5) write(query2(1,1,n));
        if (opt==6) write(query3(1,1,n));
    }
    return 0;
}