这一题看见我就想套单调数据结构什么的,然后各种做不出来,超时什么的,看了一下英文版的官方题解,体会了题解的妙处。
先找到序列中最高的作为数组的第一个元素,把圈切成一条链,然后就可以开始做了。
具体说来,设r[i]表示从i这个位置往右走第一个严格大于w[i]的位置,l[i]正相反,再记c[i]表示i到r[i]中与i相等的数的个数。
从左到右扫一遍,每个w[i]可以贡献给答案c[i],还有两对(l[i],i),(i,r[i]),但是如果l[i]==r[i]的话,就只贡献一对,ans--。注意要开long long .
%%__debug大神。
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<iostream>
const int MAXN=1e6+10;
int w[MAXN],c[MAXN],l[MAXN],r[MAXN],n;
long long ans=0;
int main()
{
int max=0,p;
scanf("%d",&n);
for(int i=0;i<n;i++)
{
scanf("%d",&w[i]);
if(w[i]>max)
{
max=w[i];
p=i;
}
}
std::rotate(w,w+p,w+n);
w[n]=max;
c[n]=0;
for(int i=n-1;i>=0;i--)
{
r[i]=i+1;
while(r[i]<n&&w[i]>w[r[i]])r[i]=r[r[i]];
if(r[i]!=n&&w[i]==w[r[i]])
{
c[i]=c[r[i]]+1;
r[i]=r[r[i]];
}
}
for(int i=1;i<=n;i++)
{
l[i]=i-1;
while(l[i]&&w[i]>=w[l[i]])
l[i]=l[l[i]];
}
for(int i=1;i<n;i++)
{
ans+=c[i];
ans+=2;
if(l[i]==0&&r[i]==n)ans--;
}
printf("%I64d\n",ans);
}