链接:https://www.luogu.org/problemnew/show/P3246
前几天xj比赛刚补了一道比这道还毒瘤的题,询问什么区间子序列前缀最大值和。。。那道题的trick学会这题就很简单了。首先维护一个st表记录区间最小值的位置,以及一个函数f,f(i)表示所有以i为开始的区间的最小值之和,同理维护g,g(i)表示所有以i为结尾的区间最小值之和。(ps:维护这个东西之前我傻乎乎的写了线段树,其实会路径压缩的小技巧一行就没了。。)考虑查询l,r时,首先找到查询区间最小值的位置记为ps,然后对于左端点在ps左右端点在ps右的就可以直接算出贡献,现在就是要考虑左右端点在ps一侧的贡献。这里举求左区间为例,考虑我们刚才求出的函数f,有可能有贡献的就是f(l~ps-1)(左端点位置),再考虑多算的贡献就是右端点在ps及ps右边的,而因为ps恰好是整个询问区间的最小值,那么左端点在l到ps之间,右端点在ps右边的区间答案一定是小于ps上的值,所以对于任取i属于(l,ps-1),它多算的贡献就是f(ps),所以再维护一个f的前缀和,就能O(1)查出ps左侧答案,右侧同理。
代码:
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1e5+100;
void rd(int &x)
{
char c=getchar();x=0;bool f=0;
while(!isdigit(c))f|=(c=='-'),c=getchar();
while(isdigit(c))x=x*10+c-48,c=getchar();
if(f)x=-x;
}
int n,q,a[N],st[N][20],lg[N],pre[N],nxt[N];
ll f[N],g[N],sf[N],sg[N];
int qry(int l,int r)
{
int tp=lg[r-l+1];
if(a[st[l][tp]]<=a[st[r-(1<<tp)+1][tp]])return st[l][tp];
return st[r-(1<<tp)+1][tp];
}
int main()
{
rd(n),rd(q);
for(int i=1;i<=n;i++)rd(a[i]),pre[i]=i-1,nxt[i]=i+1;
lg[1]=0;
for(int i=2;i<=n;i++)lg[i]=lg[i>>1]+1;
for(int i=1;i<=n;i++)st[i][0]=i;
for(int i=1,l=1;i<=20;i++,l<<=1)
{
for(int j=1;j+l<=n;j++)
{
if(a[st[j][i-1]]<=a[st[j+l][i-1]])st[j][i]=st[j][i-1];
else st[j][i]=st[j+l][i-1];
}
}
for(int i=n;i>=1;i--)
{
while(nxt[i]!=n+1&&a[i]<a[nxt[i]])nxt[i]=nxt[nxt[i]];
f[i]=f[nxt[i]]+1LL*(nxt[i]-i)*a[i];
sf[i]=sf[i+1]+f[i];
}
for(int i=1;i<=n;i++)
{
while(pre[i]&&a[i]<a[pre[i]])pre[i]=pre[pre[i]];
g[i]=g[pre[i]]+1LL*(i-pre[i])*a[i];
sg[i]=sg[i-1]+g[i];
}
int l,r,ps;ll res;
while(q--)
{
rd(l),rd(r),ps=qry(l,r);
res=1LL*a[ps]*(ps-l+1)*(r-ps+1);
res+=(sf[l]-sf[ps]-f[ps]*(ps-l))+(sg[r]-sg[ps]-g[ps]*(r-ps));
printf("%lld\n",res);
}
}