Friends and Subsequences CodeForces - 689D

http://codeforces.com/problemset/problem/689/D

分别对a和b维护最大和最小rmq 枚举左端点 右端点满足单调性 二分找第一个和最后一个使a和b区间最值相等的点即为右端点

#include <bits/stdc++.h>
using namespace std;
#define ll long long

int a[200010][20],b[200010][20];
int n;

void init()
{
    int i,j;
    for(j=1;(1<<j)<=n;j++)
    {
        for(i=1;i+(1<<j)-1<=n;i++)
        {
            a[i][j]=max(a[i][j-1],a[i+(1<<(j-1))][j-1]);
        }
    }
    for(j=1;(1<<j)<=n;j++)
    {
        for(i=1;i+(1<<j)-1<=n;i++)
        {
            b[i][j]=min(b[i][j-1],b[i+(1<<(j-1))][j-1]);
        }
    }
    return;
}

int query(int op,int l,int r)
{
    int len;
    len=log2(double(r-l+1));
    if(op==1) return max(a[l][len],a[r-(1<<len)+1][len]);
    else return min(b[l][len],b[r-(1<<len)+1][len]);
}

int main()
{
    ll ans;
    int i,pl,pr,l,r,m,rl,rr;
    scanf("%d",&n);
    for(i=1;i<=n;i++) scanf("%d",&a[i][0]);
    for(i=1;i<=n;i++) scanf("%d",&b[i][0]);
    init();
    ans=0;
    for(i=1;i<=n;i++)
    {
        pl=-1,l=i,r=n;
        while(l<=r)
        {
            m=(l+r)/2;
            rl=query(1,i,m);
            rr=query(2,i,m);
            if(rl<rr) l=m+1;
            else if(rl==rr) pl=m,r=m-1;
            else r=m-1;
        }
        pr=-1,l=i,r=n;
        while(l<=r)
        {
            m=(l+r)/2;
            rl=query(1,i,m);
            rr=query(2,i,m);
            if(rl<rr) l=m+1;
            else if(rl==rr) pr=m,l=m+1;
            else r=m-1;
        }
        if(pl!=-1) ans+=(pr-pl+1);
    }
    printf("%lld\n",ans);
    return 0;
}

/*
2
3 2
4 3
*/