luogu P1317 低洼地

 

题目描述

一组数，分别表示地平线的高度变化。高度值为整数，相邻高度用直线连接。找出并统计有多少个可能积水的低洼地？

如图：地高变化为 0 1 0 2 1 2 0 0 2 0

输入输出格式

输入格式：

 

两行，第一行n,表示有n个数。第2行连续n个数表示地平线高度变化的数据，保证首尾为0。(3<=n<=10000,0<=高度<=1000)

 

输出格式：

 

一个数，可能积水低洼地的数目。

 

输入输出样例

输入样例#1：

10
0 1 0 2 1 2 0 0 2 0

输出样例#1：

3

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>

using namespace std;
const int N = 1e4 + 10;

int n, a[N];

inline int read()
{
    int x = 0, f = 1; char c = getchar();
    while(c < '0' || c > '9'){if(c == '-')f = -1;c = getchar();} 
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}

int main()
{
    memset(a, -1, sizeof(a));
    n = read();
    for(int i = 1; i <= n; i ++)
        a[i] = read();
    int answer = 0;
    for(int i = 1; i <= n - 2; i ++)
    {
        if(a[i] > a[i + 1] && a[i + 2] > a[i + 1]) answer ++; 
        if(a[i + 1] == a[i + 2] && a[i + 1] < a[i] && i <= n - 2)
        {
            int j = i + 2;
            while(a[j] == a[j - 1]) j ++;
            if(a[j] > a[j - 1]) answer ++;
            i = j - 2;
        }
    }
    printf("%d", answer);
    return 0;
}

/*
11
10 9 8 8 8 7 8 2 1 1 2
*/