BZOJ 4826: [Hnoi2017]影魔 单调栈 主席树

时间:2021-08-12 20:27:38

https://www.lydsy.com/JudgeOnline/problem.php?id=4826

BZOJ 4826: [Hnoi2017]影魔 单调栈 主席树

年少不知空间贵,相顾mle空流泪。

和上一道主席树求的东西差不多,求两种对

1. max(a[(i,j)])<min(a[i],a[j]),[i,j]这一对贡献p1.

2. max(a[(i,j)])在a[i],a[j]之间,[i,j]这一对贡献p2.

第一种和bzoj3956那道一样,但是因为是排列所以没必要去重了。

第二种同样是单调栈求lp,rp,每个位置的lp分别和[ i+1 , rp-1 ], rp分别和[ lp+1 , i-1 ], 构成了贡献p2的数对。

因此贡献为p2的数对要区间修改,因为方便(不用downdata)(其实是因为我抄的代码就是标记永久化)所以写了标记永久化,标记永久化真的挺好用的嘻嘻。

感觉写了这道题终于有点摸到主席树的门路了,其实就是找和维护两个区间进行限制的值(找和维护一个二维块),一层线段树一层前缀和(只是有传递性不一定是实际意义的和)。

 #include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
#define LL long long
const int maxn=;
LL n,m,p1,p2;
LL a[maxn]={},sta[maxn]={},tail=;
LL lp[maxn]={},rp[maxn]={},rt[maxn]={};
LL lc[maxn*]={},rc[maxn*]={},siz[maxn*]={},ad[maxn*]={},tot=;
struct nod{
LL x,l,r,v;
}e[maxn*];
LL cnt=;
bool mcmp(nod aa,nod bb){ return aa.x<bb.x; }
LL read(){
LL w=,f=;char ch=getchar();
while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
while(ch>=''&&ch<=''){w=w*+ch-'';ch=getchar();}
return w*f;
}
void build(LL &x,LL y,LL l,LL r,LL z,LL zl,LL zr){
x=++tot;lc[x]=lc[y];rc[x]=rc[y];ad[x]=ad[y];siz[x]=siz[y]+z*(zr-zl+);
if(zl==l&&r==zr){ad[x]+=z;return;}
LL mid=(l+r)/;
if(zr<=mid)build(lc[x],lc[y],l,mid,z,zl,zr);
else if(zl>mid) build(rc[x],rc[y],mid+,r,z,zl,zr);
else {
build(lc[x],lc[y],l,mid,z,zl,mid);
build(rc[x],rc[y],mid+,r,z,mid+,zr);
}
}
LL getsum(LL x,LL y,LL l,LL r,LL zl,LL zr){
if(zl==l&&r==zr)return siz[y]-siz[x];
LL mid=(l+r)/,ans=(ad[y]-ad[x])*(zr-zl+);
if(zr<=mid)return ans+getsum(lc[x],lc[y],l,mid,zl,zr);
else if(zl>mid) return ans+getsum(rc[x],rc[y],mid+,r,zl,zr);
else {
return ans+getsum(lc[x],lc[y],l,mid,zl,mid)+getsum(rc[x],rc[y],mid+,r,mid+,zr);
}
}
//以上主席树
inline void fir(){
LL i;
a[]=a[n+]=(<<);
for(i=;i<=n;++i){
while(a[sta[tail]]<=a[i])tail--;
lp[i]=sta[tail];sta[++tail]=i;
}
sta[]=n+;tail=;
for(i=n;i>;--i){
while(a[sta[tail]]<=a[i])tail--;
rp[i]=sta[tail];sta[++tail]=i;
}
}
inline void init(LL x,LL l,LL r,LL v){
e[++cnt].x=x;e[cnt].l=l;e[cnt].r=r;e[cnt].v=v;
}
inline void fir2(){
LL i,j;
for(i=;i<=n;++i){
if(lp[i]!=&&rp[i]!=n+)init(lp[i],rp[i],rp[i],p1);
if(i<n)init(i,i+,i+,p1);
if(lp[i]!=&&rp[i]-i>)init(lp[i],i+,rp[i]-,p2);
if(rp[i]!=n+&&i-lp[i]>)init(rp[i],lp[i]+,i-,p2);
}
sort(e+,e+cnt+,mcmp);
for(i=j=;i<=n;++i){
rt[i]=rt[i-];
for(;e[j].x==i&&j<=cnt;++j)build(rt[i],rt[i],,n,e[j].v,e[j].l,e[j].r);
}
}
int main(){
LL i,x,y,ans=;
n=read();m=read();p1=read();p2=read();
for(i=;i<=n;++i)a[i]=read();
fir(); fir2();
for(i=;i<=m;++i){
x=read();y=read();
if(x>y)swap(x,y);
ans=getsum(rt[x-],rt[y],,n,x,y);
printf("%lld\n",ans);
}
return ;
}