PAT（甲级）2020年秋季考试 7-1 Panda and PP Milk (C++)

PP milk （盆盆奶）is Pandas’ favorite. They would line up to enjoy it as show in the picture. On the other hand, they could drink in peace only if they believe that the amount of PP milk is fairly distributed, that is, fatter panda can have more milk, and the ones with equal weight may have the same amount. Since they are lined up, each panda can only compare with its neighbor(s), and if it thinks this is unfair, the panda would fight with its neighbor.

Given that the minimum amount of milk a panda must drink is 200 ml. It is only when another bowl of milk is at least 100 ml more than its own that a panda can sense the difference.

Now given the weights of a line of pandas, your job is to help the breeder（饲养员）to decide the minimum total amount of milk that he/she must prepare, provided that the pandas are lined up in the given order.

Input Specification:

Each input file contains one test case. For each case, first a positive integer $n(\le 10^4)$ is given as the number of pandas. Then in the next line, $n$ positive integers are given as the weights (in kg) of the pandas, each no more than 200. the numbers are separated by spaces.

Output Specification:

For each test case, print in a line the minimum total amount of milk that the breeder must prepare, to make sure that all the pandas can drink in peace.

Sample Input:

10
180 160 100 150 145 142 138 138 138 140

Sample Output:

3000

Hint:

The distribution of milk is the following:

400 300 200 500 400 300 200 200 200 300

Solution:

// Talk is cheap, show me the code
// Created by Misdirection 2021-09-08 13:29:44
// All rights reserved.

#include <iostream>
#include <vector>
#include <cmath>

using namespace std;

int weights[10010];
int distribute[10010];

int main(){
    int n;
    scanf("%d", &n);

    int sum = 0;
    for(int i = 0; i < n; ++i) scanf("%d", &weights[i]);

    fill(distribute, distribute + 10010, 200);
    for(int i = 1; i < n; ++i){
        if(weights[i] > weights[i - 1]) distribute[i] = distribute[i - 1] + 100;
        else if(weights[i] == weights[i - 1]) distribute[i] = distribute[i - 1];
    }

    for(int i = n - 2; i >= 0; --i){
        if(weights[i] > weights[i + 1]) distribute[i] = max(distribute[i + 1] + 100, distribute[i]);
        else if(weights[i] == weights[i + 1]) distribute[i] = max(distribute[i + 1], distribute[i]);
    }

    for(int i = 0; i < n; ++i) sum += distribute[i];
    printf("%d\n", sum);

    return 0;
}