「AHOI / HNOI2017」影魔

「AHOI / HNOI2017」影魔

题目描述

解决这类比较复杂的区间贡献问题关键在于找到计算的对象。

比如这道题,我们计算的对象就是区间中间的最大值。

对于点\(i\),我们找到左边第一个比他大的位置\(L\),以及右边第一个比他大的位置\(R\)。当\(L,R\)同时被询问的区间包含是,\(i\)就会贡献\(p_1\)。当固定左端点为\(L\),右端在\([i+1,R-1]\)之间的时候会贡献\(p_2\);固定右端点\(R\)是同理。还要额外加上\(i,i+1\)贡献的\(p_1\)。

具体实现就可以使用扫描线+树状数组之类的方法。

代码:

#include<bits/stdc++.h>
#define ll long long
#define N 200005 using namespace std;
inline int Get() {int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}while('0'<=ch&&ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;} int n,m;
ll p1,p2;
int a[N];
int L[N],R[N]; void pre() {
int st[N],top;
st[top=0]=0;
for(int i=1;i<=n;i++) {
while(top&&a[st[top]]<a[i]) top--;
L[i]=st[top]+1;
st[++top]=i;
}
st[top=0]=n+1;
for(int i=n;i>=1;i--) {
while(top&&a[st[top]]<a[i]) top--;
R[i]=st[top]-1;
st[++top]=i;
}
} struct query {
int l,r;
int id;
bool operator <(const query &a)const {return l<a.l;}
}q[N];
bool cmpl(const query &a,const query &b) {return a.l<b.l;}
bool cmpL(const query &a,const query &b) {return a.l>b.l;}
bool cmpR(const query &a,const query &b) {return a.r<b.r;} struct Bit {
ll tem[N];
int low(int i) {return i&(-i);}
void add(int v,int f) {for(int i=v;i<=n;i+=low(i)) tem[i]+=f;}
ll ask(int v) {
ll ans=0;
for(int i=v;i;i-=low(i)) ans+=tem[i];
return ans;
}
void Init() {memset(tem,0,sizeof(tem));}
}T,Size;
ll ans[N];
vector<int>del[N]; int main() {
n=Get(),m=Get(),p1=Get(),p2=Get();
for(int i=1;i<=n;i++) a[i]=Get();
pre(); for(int i=1;i<=m;i++) q[i].l=Get(),q[i].r=Get(),q[i].id=i;
for(int i=1;i<=m;i++) {
ans[q[i].id]+=(q[i].r-q[i].l)*p1;
} sort(q+1,q+1+m,cmpl);
for(int i=1;i<=n;i++) del[L[i]-1].push_back(R[i]+1);
for(int i=1;i<=n;i++) T.add(R[i]+1,1); int tag=0;
for(int i=1;i<=m;i++) {
while(tag<q[i].l) {
while(del[tag].size()) {
T.add(del[tag].back(),-1);
del[tag].pop_back();
}
tag++;
}
ans[q[i].id]+=T.ask(q[i].r)*p1;
}
for(int i=1;i<=n+1;i++) del[i].clear();
T.Init(); tag=1;
for(int i=1;i<=n;i++) del[R[i]].push_back(i);
sort(q+1,q+1+m,cmpR);
for(int i=1;i<=m;i++) {
while(tag<=q[i].r) {
T.add(n-L[tag]+2,-tag);
Size.add(n-L[tag]+2,1);
while(del[tag].size()) {
int x=del[tag].back();
T.add(n-L[x]+2,x);
Size.add(n-L[x]+2,-1);
T.add(n-L[x]+2,R[x]-x);
del[tag].pop_back();
}
tag++;
}
ans[q[i].id]+=p2*(Size.ask(n-q[i].l+1)*q[i].r+T.ask(n-q[i].l+1));
} for(int i=0;i<=n+1;i++) del[i].clear();
T.Init(),Size.Init();
sort(q+1,q+1+m,cmpL);
tag=n;
for(int i=1;i<=n;i++) del[L[i]].push_back(i);
for(int i=1;i<=m;i++) {
while(tag>=q[i].l) {
T.add(R[tag]+1,tag);
Size.add(R[tag]+1,1);
while(del[tag].size()) {
int x=del[tag].back();
T.add(R[x]+1,-x);
Size.add(R[x]+1,-1);
T.add(R[x]+1,x-L[x]);
del[tag].pop_back();
}
tag--;
}
ans[q[i].id]+=p2*(T.ask(q[i].r)-Size.ask(q[i].r)*q[i].l);
}
for(int i=1;i<=m;i++) cout<<ans[i]<<"\n"; return 0;
}
上一篇:vue项目全局引入vue-awesome-swiper插件做出轮播效果


下一篇:Win7中怎么不经确认直接删除文件至回收站