51nod 1810 连续区间

51nod 1810 连续区间

区间内所有元素排序后，任意相邻两个元素值差为1的区间称为“连续区间”
如：3,1,2是连续区间，但3,1,4不是连续区间
给出一个1~n的排列，求出有多少个连续区间

Input

一个数n(n<=1,000,000)
第二行n个数,表示一个1~n的排列

Output

一个数，表示有多少个连续区间

Input示例

5
2 1 5 3 4

Output示例

9

Solution

**考虑分治，每次找出一个mid，算出左端点在[l,mid]区间内，右端点在(mid,r]区间内的答案数量。
每次以mid为分界点维护一个前后缀max和前后缀min，分四种情况讨论：
max(max[i],max[j])=max[i] min(min[i],min[j])=min[i]
则 j-i=max[i]-min[i]
max(max[i],max[j])=max[j] min(min[i],min[j])=min[i]
则 j-i=max[j]-min[i] 这种情况开个桶维护一下即可
代码实现上有不少细节要考虑**

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for (int i=a;i<=b;i++)
#define fd(i,a,b) for (int i=a;i>=b;i--)
using namespace std;
const int N=1000050;
typedef long long ll;
ll ans;
int n;
int b[3*N],a[N];
int mi[N],mx[N];
int read(){
    int sum=0;
    char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9') {
        sum=sum*10+c-'0';
        c=getchar();
    }
    return sum;
}
void solve(int l,int r){
    int i,j;
    int mid=(l+r)>>1;
    mi[mid]=a[mid],mx[mid]=a[mid];
    fd (i,mid-1,l) {
        mi[i]=min(mi[i+1],a[i]);
        mx[i]=max(mx[i+1],a[i]);
    }
    mi[mid+1]=a[mid+1],mx[mid+1]=a[mid+1];
    fo(i,mid+2,r) {
        mi[i]=min(mi[i-1],a[i]);
        mx[i]=max(mx[i-1],a[i]);
    }
    fo(i,l,mid) {
        j=mx[i]-mi[i]+i;
        if (j<=r&&mx[j]<mx[i]&&mi[j]>mi[i]&&j>mid) ans++;
    }
    for (int j=mid+1;j<=r;j++) {
        i=j-(mx[j]-mi[j]);
        if (i>=l&&mx[j]>mx[i]&&mi[j]<mi[i]&&i<=mid) ans++;
    }
    ////////////////////////// 
    int x=mid;int y=mid+1;
    fd(i,mid,l){
        while (x<r&&mx[x+1]<mx[i]) b[mi[x+1]+x+1+N]++,x++;
        while (y<=x&&mi[y]>mi[i]) b[mi[y]+y+N]--,y++;
        ans+=b[mx[i]+i+N];
    }
    while (y<=x) b[mi[y]+y+N]--,y++;
    x=mid; y=mid+1;
    fo(i,mid+1,r) {
      while (y>l&&mx[y-1]<mx[i]) y--,b[mi[y]-y+N]++;
      while (x>=y&&mi[x]>mi[i]) b[mi[x]-x+N]--,x--;
      ans+=b[mx[i]-i+N];
    }
    while (x>=y) b[mi[x]-x+N]--,x--;
    //////////////////////
    if (l==r) return;
    solve(l,mid);
    solve(mid+1,r);
}
int main(){
    freopen("1.in","r",stdin);
    freopen("1.out","w",stdout);
    int i;
    n=read();
    fo(i,1,n) a[i]=read();
    solve(1,n);
    ans+=n;
    printf("%lld\n",ans); 
}