POJ 3183 Stump Removal

水题一只。

我是模拟了一下那个过程，但这样其实麻烦了(还用了goto语句。。太丢脸了。。)

可以直接找出数组中h[i]>=h[i+1] && h[i] <= h[i-1]的数的个数，就是答案，biaaib大神告诉我的额。。。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int h[60000];
int mem[60000];
int main()
{
    int pr,n,i,j,lm;
    bool iod;
    scanf("%d",&n);
    for (i=0; i<n; i++)
    {
        scanf("%d",h+i);
    }
    h[n]=h[n-1];
    h[n+1]=-1;
    lm=0;
    i=0;
    while (h[i] == h[i+1])
    {
        mem[lm]=i;
        i++;
        lm++;
    }
    if (i == n)
    {
        goto end;
    }
    if (h[i] < h[i+1])
    {
        iod=true;
    }
    else
    {
        mem[lm]=i;
        lm++;
        iod=false;
    }
    for (i++; i<n; )
    {
        if (iod == false && h[i] <= h[i+1])
        {
            iod=true;
            i++;
        }
        else if (iod == true && h[i] == h[i+1])
        {
            while (h[i] == h[i+1])
            {
                mem[lm]=i;
                lm++;
                i++;
            }
        }
        else if (iod == true && h[i] > h[i+1])
        {
            mem[lm]=i;
            lm++;
            iod=false;
            i++;
        }
        else
            i++;
    }
end:
    for (i=0; i<lm; i++)
    {
        printf("%d\n",mem[i]+1);
    }
    return 0;
}