BZOJ4826 [Hnoi2017]影魔

16 篇文章 0 订阅
11 篇文章 0 订阅

对每个点i,单调栈求出左边和右边第一个大于i的位置,记为l[i]和r[i]

那么(l[i],r[i])会产生p1的贡献

左端点为l[i],右端点在[i+1,r-1]的点对都会产生p1的贡献

右端点为r[i],左端点在[l+1,i-1]的点对都会产生p2的贡献

将点对看成平面上的点,横坐标左端点纵坐标右端点,上述贡献分别对应单点加和线段加

查询就是矩形求和

用主席树维护即可

#include<iostream>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
#include<iomanip>
#include<cstdlib>
#include<cstdio>
#include<map>
#include<bitset>
#include<set>
#include<stack>
#include<vector>
#include<queue>
using namespace std;
#define MAXN 200010
#define MAXM 20000010
#define ll long long
#define eps 1e-8
#define MOD 1000000007
#define INF 1000000000
char xB[1<<15],*xS=xB,*xTT=xB;
#define getc() (xS==xTT&&(xTT=(xS=xB)+fread(xB,1,1<<15,stdin),xS==xTT)?0:*xS++)
#define isd(c) (c>='0'&&c<='9')
int read(){
	char xchh;
	int xaa;
    while(xchh=getc(),!isd(xchh));(xaa=xchh-'0');
    while(xchh=getc(),isd(xchh))xaa=xaa*10+xchh-'0';return xaa;
}
struct vec{
	int fro;
	int l;
	int r;
	int v;
};
vec mp[MAXN*4];
int tai[MAXN],cnt;
int n,m,p1,p2;
int a[MAXN];
ll v[MAXM],ch[MAXM];
int son[MAXM][2];
int st[MAXN],tp;
int L[MAXN],R[MAXN];
int rtl[MAXN],rtr[MAXN];
int tot;
inline void be(int x,int l,int r,int v){
	mp[++cnt].fro=tai[x];
	tai[x]=cnt;
	mp[cnt].l=l;
	mp[cnt].r=r;
	mp[cnt].v=v;
}
void change(int &x,int xx,int y,int z,int l,int r,int cv){
	if(l>r){
		return ;
	}
	x=++tot;
	v[x]=v[xx];
	ch[x]=ch[xx];
	memcpy(son[x],son[xx],sizeof(son[x]));
	v[x]+=(ll)(r-l+1)*cv;
	if(y==l&&z==r){
		ch[x]+=cv;
		return ;
	}
	int mid=y+z>>1;
	if(r<=mid){
		change(son[x][0],son[xx][0],y,mid,l,r,cv);
	}else if(l>mid){
		change(son[x][1],son[xx][1],mid+1,z,l,r,cv);
	}else{
		change(son[x][0],son[xx][0],y,mid,l,mid,cv);
		change(son[x][1],son[xx][1],mid+1,z,mid+1,r,cv);
	}
}
ll ask(int x,int y,int z,int l,int r){
	if(!x){
		return 0;
	}
	if(y==l&&z==r){
		return v[x];
	}
	ll re=ch[x]*(r-l+1);
	int mid=y+z>>1;
	if(r<=mid){
		return re+ask(son[x][0],y,mid,l,r);
	}else if(l>mid){
		return re+ask(son[x][1],mid+1,z,l,r);
	}else{
		return re+ask(son[x][0],y,mid,l,mid)+ask(son[x][1],mid+1,z,mid+1,r);
	}
}
int main(){
	int i,x,y;
	n=read();
	m=read();
	p1=read();
	p2=read();
	for(i=1;i<=n;i++){
		a[i]=read();
	}
	for(i=1;i<=n;i++){
		while(tp&&a[st[tp]]<a[i]){
			tp--;
		}
		L[i]=st[tp]+1;
		st[++tp]=i;
	}
	st[tp=0]=n+1;
	for(i=n;i;i--){
		while(tp&&a[st[tp]]<a[i]){
			tp--;
		}
		R[i]=st[tp]-1;
		st[++tp]=i;
	}
	for(i=1;i<=n;i++){
		be(L[i]-1,R[i]+1,R[i]+1,p1);
		be(L[i]-1,i+1,R[i],p2);
		be(R[i]+1,L[i],i-1,p2);
	}
	for(x=1;x<=n;x++){
		rtl[x]=rtl[x-1];
		for(i=tai[x];i;i=mp[i].fro){
			if(mp[i].l>x){
				change(rtl[x],rtl[x],1,n,max(1,mp[i].l),min(n,mp[i].r),mp[i].v);
			}
		}
		if(x!=n){
			change(rtl[x],rtl[x],1,n,x+1,x+1,p1);
		}
	}
	for(x=n;x;x--){
		rtr[x]=rtr[x+1];
		for(i=tai[x];i;i=mp[i].fro){
			if(mp[i].r<x){
				change(rtr[x],rtr[x],1,n,max(1,mp[i].l),min(n,mp[i].r),mp[i].v);
			}
		}
	}
	while(m--){
		x=read();
		y=read();
		ll ans=0;
		ans+=ask(rtl[y],1,n,x,y);
		ans-=ask(rtl[x-1],1,n,x,y);
		ans+=ask(rtr[x],1,n,x,y);
		ans-=ask(rtr[y+1],1,n,x,y);
		printf("%lld\n",ans);
	}
	return 0;
}

/*

*/


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值