题目描述
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
这道题一开始自己推出来了如果后面有比他大的那么这个数是没用的,
然后乱搞了一下搞了25分。。。
看了一下题解发现一个非常好的思路
就是把相等和大于放到一个循环里判断,
一开始num==1保证有数据读入
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<stack>
using namespace std;
int read(int & n)
{
char c='-';int x=;
while(c<''||c>'')c=getchar();
while(c>=''&&c<='')
{
x=x*+(c-);
c=getchar();
}
n=x;
}
const int MAXN=;
stack<int>s;
int main()
{
int n,h,ans=,flag=;
read(n);
for(int i=;i<=n;i++)
{
flag=;
read(h);
if(i==)
{
s.push(h);
continue;
}
if(h>s.top())
{
ans++;
s.pop();
while(s.size()>&&h>s.top())
{
ans++;
s.pop();
flag=;
}
if(s.size()!=&&h<s.top())
ans++;
s.push(h);
}
else if(h==s.top())
{
ans=ans+s.size();
s.push(h);
}
else
{
ans++;
s.push(h);
}
}
printf("%d",ans);
}
乱搞25分
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<stack>
using namespace std;
int read(int & n)
{
char c='-';int x=;
while(c<''||c>'')c=getchar();
while(c>=''&&c<='')
{
x=x*+(c-);
c=getchar();
}
n=x;
}
const int MAXN=;
stack<int>s;
int main()
{
int n,h,ans=,flag=;
read(n);
for(int i=;i<=n;i++)
{
int num=;
read(h);
while(s.size()!=&&h>=s.top())
{ if(h==s.top())
num++;
ans++;
s.pop();
}
if(s.size()!=)
ans++;
while(num--)
s.push(h);
}
printf("%d",ans);
}