首页 > 代码库 > BZOJ 1660: [Usaco2006 Nov]Bad Hair Day 乱发节

BZOJ 1660: [Usaco2006 Nov]Bad Hair Day 乱发节

Description

Input

* Line 1: 牛的数量 N。

* Lines 2..N+1: 第 i+1 是一个整数,表示第i头牛的高度。

Output

* Line 1: 一个整数表示c[1] 至 c[N]的和。

题解:

BZOJ服务器好像有问题,图片很不好显示,看上面的就好了。

设牛i看到的最远位置是see[i]。

从前向后扫,求see值,设前面有一头牛A,后面有一头牛B,如果h[B]>=h[A],那么A这头牛就永远不可能成为B牛后面的牛的see值。

可以用单调栈维护这个单调性,

从前向后扫,当栈顶牛高度小于当前牛高度时退栈,之道栈顶牛高度>=当前牛高度,那么这个栈顶牛就是当前牛的see值,将当前牛入栈。

ans=∑(see[i]-i-1)。

代码:

#include<cstdio>#include<cstring>#include<algorithm>//by zrt//problem:using namespace std;const int MAXN=80005;struct N{    int x,w;    N(int a=0,int b=0){        x=a,w=b;    }};N stk[MAXN];int top;int see[MAXN];int n;int h[MAXN];int inf=1<<30;typedef long long LL;int main(){    #ifdef LOCAL    freopen("in.txt","r",stdin);    freopen("out.txt","w",stdout);    #endif    scanf("%d",&n);    for(int i=1;i<=n;i++){        scanf("%d",&h[i]);    }    stk[top++]=N(n+1,inf);    for(int i=n;i>=1;i--){        while(stk[top-1].w<h[i]) top--;        see[i]=stk[top-1].x;        stk[top++]=N(i,h[i]);    }    LL ans=0;    for(int i=1;i<=n;i++){        ans+=see[i]-i-1;    }    printf("%lld\n",ans);    return 0;}

BZOJ 1660: [Usaco2006 Nov]Bad Hair Day 乱发节