To 洛谷.1823 音乐会的等待

题目描述

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

代码:

栈模拟。

 #include<cstdio>
using namespace std; int n,Ans;
struct Stack
{
int tp,sz[];
Stack()
{
tp=;
}
void Del()
{
--tp;
}
int Get()
{
return sz[tp];
}
bool Empty()
{
return tp==;
}
void Insert(int x)
{
sz[++tp]=x;
}
}s; void read(int &now)
{
now=;char c=getchar();
while(c<''||c>'')c=getchar();
while(c>=''&&c<='')now=now*+c-'',c=getchar();
} int main()
{
read(n);
int a;
while(n--)
{
read(a);int t=;
while(a>=s.Get() && !s.Empty())
{//当当前高度比栈顶元素大时,栈中元素要一直弹出直到栈顶大于等于当前高度
//(比当前小的都看不见后面的了)
//每次弹出Ans都++(比当前矮的可以看见当前元素,但不能看见以后的元素 so弹出)
//这里将所有等于当前元素的也一起弹出了,并记录弹出个数
if(s.Get()==a) ++t;
s.Del(),++Ans;
}
if(!s.Empty())//此时栈中还有 那一定是大于当前高度的,他们可以互相看见
++Ans;
while(t--)//将原先等于当前的重新入栈 因为t初始=1,所以当前的也会算在内 入栈
s.Insert(a);
} printf("%d",Ans);
return ;
}
05-11 16:53