题目:http://codeforces.com/contest/5/problem/E
题意:一些不同高度的山排成环,两座山之间如果没有比任何一座山高的山,则认为两座山可以互相瞭望到。求共有多少对可以互相瞭望到的山?
题解:只要对每个点求出满足条件的点即可,为了避免重复计算一对点,对每个点,只需考虑比其高的点。
首先将环转化为链,将最高的山排在第一位,这样就不用考虑这座山,因为他不会有满足上述条件的点。
对每个点,计算左边满足条件的点,也就是第一个比高于自身的点,和右边满足条件的点,也就是右边第一个比高于自身的点,这样就构成了对点,当然如果两个点重合要去掉一个。还要考虑相等的情况,计算上述两个点之间相等的点有多少个,为了避免重复计算,只计算到右边的点之间相等的点,这些点都可以为结果贡献一对点。
代码:
#include <bits/stdc++.h>
using namespace std;
const int MAX =1000000+10;
int oH[MAX];
int H[MAX],L[MAX],R[MAX];
int same[MAX];
int main()
{
int n;
scanf("%d",&n);
int maxPos = 0;
for(int i=0;i<n;i++)
{
scanf("%d",&oH[i]);
if(oH[i]>oH[maxPos])
{
maxPos = i;
}
}
for(int j=0;j<n;j++)
{
H[j] = oH[(maxPos+j)%n];
}
L[0]=0;
for(int i=1;i<n;i++)
{
L[i]=i-1;
while(L[i]>0&&H[i]>=H[L[i]])
L[i] = L[L[i]];
}
for(int i=n-1;i>=0;i--)
{
R[i]=i+1;
while(R[i]<n&&H[i]>H[R[i]])
R[i] = R[R[i]];
if(R[i]<n&&H[i]==H[R[i]])
{
same[i]=same[R[i]]+1;
R[i]=R[R[i]];
}
}
long long ans =0;
for(int i=1;i<n;i++)
{
ans+=same[i]+2;
if(L[i]==0&&R[i]==n)
{
ans--;
}
}
cout<<ans<<endl;
return 0;
}