Codeforces 5E. Bindian Signalizing

5E. Bindian Signalizing题目链接

题目大意

• 山构成一个圆，每座山都有对应的高度，而两座山相互可见的要求是两山之间没有一座山比它们任意一座山高。（顺时逆时皆可）。

输入

• $n$为山的数量
• $n$座山的高度

输出

• 相互可见的山总共有几对

样例

input
5
1 2 4 5 3
output:
7		-- 解析：12, 24, 45, 53, 31, 23, 24

思路：

• 官方题解

解环成链：

• 做法：在一个环中选择最大的那个值作为链的起点
• 理由：构成的答案中绝对不存在经过这个最高点而构成的满足条件的山，相等情况除外。
• 做法：在链的最后加一个最大值
• 理由：否则 $right[i]$ 情况会少

顺时针逆时针的处理：

• 对数组从左往右遍历得到 $left[i]$，对数组从右往左遍历得到 $right[i]$。

总体思路：

• $left[i]$ : 左边第一个大于$v[i]$的坐标
• $right[i]$: 右边第一个大于$v[i]$的坐标
• $same[i]$: $[left[i],i)$ 这个区间内和$v[i]$这个值相等的个数
• 最后结果是$(i,left[i])$ 为一对， $(i,right[i])$ 为一对，这里$left[i]==0$ 和 $right[i]==n+1$为同一种情况。再加上$same[i]$。

计算：

• 求$left[i],right[i]$，典型的单调栈
• 求$same[i]$ 就是碰到左边第一个相等 $j$ 的时候，就是$same[i]=same[j]+1$

代码

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <string>
#include <vector>
#include <cstring>
#include <map>
#include <stack>
using namespace std;

typedef long long llong;

int main()
{
	std::ios::sync_with_stdio(false);		//没有会T
	int n;
	cin >> n;

	vector<int> v(n);

	for (int i = 0; i < n; i++)
		cin >> v[i];

	rotate(v.begin(), max_element(v.begin(), v.end()), v.end());

	v.push_back(v[0]);
	//for (int i = 0; i <= n; i++)
    //    cout << v[i] << endl;

	vector<int> left(n + 1);
	vector<int> right(n + 1);
	vector<int> same(n + 1, 0);

	stack<int> s;
	s.push(0);
	left[0] = -1;
	for (int i = 1; i < n; i++)
	{
		while (!s.empty() && v[s.top()] <= v[i])
		{
			if (v[s.top()] == v[i])
				same[i] = same[s.top()] + 1;
			s.pop();
		}

		if (s.empty())
            left[i] = -1;
        else
            left[i] = s.top();
		s.push(i);
	}

	while (!s.empty())
		s.pop();

	s.push(n);
	right[n] = n;
	for (int i = n - 1; i >= 1; i--)
	{
		while (!s.empty() && v[s.top()] <= v[i])
			s.pop();

		if (s.empty())
            right[i] = n + 1;
        else
            right[i] = s.top();
		s.push(i);
	}

    //for (int i = 1; i < n; i++)
    //    printf("left[%d]=%d right[%d]=%d same[%d]=%d\n", i, left[i], i, right[i], i, same[i]);

	llong ans = 0;
	for (int i = 1; i < n; i++)
	{
		if (left[i] != -1)
            ans++;
        if (right[i] != n + 1)
            ans++;
        if (left[i] == 0 && right[i] == n)
            ans--;
		ans += same[i];
	}
	cout << ans << endl;
	return 0;
}