我的idea:先断环成链,存两遍,即Ai = Ai+n ,从左到右和从右到左各扫一遍,在这个过程中用优先队列存下来每个位置的b,在处理每个位置的时候,先把这个位置的a和优先队列中存的b比较,该出队列的就出去,然后把当前位置的b加入队列。扫完两遍后就出结果了。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<stack>
#include<algorithm>
#include<cmath>
#include<queue>
#define INF 0x3f3f3f3f
#define LL long long
#define N 251000
using namespace std;
struct node
{
int a,b,xu;
}t[N],u;
int l[151000],r[151000],n;
struct cmp
{
bool operator ()(node x,node y)//改变为小根堆
{
return x.b>y.b;
}
};
void work1()
{
node tag;
priority_queue<node,vector<node>,cmp>s;
s.push(t[1]);
for(int i=2;i<=2*n;i++)
{
bool pd = false;
int a = t[i].a,p = t[i].xu;
if( s.size()>0 )
{
u = s.top();
while( a>=u.b )//直到队中没有小于当前位置a值的b值为止
{
if( p==u.xu )//特别处理两个位置重复的情况
{
pd = true;
tag = u;
}
else l[ u.xu ] = p;
s.pop();
if( s.size()>0 ) u = s.top();
else break;
}
}
if( pd==true ) s.push(tag);
if( i<=n ) s.push(t[i]);//对于从左往右扫的情况,大于n的位置避免再入队
}
}
void work2()
{
node tag;
priority_queue<node,vector<node>,cmp>s;
s.push(t[2*n]);
for(int i=n+n-1;i>=1;i--)
{
bool pd = false;
int a = t[i].a,p = t[i].xu;
if( s.size()>0 )
{
u = s.top();
while( a>=u.b )
{
if( p==u.xu )
{
pd = true;
tag = u;
}
else r[ u.xu ] = p;
s.pop();
if( s.size()>0 ) u = s.top();
else break;
}
}
if( pd==true ) s.push(tag);
if( i>=n+1 ) s.push(t[i]);
}
}
int main()
{
// freopen("in.txt","r",stdin);
memset(l,INF,sizeof(l));
memset(r,INF,sizeof(r));
cin>>n;
for(int i=1;i<=n;i++)
{
scanf("%d",&t[i].a);
}
for(int i=1;i<=n;i++)
{
t[i].xu = i;
scanf("%d",&t[i].b);
}
for(int i=1;i<=n;i++)
{
t[i+n] = t[i];
}
work1();
work2();
for(int i=1;i<=n;i++)
{
// cout<<l[i]<<" "<<r[i]<<endl;
if( l[i]==INF&&r[i]==INF ) printf("-1 ");
else
{
int ll,rr;
if( l[i]>i ) ll = min( l[i]-i , i+n-l[i] );
else ll = min( i-l[i] , l[i]+n-i );
if( r[i]>i ) rr = min( r[i]-i , i+n-r[i] );
else rr = min( i-r[i] , r[i]+n-i );
printf("%d ",min(ll,rr));
}
}
return 0;
}
二分的办法:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define LL long long
using namespace std;
const int N=3e5+10;
int n,m,st[N][30],lg[N];
inline int query(int l,int r)
{
int q = lg[r-l+1];
return max( st[l][q] , st[ r-(1<<q)+1 ][q] );
}
inline int ef(int x,int i)
{
int l=1,r=n;
while(l<r)
{
int mid = (l+r)>>1;
if( max( query(i-mid,i-1) , query(i+1,i+mid) )>=x )//重中之重的二分,可以考虑作为模板使用
{
r = mid;
}
else
{
l = mid+1;
}
}
if( l==n ) return -1;
else return l;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
scanf("%d",&st[i][0]);
// cin>>st[i][0];
st[i+n][0] = st[i+n+n][0] = st[i][0];
}
for(int i=2;i<=n*3;i++)
{
lg[i] = lg[i>>1]+1;//预处理各个值的log2值,后面不用再用log函数
}
for(int i=1;i<=lg[n*3];i++)
{
for(int j=1;j+(1<<i)-1<=n*3;j++ )
{
st[j][i] = max( st[j][i-1] , st[j+(1<<(i-1))][i-1] );
}
}
for(int i=1;i<=n;i++)
{
int b;
scanf("%d",&b);
int ans = ef(b,i+n);
printf("%d ",ans);
}
return 0;
}