Description
十年前,北湖还只是一个深坑,未完成蓄水工作。为了确保蓄水工作的顺利进行,我们需要对北湖的蓄水量进行粗略估计。
为了简化运算,我们假设北湖的地面是一维的,每一块宽度都为1,高度是非负整数,那么可以用一个数组来表达一块地面。
例如数组[0,1,0,2,1,0,1,3,2,1,2,1]可以用来表示下图地面:
图中绿色代表地面部分,蓝色部分代表蓄水部分,蓄水量为6。
Input
样例输入有多组。
第一行输入整数 表示有组用例;
接下来,对于每组用例,输入一个正整数表示地面总宽度为。
接下来一行是个数,用空格隔开,表示地面的高度。
Output
对于每个用例输出一行一个数字,表示蓄水总量。
测试输入 | 期待的输出 | 时间限制 | 内存限制 | 额外进程 | |
---|---|---|---|---|---|
测试用例 1 | 以文本方式显示
| 以文本方式显示
| 1秒 | 64M | 0 |
思路
某位置最终蓄水量由其左右两侧的最高处的较小值决定。
可以在定义两个数组leftMax、rightMax,其中,leftMax[ i ]表示 i 左侧的最高点;rightMax[ i ] 表示 i 右侧的最高点。
然后每一处的储水量可以表示为: min(leftMax[ i ], rightMax[ i ]) - h[ i ]
蓄水总量为各个位置蓄水量之和。
注意事项
- 定义数组都是在主函数外,也就是定义成全局变量。
因为主函数内的栈内存比全局内存要少,大的内存申请定义在主函数内很容易爆栈(爆内存)。 - 第一个位置的左侧最高值和最后一个位置的右侧最高值,即leftMax[0]和rightMax[i-1]需要提前单独赋值。
- 由于是循环输入每组测试用例,在循环体内要记得变量重新初始化,不然每次循环会在上次循环的基础上继续计算。
整体代码
#include <stdio.h>
long long leftMax[100000] = { 0 };
long long rightMax[100000] = { 0 };
long long h[100000] = { 0 };
int main() {
int t, n;
scanf("%d", &t);
while (t--) {
scanf("%d", &n);
for (int i = 0; i < n; i++) {
scanf("%lld", &h[i]);
}
leftMax[0] = h[0];
rightMax[n - 1] = h[n - 1];
for (int i = 1; i < n - 1; i++) {
if (h[i] < leftMax[i - 1]) leftMax[i] = leftMax[i - 1];
else leftMax[i] = h[i];
}
for (int i = n - 2; i > 0; i--) {
if (h[i] < rightMax[i + 1]) rightMax[i] = rightMax[i + 1];
else rightMax[i] = h[i];
}
long long sum = 0;
for (int i = 1; i < n - 1; i++) {
if (rightMax[i] <= leftMax[i])
sum += rightMax[i] - h[i];
else sum += leftMax[i] - h[i];
}
printf("%lld\n", sum);
sum = 0;
}
return 0;
}