[BZOJ4826][主席树][单调栈]HNOI2017:影魔

5 篇文章 0 订阅
3 篇文章 0 订阅

BZOJ4826

又来。。。
显然考虑枚举作为区间最大值的位置 i i i,然后我们需要知道 L [ i ] L[i] L[i] R [ i ] R[i] R[i]表示 i i i左右两边第一个大于它的数的位置,用单调栈维护
考虑一个 i i i对答案的影响,显然就是以下情况:

  1. L [ i ] L[i] L[i] R [ i ] R[i] R[i]组成 p 1 p1 p1的影响
  2. i i i i + 1 i+1 i+1组成 p 1 p1 p1的影响
  3. ( L [ i ] , i ) (L[i],i) (L[i],i)的数和 R [ i ] R[i] R[i]组成 p 2 p2 p2的影响
  4. ( i , R [ i ] ) (i,R[i]) (i,R[i])的数和 L [ i ] L[i] L[i]组成 p 2 p2 p2的影响

其中圆括号表示开区间

所以这样一来,每个位置及其 L , R L,R L,R的贡献映射到二维平面上就是一个点或者一条线段,而一个询问就是询问一个矩形内的部分,直接离线+扫描线即可

当然也可以强制在线,就得用主席树了

Code(主席树):

#include<bits/stdc++.h>
#define ll long long
#define pb push_back
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f; 
}
const int N=2e5+5;
int rt[N];
namespace President_tree{
	struct seg{int l,r,add;ll val;}tr[N*80];
	int cnt=0;
	#define ls(k) tr[k].l
	#define rs(k) tr[k].r
	#define mid (l+r>>1)
	void build(int &rt,int l,int r){
		rt=++cnt;
		if(l==r) return;
		build(ls(rt),l,mid);build(rs(rt),mid+1,r);
	}
	void ins(int &rt1,int rt2,int l,int r,int ql,int qr,int x){
		rt1=++cnt;tr[rt1]=tr[rt2];tr[rt1].val+=1ll*(qr-ql+1)*x;
		if(ql==l && r==qr){
			tr[rt1].add+=x;
			return;
		}
		if(qr<=mid) ins(ls(rt1),ls(rt2),l,mid,ql,qr,x);
		else if(ql>mid) ins(rs(rt1),rs(rt2),mid+1,r,ql,qr,x);
		else ins(ls(rt1),ls(rt2),l,mid,ql,mid,x),ins(rs(rt1),rs(rt2),mid+1,r,mid+1,qr,x);
	}
	ll query(int rt,int l,int r,int ql,int qr){
		if(l==ql && r==qr) return tr[rt].val;
		ll sum=1ll*(qr-ql+1)*tr[rt].add;
		if(qr<=mid) return sum+query(ls(rt),l,mid,ql,qr);
		else if(ql>mid) return sum+query(rs(rt),mid+1,r,ql,qr);
		else return sum+query(ls(rt),l,mid,ql,mid)+query(rs(rt),mid+1,r,mid+1,qr);
	}
}
using namespace President_tree;
struct sqr{
	int l,r,val;
	sqr(){}
	sqr(int _l,int _r,int _val):l(_l),r(_r),val(_val){}
};
vector<sqr>v[N];
int n,a[N],sta[N],top=0,L[N],R[N];
int main(){
	n=read();int q=read(),p1=read(),p2=read();
	for(int i=1;i<=n;i++) a[i]=read();
	for(int i=1;i<=n;i++){
		while(top && a[sta[top]]<a[i]) --top;
		L[i]=sta[top];
		sta[++top]=i;
	}
	top=0;sta[0]=n+1;
	for(int i=n;i;i--){
		while(top && a[sta[top]]<a[i]) --top;
		R[i]=sta[top];
		sta[++top]=i;
	}
	for(int i=1;i<=n;i++){
		if(i<n) v[i].pb(sqr(i+1,i+1,p1));
		if(L[i] && R[i]<=n) v[L[i]].pb(sqr(R[i],R[i],p1));
		if(R[i]<=n && L[i]+2<=i) v[R[i]].pb(sqr(L[i]+1,i-1,p2));
		if(L[i] && i<=R[i]-2) v[L[i]].pb(sqr(i+1,R[i]-1,p2));
	}
	build(rt[0],1,n);
	for(int i=1;i<=n;i++){
		rt[i]=rt[i-1];
		for(int j=0;j<v[i].size();j++)
			ins(rt[i],rt[i],1,n,v[i][j].l,v[i][j].r,v[i][j].val); 
	}
	for(int i=1;i<=q;i++){
		int l=read(),r=read();
		cout<<query(rt[r],1,n,l,r)-query(rt[l-1],1,n,l,r)<<"\n";
	}
	return 0;
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值