分糖果问题进阶问题

分糖果问题进阶问题

题目描述

一群孩子做游戏，现在请你根据游戏得分来发糖果，要求如下：

1. 每个孩子不管得分多少，起码分到一个糖果。

2. 任意两个相邻的孩子之间，得分较多的孩子必须拿多一些糖果。

3. 任意两个相邻的孩子之间的得分如果一样多，糖果数必须相同

给定一个数组arr代表得分数组，请返回最少需要多少糖果。

输入描述：

第一行一个整数N表示数组大小
接下来一行N个整数表示数组内的元素

输出描述：

输出一个整数表示答案

示例1

输入

3
1 2 2

输出

5

说明

最优分配方案为1, 2, 2

示例2

输入

13
0 1 2 3 3 3 2 2 2 2 2 1 1

输出

30

说明

最优的分配方案为
1 2 3 4 4 4 2 2 2 2 2 1 1

备注：

$1⩽N⩽10^5$
$0⩽a_i⩽10^5$

---

题解：

同样的是 分糖果问题 思路，正反两次遍历。与基础版问题不同的是，进阶版多了一个要求：相邻分数相同的孩子分得的糖果必须相同。那么我们可以在基础版的解法上这么做：

• 正向遍历，若 a[i] == a[i - 1] ，f[i] = f[i - 1]
• 反向遍历，若 a[i] == a[i + 1]，f[i] = f[i + 1]

其余部分不需要改变，这样就好了。。。

代码：

#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 100000;

int n;
int a[N];
int f[N];

int main(void) {
    scanf("%d", &n);
    for ( int i = 0; i < n; ++i ) {
        scanf("%d", a + i );
        f[i] = 1;
        if ( i && a[i] >= a[i - 1] ) {
            f[i] = f[i - 1];
            f[i] += ( a[i] > a[i - 1] );
        }
    }
    int sum = f[n - 1];
    for ( int i = n - 2; i >= 0; --i ) {
        if ( a[i] > a[i + 1] && f[i] <= f[i + 1] ) f[i] = f[i + 1] + 1;
        else if ( a[i] == a[i + 1] ) f[i] = f[i + 1];
        sum += f[i];
    }
    printf("%d\n", sum);
    return 0;
}