POJ3250--Bad Hair Day(栈)

题目大意:有n头牛,身高不一样,排成一排,每头牛可以看见后面一头牛的个数之和,如果相等就看不见,因为是看头发啦~,头发

在眼睛上面。嗯~出题人应该是这样想的,23333

 

分析:一眼就能看出来是用栈来做。我是从后往前做的,栈顶的元素小于当前身高,栈顶元素出栈,就说明当前的牛看得见栈顶的

牛,接着比较,直到栈顶的牛的身高比当前身高要高,那么就找到了当前牛能看见的最远距离。

不要问我为什么全部用%lld,因为我没用全用的时候,一直WA。。。


代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn = 80010;
long long h[maxn];
int R[maxn];
int st[maxn];
long long n;

int main() {
    while(~scanf("%lld", &n)) {
        for(int i = 0; i < n; i++)
            scanf("%lld", &h[i]);
        int t = 0;
        long long sum = 0;
        for(int i = n-1; i >= 0; i--) {
            while(t > 0 && h[st[t-1]] < h[i]) t--;
            R[i] = t == 0 ? n : st[t-1];
            st[t++] = i;
            sum += R[i]-1-i;
        }
        printf("%lld\n", sum);
    }
    return 0;
}


你可能感兴趣的:(POJ3250--Bad Hair Day(栈))