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
****************************************************************/