首页 > 代码库 > Codeforces 319B. Psychos in a Line【栈】

Codeforces 319B. Psychos in a Line【栈】

题目大意:

一串数列,每一个值如果大于相邻右一位的值的话,那么就可以把右边这个值“吃掉”(右一位消失,原来的值不变),问需要吃多少次才能到达无法再吃的状态。


做法:

利用栈。遍历一遍数组,处理每个值的时候,如果栈顶的元素小于该值,那么将其弹出,知道栈顶元素大于该值或者栈为空,栈内的每个元素记录下一个属性:他是在第几次被“吃掉”,进栈的新元素的被吃次数就是它弹出去的元素中的属性的最大值+1,如果没有弹任何值则为1;如果栈空,则值为0;

代码:
#include <iostream>
#include <cstdio>
#define N 100010
using namespace std;
int a[N];
int ans,n;
struct aa
{
    int x,eat;
}que[N];
int main()
{
    int tail=0;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%d",a+i);
    int i=1;
    while(a[i-1]<a[i] && i<n) i++;
    que[tail].x=a[i-1],que[tail++].eat=0;
    for(;i<n;i++)
    {
        if(a[i]<que[tail-1].x)
        {
            que[tail].x=a[i],que[tail++].eat=1;
            ans=max(ans,que[tail-1].eat);
            continue;
        }
        int ma=0;
        while(tail>0 && a[i]>que[tail-1].x) ma=max(ma,que[tail-1].eat),tail--;
        if(tail==0) que[tail].eat=0;
        else que[tail].eat=ma+1;
        que[tail++].x=a[i];
        ans=max(ans,que[tail-1].eat);
    }

    cout<<ans<<endl;
    return 0;
}

Codeforces 319B. Psychos in a Line【栈】