题目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
, cnt
;

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;
}