糟糕的一天【栈】

题目大意:

农民约翰的某N头奶牛正在过乱头发节!由于每头牛都意识到自己凌乱不堪的发型,约翰希望统计出能够看到其他牛的头发的牛的数量。

每一头牛i有一个高度所有N头牛面向东方排成一排,牛N在最前面,而 牛1在最后面.第i头牛可以看到她前面的那些牛的头,只要那些牛的高度严格小于她的高度,而且 中间没有比hi高或相等的奶牛阻隔.

让N表示第i头牛可以看到发型的牛的数量;请输出Ci的总和。

思路:

很明显,这道题是一个单调栈。 
将奶牛一个一个进栈,同时保证栈里面的数字单调递减(后面必须看得见前面),每次进栈之后求出最后面一头牛能看见前面的牛的数量,输出答案即可。 
时间复杂度:O(n)

代码:

#include <cstdio>
#include <iostream>
#define fre(x) freopen(#x".in","r",stdin),freopen(#x".out","w",stdout);
using namespace std;

long long s[100011],n,a[100011],sum,tail;

int main()
{
    fre(badhair);
    cin>>n;
    tail=0;
    a[0]=1e17;
    for (long long i=1;i<=n;i++)
     cin>>a[i];
    for (long long i=1;i<=n;i++)
    {
        while (a[i]>=s[tail]&&tail) tail--;  //模拟栈(可以用queue)
        tail++; 
        s[tail]=a[i];  //进栈
        sum=sum+tail-1;  //求出答案
    }
    cout<<sum<<"\n";
    return 0;
}

原文地址:https://www.cnblogs.com/hello-tomorrow/p/9323734.html