5. 北湖深坑

Description

十年前,北湖还只是一个深坑,未完成蓄水工作。为了确保蓄水工作的顺利进行,我们需要对北湖的蓄水量进行粗略估计。

为了简化运算,我们假设北湖的地面是一维的,每一块宽度都为1,高度是非负整数,那么可以用一个数组来表达一块地面。

例如数组[0,1,0,2,1,0,1,3,2,1,2,1]可以用来表示下图地面:

图中绿色代表地面部分,蓝色部分代表蓄水部分,蓄水量为6。

Input

样例输入有多组。

第一行输入整数 表示有组用例;

接下来,对于每组用例,输入一个正整数表示地面总宽度为。

接下来一行是个数,用空格隔开,表示地面的高度。

Output

对于每个用例输出一行一个数字,表示蓄水总量。

测试输入 期待的输出 时间限制 内存限制 额外进程
测试用例 1 以文本方式显示
  1. 2↵
  2. 12↵
  3. 0 1 0 2 1 0 1 3 2 1 2 1↵
  4. 5↵
  5. 5 2 3 2 4↵
以文本方式显示
  1. 6↵
  2. 5↵
1秒 64M 0

思路

某位置最终蓄水量由其左右两侧的最高处的较小值决定。

可以在定义两个数组leftMax、rightMax,其中,leftMax[ i ]表示 i 左侧的最高点;rightMax[ i ] 表示 i 右侧的最高点。

 然后每一处的储水量可以表示为: min(leftMax[ i ], rightMax[ i ]) - h[ i ]

蓄水总量为各个位置蓄水量之和。

注意事项

  1. 定义数组都是在主函数外,也就是定义成全局变量。
    因为主函数内的栈内存比全局内存要少,大的内存申请定义在主函数内很容易爆栈(爆内存)。
  2. 第一个位置的左侧最高值和最后一个位置的右侧最高值,即leftMax[0]和rightMax[i-1]需要提前单独赋值。
  3. 由于是循环输入每组测试用例,在循环体内要记得变量重新初始化,不然每次循环会在上次循环的基础上继续计算。

整体代码

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

你可能感兴趣的:(程序设计方法与实践,算法,数据结构,c语言)