AC日记——[Hnoi2017]影魔 bzoj 4826

4826

思路：

　　主席树矩阵加减+单调栈预处理；

代码：

#include <bits/stdc++.h>
using namespace std;
#define maxn 200005
#define ll long long
#define maxtree maxn*30
class PTreeType
{
    private:
        int ch[maxtree][],root[maxn],tot,head[maxn],li[maxn<<],ri[maxn<<],E[maxn<<],cnt,val_[maxn<<];
        ll val[maxtree],tag[maxtree];
    public:
        void add(int &now,int pre,int l,int r,int tl,int tr,ll x)
        {
            now=++tot,val[now]=val[pre],tag[now]=tag[pre];
            val[now]+=((r<tr?r:tr)-(l>tl?l:tl)+)*x;
            ch[now][]=ch[pre][],ch[now][]=ch[pre][];
            if(l>=tl&&r<=tr){tag[now]+=x;return;}int mid=l+r>>;
            if(tl<=mid) add(ch[now][],ch[pre][],l,mid,tl,tr,x);
            if(tr>mid) add(ch[now][],ch[pre][],mid+,r,tl,tr,x);
        }
        ll query(int now,int pre,int l,int r,int tl,int tr)
        {
            if(l>=tl&&r<=tr) return val[now]-val[pre];
            int mid=l+r>>;ll res=((r<tr?r:tr)-(l>tl?l:tl)+)*(tag[now]-tag[pre]);
            if(tl<=mid) res+=query(ch[now][],ch[pre][],l,mid,tl,tr);
            if(tr>mid) res+=query(ch[now][],ch[pre][],mid+,r,tl,tr);
            return res;
        }
        void add(int l,int r,int size,int x,int to)
        {
            add(root[to],root[to],,size,l,r,x);
        }
        ll query(int l,int r,int size,int now,int pre)
        {
            return query(root[now],root[pre],,size,l,r);
        }
        void operation_add(int to,int l,int r,int pi)
        {
            E[++cnt]=head[to],li[cnt]=l,ri[cnt]=r,val_[cnt]=pi,head[to]=cnt;
        }
        void makeit(int size)
        {
            for(int i=;i<=size;i++)
            {
                root[i]=root[i-];
                for(int v=head[i];v;v=E[v])
                {
                    add(li[v],ri[v],size,val_[v],i);
                }
            }
        }
};
class PTreeType xtree,ytree;
struct NodeType {
    int l,r;
};
struct NodeType ai[maxn];
int n,m,p1,p2,ki[maxn],sta[maxn];
inline void in(int &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'') Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}
inline void in(ll &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'')Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}
int main()
{
    in(n),in(m),in(p1),in(p2);int top=,op,l,r;ll ans;
    for(int i=;i<=n;i++)
    {
        in(ki[i]);
        while(top&&ki[sta[top]]<ki[i]) ai[sta[top]].r=i,top--;
        ai[i].l=sta[top],sta[++top]=i;
    }
    while(top) ai[sta[top]].r=n+,top--;
    for(int i=;i<=n;i++)
    {
        if(ai[i].l>&&ai[i].r<=n) xtree.operation_add(ai[i].l,ai[i].r,ai[i].r,p1);
        if(i+<=ai[i].r-) xtree.operation_add(ai[i].l,i+,ai[i].r-,p2);
        if(ai[i].l+<=i-) ytree.operation_add(ai[i].r,ai[i].l+,i-,p2);
    }
    xtree.makeit(n),ytree.makeit(n);
    while(m--)
    {
        in(l),in(r),ans=;
        ans+=xtree.query(l,r,n,r,l-);
        ans+=ytree.query(l,r,n,r,l-);
        ans+=(ll)p1*(r-l);
        printf("%lld\n",ans);
    }
    return ;
}