POJ 3250 Bad Hair Day(单调栈)

标签：acm   c语言   算法   编程   单调栈   

题目地址：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(单调栈)

标签：acm   c语言   算法   编程   单调栈   

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