刷题31-收集雨水

原题链接

扩展题目一------种花时间

题目描述

给出n个数字，表示一个高程图，高程图中每一条的宽度为1，
请计算下雨之后这个地形可以存储多少水
例如
给出[0,1,0,2,1,0,1,3,2,1,2,1],返回6.

上面的高程图用数组[0,1,0,2,1,0,1,3,2,1,2,1]表示。
在这种情况下，6个单位的雨水（蓝色部分）被存储。

示例

输入：
[0,1,0,2,1,0,1,3,2,1,2,1]
输出：
6

参考解法

public class Main {
	public static void main(String[] args) {
		int a[] = {0,1,0,2,1,0,1,3,2,1,2,1};
		System.out.println(trap(a));
	}

	public static int trap(int[] a) {
		// 数组中最大值的下标
		int max_index = 0;
		// 最大值左侧的极大值
		int l = 0;
		// 最大值右侧的极大值
		int r = 0;
		int sum = 0;
		for (int i = 0; i < a.length; i++) {
			max_index = a[i] > a[max_index] ? i : max_index;
		}
//		System.out.println(max_index);
		// 左侧的和
		for (int i = 0; i < max_index; i++) {
			if (a[i] < l)
				sum += l - a[i];
			else
				l = a[i];
		}

		// 右侧的和
		for (int i = a.length-1; i > max_index; i--) {
			if (a[i] < r)
				sum += r - a[i];
			else
				r = a[i];
		}

		return sum;
	}
}