【JZOJ5073】【GDOI2017第三轮模拟day1】影魔

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]。

Data Constraint

30%：1<= n,m <= 500。
另30%: p1=2*p2。
100%:1 <= n,m <= 200000；1 <= p1,p2 <= 1000。

Solution

这道题考场推出了大半，但没看到输入是排列，所以就GG了。
这里只讲右边的情况，左边同理,我们考虑作为最高点的情况。
我们先满足条件1：我们可以求出一个点最多可以往右延伸多远遇到第一个比他高的点r[i]。对于一个询问，我们只要查询[l,r]范围内有多少数的r[i]∈[l,r]即可。式子为

$$ans=\sum_{i=l}^{r}\sum_{j=i+1}^{r[i]-1}[max(a[k],k∈(i,j))<min(a[i],a[j])]$$
我们先满足条件2:我们写出右边的式子 $$ans=\sum_{i=l}^{r}\sum_{j=i+1}^{r[i]-1}[min(a[i],a[j])<max(a[k],k∈(i,j))<max(a[i],a[j])]$$ 这式子好烦，我们考虑把min去掉
那么条件2就包括了条件一，我们只要求出条件2的方案，真实条件2的方案=条件2的方案-条件一的方案。那么显然在i+1到r[i]-1的数都小于a[i]，所以 $$ans=\sum_{i=l}^{r}\sum_{j=i+1}^{r[i]-1}[max(a[k],k∈(i,j))<max(a[i],a[j])]=\sum_{i=l}^{r}min(r,r[i]-1)-i$$ 。这个可以拆开来做，用主席树轻松过。

Code

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const ll maxn=2e5+5;
struct code{
    ll l,r,num,sum;
}f[maxn*20],f1[maxn*20];
ll d[maxn],d1[maxn],l[maxn],a[maxn],r[maxn];
ll n,m,p1,p2,i,t,j,k,x,y,z,num,num1,ans,q,q1;
void insert(int l,int r,ll &v,int v1,int x){
    int mid=(l+r)/2;v=++num;f[v]=f[v1];
    if (l==r){
        f[v].num++;f[v].sum+=l+1;return;
    }
    if (mid>=x) insert(l,mid,f[v].l,f[v1].l,x);
    else insert(mid+1,r,f[v].r,f[v1].r,x);
    f[v].num=f[f[v].l].num+f[f[v].r].num;f[v].sum=f[f[v].l].sum+f[f[v].r].sum;
}
void insert1(int l,int r,ll &v,int v1,int x){
    int mid=(l+r)/2;v=++num1;f1[v]=f1[v1];
    if (l==r){
        f1[v].num++;f1[v].sum+=l-1;return;
    }
    if (mid>=x) insert1(l,mid,f1[v].l,f1[v1].l,x);
    else insert1(mid+1,r,f1[v].r,f1[v1].r,x);
    f1[v].num=f1[f1[v].l].num+f1[f1[v].r].num;f1[v].sum=f1[f1[v].l].sum+f1[f1[v].r].sum;
}
void find(int l,int r,int v,int v1,int x,int y){
    int mid=(l+r)/2;
    if (f[v].num==f[v1].num) return;
    if (l>=x && r<=y){
        t+=f[v].num-f[v1].num;k+=f[v].sum-f[v1].sum;
        return;
    }
    if (l<=y && mid>=x) find(l,mid,f[v].l,f[v1].l,x,y);
    if (mid<y && r>=x) find(mid+1,r,f[v].r,f[v1].r,x,y);
}
void find1(int l,int r,int v,int v1,int x,int y){
    int mid=(l+r)/2;
    if (f1[v].num==f1[v1].num) return;
    if (l>=x && r<=y){
        t+=f1[v].num-f1[v1].num;k+=f1[v].sum-f1[v1].sum;
        return;
    }
    if (l<=y && mid>=x) find1(l,mid,f1[v].l,f1[v1].l,x,y);
    if (mid<y && r>=x) find1(mid+1,r,f1[v].r,f1[v1].r,x,y);
}
int main(){
    freopen("sf.in","r",stdin);freopen("sf.out","w",stdout);
    scanf("%lld%lld%lld%lld",&n,&m,&p1,&p2);
    for (i=1;i<=n;i++){
        scanf("%lld",&a[i]);    
        while (d[0] && a[d[d[0]]]<=a[i]) r[d[d[0]--]]=i;
        if (d[0]) l[i]=d[d[0]];
        d[++d[0]]=i;
    }
    while (d[0]) r[d[d[0]--]]=n+1;
    memset(d,0,sizeof(d));
    for (i=1;i<=n;i++)
        insert(0,n+1,d[i],d[i-1],l[i]),insert1(0,n+1,d1[i],d1[i-1],r[i]);
    for (i=1;i<=m;i++){
        scanf("%lld%lld",&x,&y);t=k=0;
        find(0,n+1,d[y],d[x-1],x,y);ans=t*p1;q=-(x+y)*(y-x+1)/2-t;
        t=k=0;find1(0,n+1,d1[y],d1[x-1],y+1,n+1);q+=t*y;
        t=k=0;find1(0,n+1,d1[y],d1[x-1],x,y);q+=k;ans+=q*p2;
        t=k=0;find1(0,n+1,d1[y],d1[x-1],x,y);ans+=t*p1;q=(x+y)*(y-x+1)/2-t;
        t=k=0;find(0,n+1,d[y],d[x-1],0,x-1);q-=t*x;
        t=k=0;find(0,n+1,d[y],d[x-1],x,y);q-=k;ans+=q*p2;
        printf("%lld\n",ans);
    }
}