ZZULI 2400小G的日常之接雨水
**
题目描述
下雨了,小G想到了一个问题,给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
你能帮帮小G吗?
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。
输入
第一行,输入一个数n(1 <= n <= 1,000,000)
第二行,输入n个数 (这n个数小于1e9)
输出
可以接多少雨水。
样例输入 Copy
12
0 1 0 2 1 0 1 3 2 1 2 1
样例输出 Copy``
6
思路:一块柱子能接水的量取决于它左右两边最高的柱子中较短的一个。
我们进行两次for循环,第一次找到最高的柱子,这样对于它左边的柱子相当于确定了右边的最高值,对于它右边的柱子相当于确定了左边的最高值,以最高的柱子的位置为分割点,将数组分成两部分。
对于左半部分,不断往右逼近,更新左边最大值;对于右半部分,不断往左逼近,更新右边最大值。累加求乘水量和。
原文:https://blog.csdn.net/Scarlett_Guan/article/details/80083487
import java.util.*;
public class Main{
public static void main(String args[]) {
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
int a[]=new int[n+1];
int max=-1;
int maxi=-1;
long sum=0;
for(int i=1;i<=n;i++)
{
a[i]=sc.nextInt();
if(a[i]>max)
{
max=a[i];
maxi=i;
}
}
int x=a[1];
for(int i=2;i<maxi;i++)
{
if(x>a[i])
sum+=x-a[i];
else
x=a[i];
}
x=a[n];
for(int i=n-1;i>maxi;i--)
{
if(x>a[i])
sum+=x-a[i];
else
x=a[i];
}
System.out.println(sum);
}
}