天天看点

[HNOI2017]影魔题目大意题目分析代码实现

题目大意

给定一个长度为 n 的排列,对于一个区间,如果两个端点分别是这个区间的最大值和次大值,有p1的贡献,否则如果其中一个端点是这个区间的最大值,有 p2 的贡献。

现在给你 q 个询问,每次询问一个区间[L,R],问所有被区间完全包含的区间的贡献和。

1≤n,q≤2×105,1≤p1,p2≤103

题目分析

首先,我们可以考虑把 p1 这种贡献先算在 p2 里面,具体是怎么样呢?

我们将贡献 p2 重定义为:一个区间,如果它有一个端点比这个区间内除了端点的数都大,那么对答案就有一个 p2 的贡献,注意,左端点和右端点可能都对答案有贡献。

这个怎么计算呢?显然我们要正着做一遍反着做一遍来把左右端点的贡献都给求出来,这里只讨论正着做,反着的话你把数组reverse一下,把询问区间翻转一下就一样了。

首先我们用单调栈处理出一个点后面的第一个比它大的数的位置,记为 ri 。然后我们把询问都按照左端点排序,然后从右到左扫描线。扫到一个位置,现在线段树 [i+1,ri] 位置区间加上 p2 ,然后处理挂在这个左端点的所有询问,左端点对于这个询问的贡献就是线段树上 [1,R] ( R 是查询的右端)的和。

这样做,我们能把所有p2都很好地求出来,可是 p1 都被我们当成了两倍的 p2 ,怎么办呢?可以发现,我们枚举作为次大值的左/右端点,那这种区间只会有一个,就是 [i,ri] ,因此我们在做上面扫描线过程中,除了区间加之外,还给位置 ri 单点减去 p1−2p2 就好了。

时间复杂度 O(qlogn) 。

代码实现

#include <algorithm>
#include <iostream>
#include <cstdio>
#include <cctype>

using namespace std;

typedef long long LL;

int read()
{
    int x=,f=;
    char ch=getchar();
    while (!isdigit(ch)) f=ch=='-'?-:f,ch=getchar();
    while (isdigit(ch)) x=x*+ch-'0',ch=getchar();
    return x*f;
}

int buf[];

void write(LL x)
{
    if (x<) putchar('-'),x=-x;
    for (;x;x/=) buf[++buf[]]=x%;
    if (!buf[]) buf[++buf[]]=;
    for (;buf[];putchar('0'+buf[buf[]--]));
}

const int N=;
const int Q=;

struct segment_tree
{
    LL tag[N<<],sum[N<<];
    int size[N<<];

    void ADD(int x,LL delta){sum[x]+=l*delta*size[x],tag[x]+=delta;}

    void clear(int x,int l,int r)
    {
        if (tag[x])
        {
            if (l!=r) ADD(x<<,tag[x]),ADD(x<<|,tag[x]);
            tag[x]=;
        }
    }

    void update(int x){size[x]=size[x<<]+size[x<<|],sum[x]=sum[x<<]+sum[x<<|];}

    void modify(int x,int st,int en,int l,int r,int delta)
    {
        clear(x,l,r);
        if (st==l&&en==r)
        {
            ADD(x,delta),clear(x,l,r);
            return;
        }
        int mid=l+r>>;
        if (en<=mid) modify(x<<,st,en,l,mid,delta);
        else if (mid+<=st) modify(x<<|,st,en,mid+,r,delta);
        else modify(x<<,st,mid,l,mid,delta),modify(x<<|,mid+,en,mid+,r,delta);
        update(x);
    }

    LL query(int x,int st,int en,int l,int r)
    {
        clear(x,l,r);
        if (st==l&&en==r) return sum[x];
        int mid=l+r>>;
        if (en<=mid) return query(x<<,st,en,l,mid);
        else if (mid+<=st) return query(x<<|,st,en,mid+,r);
        else return query(x<<,st,mid,l,mid)+query(x<<|,mid+,en,mid+,r);
    }

    void build(int x,int l,int r)
    {
        tag[x]=;
        if (l==r)
        {
            sum[x]=,size[x]=;
            return;
        }
        int mid=l+r>>;
        build(x<<,l,mid),build(x<<|,mid+,r),update(x);
    }
}t;

LL res[Q];
int stack[N],a[N],R[N];
int top,n,q,p1,p2;

struct qy
{
    int l,r,id;

    bool operator<(qy const x)const{return l<x.l;}
}ask[Q];

void calc()
{
    t.build(,,n+),sort(ask+,ask++q),a[stack[top=]=n+]=n+;
    for (int i=n;i>=;--i)
    {
        for (;top&&a[stack[top]]<a[i];--top);
        R[i]=stack[top],stack[++top]=i;
    }
    for (int i=n,cur=q;i>=;--i)
    {
        t.modify(,i+,R[i],,n+,p2),t.modify(,R[i],R[i],,n+,p1-*p2);
        for (;cur&&ask[cur].l==i;--cur) res[ask[cur].id]+=t.query(,,ask[cur].r,,n+);
    }
}

int main()
{
    freopen("sf.in","r",stdin),freopen("sf.out","w",stdout);
    n=read(),q=read(),p1=read(),p2=read();
    for (int i=;i<=n;++i) a[i]=read();
    for (int i=;i<=q;++i) ask[i].l=read(),ask[i].r=read(),ask[i].id=i;
    calc(),reverse(a+,a++n);
    for (int i=;i<=q;++i) ask[i].l=n+-ask[i].l,ask[i].r=n+-ask[i].r,swap(ask[i].l,ask[i].r);
    calc();
    for (int i=;i<=q;++i) write(res[i]),putchar('\n');
    fclose(stdin),fclose(stdout);
    return ;
}