题目1550：分糖果

题目描述：

给从左至右排好队的小朋友们分糖果，
要求：
1.每个小朋友都有一个得分，任意两个相邻的小朋友，得分较高的所得的糖果必须大于得分较低的，相等则不作要求。
2.每个小朋友至少获得一个糖果。
求，至少需要的糖果数。

输入：

输入包含多组测试数据，每组测试数据由一个整数n（1<=n<=100000）开头，接下去一行包含n个整数，代表每个小朋友的分数Si（1<=Si<=10000）。

输出：

对于每组测试数据，输出一个整数，代表至少需要的糖果数。

样例输入：

3
1 10 1
3
6 2 3
2
1 1

样例输出：

4
5
2

#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 100010;

int a[N], cnt[N];

int main()
{
    int n;

    #ifndef ONLINE_JUDGE
        //freopen("d:\\OJ\\uva_in.txt", "r", stdin);
    #endif // ONLINE_JUDGE

    while (scanf("%d", &n) == 1) {
        for (int i = 0; i < n; i++) {
            scanf("%d", &a[i]);
            cnt[i] = 1;

            if (i - 1 >= 0) {
                if (a[i] > a[i - 1]) {
                    cnt[i] = cnt[i - 1] + 1;
                } else if (a[i] < a[i - 1]) {
                    for (int j = i - 1;  j >= 0 && a[j] > a[j + 1]; j--) {
                        cnt[j] = max(cnt[j], cnt[j + 1] + 1);
                    }
                }
            }
        }

        int sum = 0;
        for (int i = 0; i < n; i++) sum += cnt[i];

        printf("%d\n", sum);
    }
    return 0;
}