有一组不同高度的台阶,有一个整数数组表示,数组中每个数是台阶的高度,当开始下雨了(雨水足够多)台阶之间的水坑会积水多少呢?如下图,可以表示为数组[0,1,0,2,1,0,1,3,2,1,2,1],返回积水量6.
方法一:(两次遍历)
先在这个数组中找到最大值,然后从左右两边遍历。对于左边,只要当前的数字比下一个数字大,那么这个数字的右边就可以存水,按照这个思路去分析就可以了,对于右边,只要当前的数字比下一个数字大,那么这个数字的左边就可以存水。
#include <stdio.h>
#include <stdlib.h>
int main()
{
int a[12]={0,1,0,2,1,0,1,3,2,1,2,1};
int length=12,sum=0,i;
int arr_max,arr_max_pos,arr_max_left,arr_max_right;
arr_max=a[0];//定义数组的最大值
arr_max_pos=0;//定义数组最大值的下标
for(i=1;i<length;i++)
{
if(arr_max<a[i])
{
arr_max=a[i];
arr_max_pos=i;
}
}
arr_max_left=a[0];//定义从最左边开始遍历的极大值
for(i=1;i<arr_max_pos;i++)
{
if(arr_max_left<a[i])
arr_max_left=a[i];
else if(arr_max_left>a[i])
sum+=(arr_max_left-a[i]);
}
arr_max_right=a[length-1];//定义从最右边开始遍历的极大值
for(i=length-2;i>arr_max_pos;i--)
{
if(arr_max_right<a[i])
arr_max_right=a[i];
else if(arr_max_right>a[i])
sum+=(arr_max_right-a[i]);
}
printf("%d",sum);
}
方法二:(一次遍历)
整个的原理和方法一是一样的,就是从左右两边同时遍历,这样一次就可以完成。
#include <stdio.h>
#include <stdlib.h>
int getArea(int array[], int size)
{
if (size < 3)
{
return 0;
}
int i = 0;
int j = size - 1;
int maxLeft = array[i];
int maxRight = array[j];
int sum = 0;
while (i <= j)
{
if (maxLeft < maxRight)
{
if (maxLeft < array[i])
{
maxLeft = array[i];
}
else if (maxLeft > array[i])
{
sum += (maxLeft - array[i]);
}
i++;
}
else
{
if (maxRight < array[j])
{
maxRight = array[j];
}
else if (maxRight > array[j])
{
sum += (maxRight - array[j]);
}
j--;
}
}
return sum;
}
int main()
{
int result;
int a[12] = {0, 1, 0, 2, 1, 0, 1, 3, 2, 1, 2, 1};
result=getArea(a,12);
printf("%d",result);
}