PAT乙级 1119 胖达与盆盆奶

大熊猫，俗称“胖达”，会排队吃盆盆奶。它们能和谐吃奶的前提，是它们认为盆盆奶的分配是“公平”的，即：更胖的胖达能吃到更多的奶，等胖的胖达得吃到一样多的奶。另一方面，因为它们是排好队的，所以每只胖达只能看到身边胖达的奶有多少，如果觉得不公平就会抢旁边小伙伴的奶吃。

已知一只胖达每次最少要吃 200 毫升的奶，当另一份盆盆奶多出至少 100 毫升的时候，它们才能感觉到是“更多”了，否则没感觉。

现在给定一排胖达的体重，请你帮饲养员计算一下，在保持给定队形的前提下，至少应该准备多少毫升的盆盆奶？

输入格式：

输入首先在第一行给出正整数 n（≤104），为胖达的个数。随后一行给出 n 个正整数，表示 n 只胖达的体重（公斤）。每个数值是不超过 200 的正整数，数字间以空格分隔。

输出格式：

在一行中输出至少应该准备多少毫升的盆盆奶。

输入样例：

10
180 160 100 150 145 142 138 138 138 140

输出样例：

3000

样例解释：

盆盆奶的分配量顺序为：

400 300 200 500 400 300 200 200 200 300

AC代码：

#include <bits/stdc++.h>
using namespace std;

int main() {

	int num;
	cin >> num;

	int weiths[num + 2], foods[num + 2];
	weiths[0] = 0;
	weiths[num + 1] = 0;
	foods[0] = 0;
	foods[num + 1] = 0;

	for (int i = 1; i < num + 1; i ++ ) {
		cin >> weiths[i];
	}

	for (int i = 1; i < num + 1; i ++ ) {
		foods[i] = 200;
	}


	while (true) {
		bool isOK = true;
		for (int i = 1; i < num + 1; i ++ ) {
			if (weiths[i] == weiths[i - 1]) {
				if (foods[i] != foods[i - 1]) {
					isOK = false;
				}
			}
			if (weiths[i] == weiths[i + 1]) {
				if (foods[i] != foods[i + 1]) {
					isOK = false;
				}
			}
			if (weiths[i] > weiths[i + 1]) {
				if (foods[i] <= foods[i + 1]) {
					isOK = false;
				}
			}
			if (weiths[i] > weiths[i - 1]) {
				if (foods[i] <= foods[i - 1]) {
					isOK = false;
				}
			}
		}
		if (isOK) {
			break;
		}

		for (int i = 1; i < num + 1; i ++ ) {
			if (weiths[i] == weiths[i - 1]) {
				if (foods[i] != foods[i - 1]) {
					foods[i] += 100;
					continue;
				}
			}
			if (weiths[i] == weiths[i + 1]) {
				if (foods[i] != foods[i + 1]) {
					foods[i] += 100;
					continue;
				}
			}
			if (weiths[i] > weiths[i + 1]) {
				if (foods[i] <= foods[i + 1]) {
					foods[i] += 100;
					continue;
				}
			}
			if (weiths[i] > weiths[i - 1]) {
				if (foods[i] <= foods[i - 1]) {
					foods[i] += 100;
					continue;
				}
			}
		}
	}

	long long sum = 0;
	for (int i = 1; i < num + 1; i ++ ) {
		sum += foods[i];
	}

	cout << sum;
	return 0;
}