poj 3250 单调栈(看到前面的牛的数量)

题意:一群高度不完全相同的牛从左到右站成一排,每头牛只能看见它右边的比它矮的牛的发型,若遇到一头高度大于或等于它的牛,则无法继续看到这头牛后面的其他牛。给出这些牛的高度,要求每头牛可以看到的牛的数量的和。
思路:单调栈(类似2559和3934),栈中按照递减顺序保存牛的高度,对于当前牛判断它入栈的位置。其实stack不必设置h域,stack只存下标即可,找栈中高度用s[下标]就行了。

看到另一种思路:把要求作一下转换,其实就是要求每头牛被看到的次数之和。这个也可以使用单调栈来解决。从左到右依次读取当前牛的高度,从栈顶开始把高度小于或等于当前牛的高度的那些元素删除,此时栈中剩下的元素的数量就是可以看见当前牛的其他牛的数量。把这个数量加在一起,就可以得到最后的答案了。按照这种思路写出的代码更简单。

#include <cstdio>
using namespace std;
#define N 80005
struct point{
    int id,h;
}stack[N];
int n,s[N];
long long  test(){
    int i,top=0;
    long long res = 0;
    stack[top].h = s[0];
    stack[top].id = 0;
    for(i = 1;i<=n;i++){
        while(top>=0 && s[i]>=stack[top].h){
            res += i-stack[top].id-1;
            top--;
        }
        stack[++top].h = s[i];
        stack[top].id = i;
    }
    return res;
}
int main(){
    int i;
    scanf("%d",&n);
    for(i = 0;i<n;i++)
        scanf("%d",&s[i]);
    s[n] = 1000000001;
    printf("%lld\n",test());
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值