保卫方案

保卫方案

时间限制：2秒

空间限制：65536K

战争游戏的至关重要环节就要到来了，这次的结果将决定王国的生死存亡，小B负责首都的防卫工作。首都位于一个四面环山的盆地中，周围的n个小山构成一个环，作为预警措施，小B计划在每个小山上设置一个观察哨，日夜不停的瞭望周围发生的情况。 一旦发生外地入侵事件，山顶上的岗哨将点燃烽烟，若两个岗哨所在的山峰之间没有更高的山峰遮挡且两者之间有相连通路，则岗哨可以观察到另一个山峰上的烽烟是否点燃。由于小山处于环上，任意两个小山之间存在两个不同的连接通路。满足上述不遮挡的条件下，一座山峰上岗哨点燃的烽烟至少可以通过一条通路被另一端观察到。对于任意相邻的岗哨，一端的岗哨一定可以发现一端点燃的烽烟。 小B设计的这种保卫方案的一个重要特性是能够观测到对方烽烟的岗哨对的数量，她希望你能够帮她解决这个问题。 

输入描述:

输入中有多组测试数据，每一组测试数据的第一行为一个整数n(3<=n<=10^6),为首都周围的小山数量，第二行为n个整数，依次表示为小山的高度h（1<=h<=10^9）.

输出描述:

对每组测试数据，在单独的一行中输出能相互观察到的岗哨的对数。

输入例子1:

5
1 2 4 5 3

输出例子1:

7

解题思路：先从最大的数字开始遍历，维护一个单调递减的单调栈，用pair来记录连续的相同的数字出现了几次

#include <iostream>   
#include <cstdio>   
#include <cstring>   
#include <string>   
#include <algorithm>   
#include <map>   
#include <set>   
#include <stack>   
#include <queue>   
#include <vector>   
#include <bitset>   
#include <functional>   
 
using namespace std;
 
#define LL long long   
const int INF = 0x3f3f3f3f;

int n;

int main()
{
	while (~scanf("%d", &n))
	{
		int ma = -1;
		stack<pair<int, int> >s;
		LL ans = 0;
		vector<int> g(n, 0);
		for (int i = 0; i < n; i++)
		{
			scanf("%d", &g[i]);
			ma = max(ma, g[i]);
		}
		vector<int>::iterator it = g.begin();
		while (*it != ma)
		{
			int tmp = *it;
			it = g.erase(it);
			g.push_back(tmp);
		}
		s.push(pair<int, int>(g[0], 1));
		for (int i = 1; i < n; i++)
		{
			if (g[i] == s.top().first) s.top().second++;
			else if (g[i] < s.top().first) s.push(pair<int, int>(g[i], 1));
			else
			{ 
				while (g[i] > s.top().first)
				{
					LL tmp = s.top().second;
					ans += tmp * (tmp - 1) / 2 + 2 * tmp;
					s.pop();
				}
				if (g[i] == s.top().first) s.top().second++;
				else s.push(pair<int, int>(g[i], 1));
			}
		}
		if (s.size() == 1)
		{
			LL tmp = s.top().second;
			ans += tmp * (tmp - 1) / 2;
			s.pop();
		}
		else
		{
			while (s.size() > 2)
			{
				LL tmp = s.top().second;
				ans += tmp * (tmp - 1) / 2 + 2 * tmp;
				s.pop();
			}
			LL x = s.top().second;
			s.pop();
			LL y = s.top().second;
			s.pop();
			if(y == 1) ans += x * (x - 1) / 2 + y * (y - 1) / 2 + x;
			else ans += x * (x - 1) / 2 + y * (y - 1) / 2 + 2 * x;
		}
		printf("%lld\n", ans);
	}
    return 0;
}