其他分享
首页 > 其他分享> > 音乐会排队

音乐会排队

作者:互联网

【问题描述】

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

  输入: 输入的第一行包含一个整数N(1≤N≤500000),表示队伍中共有N个人。接下来的N行中,每行包含一个整数,表示人的高度,以毫微米(等于10的-9次方米)为单位,每个人的调度都小于231毫微米。这些高度分别表示队伍中人的身高。

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

【输入样例】

  7

  2

  4

  1

  2

  2

  5

  1

【输出样例】

  10

#include<iostream>
using namespace std;

int main(){
    int st[500001]={},n,m,top=1,cnt=0;
    cin>>n;
    cin>>st[1]; // 默认栈顶为1:
    for(int i=2; i<=n; i++){
        cin>>m;
        if(st[top]>m){ // 如果输入的元素依次递减,直接入栈,cnt加1。 
            cnt++;
            st[++top]=m;
        }else{ // 如果输入的元素大于或等于栈顶元素 
            int rear=top;
            while(rear>1&&m>=st[rear]) rear--; // 计算比当前元素小的栈里元素个数。
            cnt+=top-rear+1;
            while(top>0&&st[top]<m) top--; // 出栈,删除元素。
            st[++top]=m; 
        }
    } 
    cout<<cnt;
    return 0;
}

 

标签:音乐会,int,top,排队,cnt,st,输入,rear
来源: https://www.cnblogs.com/dks0313/p/16496235.html