音乐会的等待【单调栈】

1308: 音乐会的等待

时间限制: 1 Sec   内存限制: 128 MB
提交: 297   解决: 89
[ 提交][ 状态][ 讨论版]

题目描述

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

输入

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

输出

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

样例输入

7 
2 
4 
1 
2 
2 
5 
1 

样例输出

10

提示

来源

简单数据结构

思路：

单调栈！思路还是要缜密啊！wa了好久。。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;

struct asd{
    int w;
    LL Left;    //就是表它向右的答案
    LL cet;    //右边重复个数
};
int n,x;
stack<asd>q;
LL ans;
int main()
{
    while(!q.empty()) q.pop();
    scanf("%d",&n);
    asd now,nex;
    ans=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        now.Left=0; //都有它一个。 
        now.cet=0;
        now.w=x;
        if(q.empty())
            q.push(now);
        else
        {
            if(q.top().w>=now.w)
                q.push(now);
            else
            {
                while(!q.empty()&&q.top().w<now.w)
                {
                    nex=q.top();q.pop();
                    ans=ans+nex.Left+1LL;   //直接+答案咯  
                    if(!q.empty())
                    {
                        q.top().Left+=nex.cet+1LL;
                        if(q.top().w==nex.w)
                            q.top().cet+=nex.cet+1LL;
                    }
                    //printf("%d %lld\n",x,ans);
                }
                q.push(now);
            }
        }
    }
    while(!q.empty())
    {
        nex=q.top();q.pop();
        if(!q.empty())
        {
            q.top().Left+=nex.cet+1LL;
            if(q.top().w==nex.w)
                q.top().cet+=nex.cet+1LL;
        }
        ans=ans+nex.Left;
    }
    printf("%lld\n",ans);
    return 0;
}
/*
12
2 4 1 2 2 5 4 1 2 2 5 4
4
1 2 3 4
4
4 1 1 4
*/
/**************************************************************
    Problem: 1308
    User: 135135135135
    Language: C++
    Result: 正确
    Time:320 ms
    Memory:9620 kb
****************************************************************/