Description
影魔,奈文摩尔,据说有着一个诗人的灵魂。事实上,他吞噬的诗人灵魂早已成千上万。千百年来,他收集了各式各样的灵魂,包括诗人、牧师、帝王、乞丐、奴隶、罪人,当然,还有英雄。每一个灵魂,都有着自己的战斗力,而影魔,靠这些战斗力提升自己的攻击。奈文摩尔有 n 个灵魂,他们在影魔宽广的体内可以排成一排,从左至右标号 1 到 n。
第 i个灵魂的战斗力为 k[i],灵魂们以点对的形式为影魔提供攻击力,对于灵魂对 i,j ( i < j ) 来说,若不存在 k[s] ( i < s < j ) 大于 k[i]或者 k[j],则会为影魔提供 p1 的攻击力(可理解为:当 j=i+1 时,因为不存在满足 i < s < j 的 s,从而 k[s]不存在,这时提供 p1 的攻击力;当 j>i+1 时,若max{k[s]| i < s < j}< =min{k[i],k[j]} , 则 提 供 p1 的 攻击 力 ); 另 一 种 情 况 , 令 c 为k[i+1],k[i+2],k[i+3]……k[j-1]的最大值,若 c 满足:k[i] < c < k[j],或者 k[j] < c< k[i],则会为影魔提供 p2 的攻击力,当这样的 c 不存在时,自然不会提供这 p2 的攻击力;其他情况的点对,均不会为影魔提供攻击力。影魔的挚友噬魂鬼在一天造访影魔体内时被这些灵魂吸引住了,他想知道,对于任意一段区间[a,b],1< =a< b< =n,位于这些区间中的灵魂对会为影魔提供多少攻击力,即考虑 所有满足a< =i< j< =b 的灵魂对 i,j 提供的攻击力之和。顺带一提,灵魂的战斗力组成一个 1 到 n 的列:k[1],k[2],…,k[n]。
Input
第一行 n,m,p1,p2
第二行 n 个数:k[1],k[2],…,k[n]
接下来 m 行,每行两个数 a,b,表示询问区间[a,b]中的灵魂对会为影魔提供多少攻击力。
1 <= n,m <= 200000;1 <= p1,p2 <= 1000
Output
共输出 m 行,每行一个答案,依次对应 m 个询问。
Sample Input
10 5 2 3
7 9 5 1 3 10 6 8 2 4
1 7
1 9
1 3
5 9
1 5
Sample Output
30
39
4
13
16
I think
首先我们发现k[i]是一个排列,也就是该正项数列中没有重复元素。我们找出对于每一个k[i],
那么
1)区间[L[i],r[i]] 有贡献p1
2)区间[L[i],j],j∈[i+1,R[i]-1] 有贡献p2
3)区间[j,R[i]],j∈[L[i]+1,i-1] 有贡献p2
于是我们可以把区间看作一个xy坐标轴上的点,对于每一个被询问区间Q(a,b),其答案是以(a,b)与(n,1)为对角线的矩形内的点权和。一段连续的点可以用扫描线+线段树更新,一边更新一边更新包含该段区间的询问对应答案。扫描线沿xy轴分别扫一遍就ok。
代码中,由于有p1贡献的点和横坐标一定为L[i]的点同时存储在LX中,所以空间开两倍。
用树状数组实现更快,代码也更简洁。
Code
#include<vector>
#include<cstdio>
#include<algorithm>
#define s second
using namespace std;
#define int long long
const int sm = 2e5+10;
const int sn = sm << 1;
typedef long long ll;
typedef pair<int,int> pii;
int n,m,p1,p2,t1,t2;
int k[sm],L[sm],R[sm];//注意k为一个排列元素不重复
ll ans[sm],val[sm<<2],mk[sm<<2];
vector<int> xx[sm],yy[sm];//存询问
vector<int> MX[sm],MY[sm];
pii st[sm],t;
struct line {
int a,b1,b2,c;
}LX[sm << 1],LY[sm];
struct query {
int l,r,id;
}Q[sm];
char ch;
void read(int &x) {
x=0,ch=getchar();
while(ch>'9'||ch<'0')ch=getchar();
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
}
void calc() {
//利用单调栈求L[i],R[i]
//L[i]=max{k[j]},j<i&&k[j]>k[i]
//R[i]=min{k[j]},j>i&&k[j]>k[i]
int tp;
st[tp=1] = make_pair(k[1],1);
for(int i=2;i<=n;++i) {
t=make_pair(k[i],i);
while(tp&&st[tp]<t) --tp;
if(!tp) L[i]=0;
else L[i]=st[tp].s;
st[++tp]=t;
}
R[n]=n+1;
st[tp=1] = make_pair(k[n],n);
for(int i=n-1;i>=1;--i) {
t=make_pair(k[i],i);
while(tp&&st[tp]<t) --tp;
if(!tp) R[i]=n+1;
else R[i]=st[tp].s;
st[++tp]=t;
}
}
void pushdown(int k,int l,int r) {
int ls=k<<1,rs=k<<1|1,m=(l+r)>>1;
mk[rs]+=mk[k],val[rs]+=mk[k]*(r-m);
mk[ls]+=mk[k],val[ls]+=mk[k]*(m-l+1);
mk[k]=0;
}
void add(int k,int l,int r,line L) {
if(L.b1<=l&&r<=L.b2) {
val[k]+=L.c*(r-l+1); //printf("*%d %d %d\n",L.c,l,r);
mk[k]+=L.c;return;
}
if(l==r)return;
if(mk[k])pushdown(k,l,r);
int m=(l+r)>>1;
if(L.b1<=m) add(k<<1,l,m,L);
if(L.b2> m) add(k<<1|1,m+1,r,L);
val[k]=val[k<<1]+val[k<<1|1];
}
int query(int k,int l,int r,int ll,int rr) {
if( l > rr || r < ll )return 0;
if(ll<=l&&r<=rr) return val[k];
if(mk[k]) pushdown(k,l,r);
int m=(l+r)>>1,tmp=0;
if(ll<=m) tmp+=query(k<<1,l,m,ll,rr);
if(rr> m) tmp+=query(k<<1|1,m+1,r,ll,rr);
return tmp;
}
void build(int k,int l,int r) {
val[k]=mk[k]=0;
if(l==r)return;
int m=(l+r)>>1;
build(k<<1,l,m);
build(k<<1|1,m+1,r);
}
void work() {
int g;
for(int i=n;i>=1;--i) { //扫描线沿x轴方向由右至左扫
for(int j=0;j<MX[i].size();++j)
g=MX[i][j],add(1,1,n,LX[g]);
for(int j=0;j<xx[i].size();++j)
g=xx[i][j],ans[g]+=query(1,1,n,1,Q[g].r);// ind=1 取[1,r]
}
build(1,1,n);
for(int i=1;i<=n;++i) {
for(int j=0;j<MY[i].size();++j)
g=MY[i][j],add(1,1,n,LY[g]);
for(int j=0;j<yy[i].size();++j)
g=yy[i][j],ans[g]+=query(1,1,n,Q[g].l,n);// ind=0 取[l,n]
}
}
main() {
read(n),read(m);
read(p1),read(p2);
for(int i=1;i<=n;++i) read(k[i]);
calc();
for(int i=1,x,y;i<=m;++i) {
read(x),read(y);
Q[i].l=x,Q[i].r=y,Q[i].id=i;
xx[x].push_back(i),yy[y].push_back(i);
//即将询问i加入以x为询问左端的询问序列xx[x]中
}
for(int i=1;i<=n;++i) {//存贡献值为p1的点
if(L[i]!=0&&R[i]!=n+1) {
LX[++t1].a=L[i], LX[t1].c=p1;
LX[t1].b1=LX[t1].b2=R[i],MX[L[i]].push_back(t1);//不要漏掉加入MX
}
if(L[i]!=0||R[i]!=n+1) {
if(i+1<=R[i]-1) {//将贡献值为p2的点分别以平行与x|y轴线段的形式存储
LX[++t1].a=L[i], LX[t1].b1=i+1, LX[t1].b2=R[i]-1;
LX[t1].c=p2, MX[L[i]].push_back(t1); //将t1存入以L[i]为a的序列中
}
if(L[i]+1<=i-1) {
LY[++t2].a=R[i], LY[t2].b1=L[i]+1, LY[t2].b2=i-1;
LY[t2].c=p2, MY[R[i]].push_back(t2);
}
}
}
work();
for(int i=1;i<=m;++i)
printf("%lld\n",ans[i]+(Q[i].r-Q[i].l)*p1);
return 0;
}
再贴一份更简洁的树状数组代码……
#include<vector>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define s second
#define lowbit(x) x&(-x)
using namespace std;
const int sm = 2e5+10;
const int sn = sm << 1;
typedef long long ll;
typedef pair<int,int> pii;
int n,m,p1,p2,t1,t2;
int k[sm],L[sm],R[sm];//注意k为一个排列元素不重复
ll sum,ans[sm],b[sm],c[sm];
vector<int> xx[sm],yy[sm];//存询问
vector<int> MX[sm],MY[sm];
pii st[sm],t;
struct line {
int a,b1,b2,c;
}LX[sm<<1],LY[sm];
struct query {
int l,r,id;
}Q[sm];
char ch;
void read(int &x) {
x=0,ch=getchar();
while(ch>'9'||ch<'0')ch=getchar();
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
}
void calc() {
//利用单调栈求L[i],R[i]
//L[i]=max{k[j]},j<i&&k[j]>k[i]
//R[i]=min{k[j]},j>i&&k[j]>k[i]
int tp;
st[tp=1] = make_pair(k[1],1);
for(int i=2;i<=n;++i) {
t=make_pair(k[i],i);
while(tp&&st[tp]<t) --tp;
if(!tp) L[i]=0;
else L[i]=st[tp].s;
st[++tp]=t;
}
R[n]=n+1;
st[tp=1] = make_pair(k[n],n);
for(int i=n-1;i>=1;--i) {
t=make_pair(k[i],i);
while(tp&&st[tp]<t) --tp;
if(!tp) R[i]=n+1;
else R[i]=st[tp].s;
st[++tp]=t;
}
}
void add(int x,int y) {
for(int i=x;i<=n;i+=lowbit(i))
b[i]+=y,c[i]+=x*y;
}
ll Query(int x) {
sum=0;
for(int i=x;i;i-=lowbit(i))
sum+=(x+1)*b[i]-c[i];
return sum;
}
void work() {
int g;
line x;
query y;
for(int i=n;i>=1;--i) { //扫描线沿x轴方向由右至左扫
for(int j=0;j<MX[i].size();++j) {
g=MX[i][j]; x=LX[g];
add(x.b1,x.c),add(x.b2+1,-x.c);//add(1,1,n,LX[g]);
}
for(int j=0;j<xx[i].size();++j) {
g=xx[i][j]; y=Q[g];
ans[g]+=Query(y.r)-Query(y.l-1);//query(1,1,n,1,Q[g].r);// ind=1 取[1,r]
}
}
memset(b,0,sizeof(b));
memset(c,0,sizeof(c));
for(int i=1;i<=n;++i) {
for(int j=0;j<MY[i].size();++j) {
g=MY[i][j]; x=LY[g];
add(x.b1,x.c),add(x.b2+1,-x.c);
}
for(int j=0;j<yy[i].size();++j) {
g=yy[i][j]; y=Q[g];
ans[g]+=Query(y.r)-Query(y.l-1);// ind=0 取[l,n]
}
}
}
int main() {
read(n),read(m);
read(p1),read(p2);
for(int i=1;i<=n;++i) read(k[i]);
calc();
for(int i=1,x,y;i<=m;++i) {
read(x),read(y);
Q[i].l=x,Q[i].r=y,Q[i].id=i;
xx[x].push_back(i),yy[y].push_back(i);
//即将询问i加入以x为询问左端的询问序列xx[x]中
}
for(int i=1;i<=n;++i) {//存贡献值为p1的点
if(L[i]!=0&&R[i]!=n+1) {
LX[++t1].a=L[i], LX[t1].c=p1;
LX[t1].b1=LX[t1].b2=R[i],MX[L[i]].push_back(t1);//不要漏掉加入MX
}
if(L[i]!=0||R[i]!=n+1) {
if(i+1<=R[i]-1) {//将贡献值为p2的点分别以平行与x|y轴线段的形式存储
LX[++t1].a=L[i], LX[t1].b1=i+1, LX[t1].b2=R[i]-1;
LX[t1].c=p2, MX[L[i]].push_back(t1); //将t1存入以L[i]为a的序列中
}
if(L[i]+1<=i-1) {
LY[++t2].a=R[i], LY[t2].b1=L[i]+1, LY[t2].b2=i-1;
LY[t2].c=p2, MY[R[i]].push_back(t2);
}
}
}
work();
for(int i=1;i<=m;++i)
printf("%lld\n",ans[i]+(Q[i].r-Q[i].l)*p1);
return 0;
}