【来源】
未知。
【题目描述】
给你n个正整数,不超过 1 0 9 10^9 109, 这n个数构成了一个环。让你找数对的个数,这些数对满足的条件是: 对于数对(a,b),只要a到b的中间没有比a大和比b大的数,那么这个数对就是满足条件的,由于环上有2个弧,只要a到b的任意一个弧满足条件即可。
求这样的数对的个数。
【输入格式】
第一个整数n。
然后有n个整数。
【输出格式】
满足条件的数对的个数。
【样例输出】
5
1 2 4 5 3
【样例输出】
7
【数据范围】
对于30%的数据,n不超过1000。
对于100%的数据,n不超过1000000。
【题目大意】
给你⼀个长度度为n的环形序列,求有多少个数对,他们中间没有比他们其中⼀个大(只要
有⼀个弧满足条件即可)。
【解析】
问题的难点在于n很大,而且是环形。
首先我们来考虑如果是线性的情况应该怎么做?
比如如我们枚举左端点,然后去找右端点,这样利用单调性,可以做到O(n)。
但是这个问题是环形的,比较麻烦,我们如何来处理呢?
我们⾸先找到最⾼的⼀个,然后把它旋转到最左边。
我们先来考虑相等的情况,相等的⼀对有时候也是可以的,只要他们中间没有比他们大的,如果要逆时针寻找,那这个数必须和最高的⼀样高,这样就不用逆时针了,所以相等的情况我们不需要考虑逆时针。
接下来我们考虑不相等的情况,我们用l[x]表示x左边第⼀个比⾼的,同理r[x]表示x位置右边比第⼀个比它高的。 这样显然(x,l[x])和(x,r[x])都构成了一对。接下来我们们要考虑逆时针的情况,和上面⼀样,也是不⽤考虑的,因为最大的数在最左边。
找相等的有几个和l[x]、r[x]都只需要⽤类似于路径压缩的方法来寻找,可以做到接近O(n)(不会证明)。
这⾥我们处理⼀下特殊情况,当某个x的l和r都是最左端和最右端的时候,我们可以只算⼀边的。
【代码】
#pragma GCC optimize(3,"Ofast","inline")
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N=1e6+5;
const int inf=1e9+10;
int n;
int l[N],r[N],c[N],a[N],b[N];
int main() {
scanf("%d",&n);
for(int i=0; i<n; i++) scanf("%d",&a[i]);
int x=0;
for(int i=0; i<n; i++)
if(a[i]>a[x]) x=i;
for(int i=0; i<n; i++)
b[i]=a[(x+i)%n];
b[n]=a[x];
b[n+1]=inf;
for(int i=n+1; i>=0; i--)
b[i+1]=b[i];
b[0]=inf;
for(int i=n+1; i>=1; i--) {
r[i]=i+1;
while(i<=n && b[i]>b[r[i]]) r[i]=r[r[i]];
if(r[i]<=n+1 && b[i]==b[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(b[i]>b[l[i]]) l[i]=l[l[i]];
if(b[i]==b[l[i]]) l[i]=l[l[i]];
}
LL ans=0;
for(int i=2; i<=n; i++)
ans+=c[i];
for(int i=2; i<=n; i++) {
int x=2;
if(l[i]==0) x--;
if(r[i]==n+2) x--;
if(l[i]==1 && r[i]==n+1) x--;
ans+=x;
}
printf("%lld\n",ans);
return 0;
}