首页 > 其他 > 详细

POJ 3250 Bad Hair Day(单调栈)

时间:2014-09-22 23:15:04      阅读:425      评论:0      收藏:0      [点我收藏+]

题目地址:POJ 3250

初学单调栈。多校和网络赛已经碰到两次了。

单调栈的原理简单的不能再简单了。。就是让栈里的元素从栈顶到栈底呈单调性。

比如说递增单调栈。

每次放进一个数的时候,如果栈顶的数小于要放的数,就把栈顶的数pop出来使得栈里保持单调性。

对于这道题来说,就从右往左开始遍历,建一个递增单调栈。那么每次pop出来的就是当前的牛可以看到的牛数。然后累加即可。

代码如下:

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <set>
#include <algorithm>
#include <stack>
using namespace std;
#define LL long long
LL a[100000], dp[100000];
stack<int>q;
int main()
{
    int n, i, j;
    LL sum=0;
    scanf("%d",&n);
    for(i=0;i<n;i++)
    {
        scanf("%lld",&a[i]);
    }
    memset(dp,0,sizeof(dp));
    for(i=n-1;i>=0;i--)
    {
        dp[i]=1;
        while(!q.empty()&&a[q.top()]<a[i])
        {
            dp[i]+=dp[q.top()];
            sum+=dp[q.top()];
            q.pop();
        }
        q.push(i);
    }
    /*for(i=0;i<n;i++)
    {
        printf("%d ",dp[i]);
    }*/
    printf("%lld\n",sum);
    return 0;
}


POJ 3250 Bad Hair Day(单调栈)

原文:http://blog.csdn.net/scf0920/article/details/39483569

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!