洛谷 P1823 音乐会的等待

题目描述

N个人正在排队进入一个音乐会。人们等得很无聊，于是他们开始转来转去，想在队伍里寻找自己的熟人。队列中任意两个人A和B，如果他们是相邻或他们之间没有人比A或B高，那么他们是可以互相看得见的。

写一个程序计算出有多少对人可以互相看见。

输入输出格式

输入格式：

输入的第一行包含一个整数N (1 ≤ N ≤ 500 000), 表示队伍中共有N个人。

接下来的N行中，每行包含一个整数，表示人的高度，以毫微米(等于10的-9次方米)为单位，每个人的调度都小于2^31毫微米。这些高度分别表示队伍中人的身高。

输出格式：

输出仅有一行，包含一个数S，表示队伍中共有S对人可以互相看见。

输入输出样例

输入样例#1：

7 
2 
4 
1 
2 
2 
5 
1

输出样例#1：

10

说明

数据制作: @w

辣鸡题目，本来我是冲着线段树来的，结果是单调栈！！！

放一下洛谷题解吧https://www.luogu.org/wiki/show?name=%E9%A2%98%E8%A7%A3+P1823

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=500005;
int n,ans,tp,h[maxn],stk[maxn];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&h[i]);
	for(int i=1;i<=n;i++)
	{
		int k=1;
		while(tp>0&&h[i]>stk[tp])
			tp--,ans++;
		while(tp>0&&h[i]==stk[tp])
			tp--,ans++,k++;
		if(tp!=0)
			ans++;
		tp+=k;
		stk[tp]=h[i];
	}
	printf("%d\n",ans);
	return 0;
}