小a的排列（线段树维护）

链接：https://ac.nowcoder.com/acm/contest/317/G
来源：牛客网
 

题目描述

小a有一个长度为nn的排列。定义一段区间是"萌"的，当且仅当把区间中各个数排序后相邻元素的差为11
现在他想知道包含数x,yx,y的长度最小的"萌"区间的左右端点

也就是说，我们需要找到长度最小的区间[l,r][l,r]，满足区间[l,r][l,r]是"萌"的，且同时包含数xx和数yy

如果有多个合法的区间，输出左端点最靠左的方案。

 

 

输入描述:

第一行三个整数N,x,yN,x,y，分别表示序列长度，询问的两个数
第二行有nn个整数表示序列内的元素，保证输入为一个排列

输出描述:

输出两个整数，表示长度最小"萌"区间的左右端点

示例1

输入

复制

5 2 3
5 2 1 3 4

输出

复制

2 4

说明

区间[2,4]={2,1,3}[2,4]={2,1,3}包含了2,32,3且为“萌”区间，可以证明没有比这更优的方案

示例2

输入

复制

8 3 5
6 7 1 8 5 2 4 3

输出

复制

5 8

备注:

保证2⩽n⩽105,1⩽x,y⩽n

思路：

可以先找到x，y的区间范围st，end。

然后再找st，end区间范围内的最小值x和最大值y。

再在x，y区间内找到区间范围st，end。

不断这样进行，直到end-st==y-x。

最大值和最小值，我是用线段树维护的。由于没有修改操作，所以用RMQ更好。

 

代码：

 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <map>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define root 1,N,1
#define LL long long
const int maxn=1e5+10;
const int INF=0x3f3f3f3f;
 int cnt=0;
LL max_[maxn<<2][2];
LL min_[maxn<<2][2];
int a[maxn];
int b[maxn];
void pushup0(int rt)
{
    max_[rt][0]=max(max_[rt<<1][0],max_[rt<<1|1][0]);
    min_[rt][0]=min(min_[rt<<1][0],min_[rt<<1|1][0]);
}
 
void pushup1(int rt)
{
    max_[rt][1]=max(max_[rt<<1][1],max_[rt<<1|1][1]);
    min_[rt][1]=min(min_[rt<<1][1],min_[rt<<1|1][1]);
}
 
void build0(int l,int r,int rt)
{
    if(l==r)
    {
     max_[rt][0]=a[cnt];
     min_[rt][0]=a[cnt++];
 
     return;
    }
    int m=(l+r)>>1;
    build0(lson);
    build0(rson);
    pushup0(rt);
}
 
void build1(int l,int r,int rt)
{
    if(l==r)
    {
     max_[rt][1]=b[cnt];
     min_[rt][1]=b[cnt++];
     return;
    }
    int m=(l+r)>>1;
    build1(lson);
    build1(rson);
    pushup1(rt);
}
 
int quert0_min(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return min_[rt][0];
    }
    int m=(l+r)>>1;
    int ret=INF;
    if(L<=m)ret=min(ret,quert0_min(L,R,lson));
    if(R>m)ret=min(ret,quert0_min(L,R,rson));
 
    return ret;
}
 
 
 int quert0_max(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return max_[rt][0];
 
    }
    int m=(l+r)>>1;
    int ret=0;
    if(L<=m)ret=max(ret,quert0_max(L,R,lson));
    if(R>m)ret=max(ret,quert0_max(L,R,rson));
 
    return ret;
}
int quert1_min(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return min_[rt][1];
    }
    int m=(l+r)>>1;
    int ret=INF;
    if(L<=m)ret=min(ret,quert1_min(L,R,lson));
    if(R>m)ret=min(ret,quert1_min(L,R,rson));
 
    return ret;
}
 
int quert1_max(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return max_[rt][1];
    }
    int m=(l+r)>>1;
    int ret=0;
    if(L<=m)ret=max(ret,quert1_max(L,R,lson));
    if(R>m)ret=max(ret,quert1_max(L,R,rson));
    return ret;
}
 
struct node
{
    int w,id;
}G[maxn];
 
int cmp(node a,node b)
{
    return a.w<b.w;
}
int x,y;
 
map<int,int>Map;
bool vis[maxn];
 
int main()
{
    int n;
    scanf("%d%d%d",&n,&x,&y);
    int st=1,end=n;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        G[i].w=a[i];
        G[i].id=i;
        if(a[i]==x)st=i;
        if(a[i]==y)end=i;
        Map[a[i]]=i;
    }
    sort(G+1,G+1+n,cmp);
    for(int i=1;i<=n;i++)
    {
        b[i]=G[i].id;
    }
 
    cnt=1;
    build0(1,n,1);
 
    cnt=1;
    build1(1,n,1);
 
    if(st>end)swap(st,end);
    x=quert0_min(st,end,1,n,1);
    y=quert0_max(st,end,1,n,1);
 
    int pst=st,pend=end;
    while(y-x!=end-st)
    {
        st=quert1_min(x,y,1,n,1);
        end=quert1_max(x,y,1,n,1);
 
        x=quert0_min(st,end,1,n,1);
        y=quert0_max(st,end,1,n,1);
    }
 
    printf("%d %d\n",st,end);
 
    return 0;
}