POJ - 3250:Bad Hair Day(单调栈)

原题链接
Farmer John的奶牛在风中凌乱了它们的发型……
每只奶牛都有一个身高hi(1 ≤ hi ≤ 1,000,000,000),现在在这里有一排全部面向右方的奶牛,一共有N只(1 ≤ N ≤ 80,000)。对于奶牛i来说,如果奶牛i+1,i+2,……,N这些奶牛的身高严格小于奶牛i,则奶牛i可以看到它们凌乱的发型。
比如下面这个例子:

=
= =
= * - * = * 奶牛面向这边–>
= * = * = *
= - = = = *
= = = = = =
1 2 3 4 5 6

令ci表示第i只奶牛能够看到的发型数量,请计算c1 + c2 + c3 + … + cN的值
对于上面这个例子,答案为3 + 0 + 1 + 0 + 1 + 0=5。

输入
第一行:奶牛数量N
第二到N+1行:第i+1行输入奶牛i的身高

输出
第一行:一个整数即c1到cN的和

样例输入
6
10
3
7
4
12
2

样例输出
5

题目描述

一群身高不等的奶牛排成一排,向右看,每个奶牛只能看到身高小于自己的奶牛发型,问这些奶牛能够看到的发型总和是多少
思路:利用单调栈(栈中元素从栈顶往下越来越大)的思想,可以计算出,每只奶牛能被他前面的多少只奶牛看到自己的发型,就反向得到了奶牛看到的发型总和
借用别人的解释:首先弹出元素是因为它右边相邻牛比它高(看不到它的头发并且挡住了该牛的视线),“挡住”这个关键字很重要,这就是说该牛已经看不到后面的其他牛的头发啦,而思路一是按顺序比较身高,统计每头牛前面能看到它头发的牛数,既然该牛望不到后面,那么把它出栈对整个结果没有影响。
例如:10 3 7 4 12 2
①Height_List:10 入栈首身高

②3<10,不弹出,num=0+1(当前栈中元素数),3入栈后 Height_List:10 3

③7 > 3(3 弹出) 7<10(10保留),栈中剩 10 ,num=1+1,7入栈后 Height_List:10 7

④4 < 7,没有弹出,栈中10 7都能看到4,num=2+2 ,4入栈后 Height_List:10 7 4

⑤12大于栈中全部元素,表示栈中所有元素看不到12,全部出栈,num不变,12入栈后 Height_List:12

⑥2 < 12,没有弹出,12能看到2,num=4+1

最后结果num=5

代码如下

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <stack>

using namespace std;

stack<long long> st;
int a[80005];

int main()
{
    int i,n;
    long long sum = 0;
    scanf("%d",&n);
    for(i = 0;i < n;i++)
        scanf("%d",&a[i]);
    for(i = 0;i < n;i++)
    {
        while(!st.empty() && st.top() <= a[i])	//如果栈顶元素小于即将入栈元素,那么就不断出栈
            st.pop();
        sum += st.size();//st.size()表示将要入栈的元素,它后面能看到他发型的奶牛个数
        st.push(a[i]);
    }
    printf("%lld\n",sum);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值