雨水灌墙算法实现

之前面试碰到的一道算法题,题目大概是: 假设最左边有一个无限高的墙,右侧有一排不同高度的紧挨着的墙,求下雨后雨水注满在墙内的量。 例如在下图中,例如给定数组为 {1, 2, 9, 4, 5, 5, 3, 6, 4, 3, 2, 4}, 则应该输出 25


由于本人平时算法底子较薄,刚看到题目时没有在很短的时间内想到思路,结果越紧张越没思路,越没思路越紧张,导致最后虽然写出,但只在某种情况下是正确的。回到家第二天醒来心有不甘,花了几分钟时间就写对了,但当时写的不是最优解。

先说一下我最开始的思路吧,其实对于这种题目,我们刚开始不要去想代码怎么写,而是先用一种口算的方法,找到一种规律。例如对于这道题,我刚开始就想,雨水的量肯定是较高的两面墙中间的部分,而且是以两侧墙的最矮的那个墙为基准。

以下是我刚开始的思路:

第一步:从左往右,如果一直上升,则继续向下找,如果碰到矮的墙,例如图中的第 3 号墙,则记下 ‘2’,继续走,每次记下前,必须要有这种情况: 先下降,再上升,那么之前下降部分就是要累积的部分。所以第二次我记下的就会是 序号'5', 到序号 '7' 的时候,循环抛出之前记下的比当前值小的,因为对于第 '3' 号墙来说,我们要计算的是和 '7’ 号墙的差距,而不是和 '5' 号墙的差距。  最后循环完成时,我记下的为 '2' '7' '11' ,用 list 保存。  抛出时 pop_back. 插入时 push_back.

第二部,遍历 list, 通过 front() 取得当前最高值,累积当前最高值之前的差值就可以了。例如对于 '2' 号墙,之前累积2个墙的高度差,然后更改当前累积的初始 index 为 '2'+1,下一次取出的是 '7' 号墙,从 index 开始一直算到 ‘7’ 号墙的前一个墙。 循环到 list 为空即可。


下面是我第二天早上写出的代码。 但是写完之后感觉解法不好,肯定是有 O(n) 的做法,我这种做法因为需要循环 pop_back,时间复杂度大于 O(n)。  空间复杂度大于O(1).     这段代码若有问题还请指出,多谢。

int data[] = { 1, 2, 9, 4, 5, 5, 3, 6, 4, 3, 2, 4};
int nLen = sizeof(data) / sizeof(int);

list<int> listIndex;
int upIndex = -1;
for (int i = 0; i < nLen - 1; i++)
{
	if (data[i] > data[i + 1] && upIndex > 0)
	{
		while (!listIndex.empty() && data[listIndex.back()] < data[i])
		{
			listIndex.pop_back();
		}
		listIndex.push_back(i);
		upIndex = -1;
	}
	else if (data[i] < data[i + 1])
	{
		upIndex = i + 1;
	}
}
if (upIndex > 0)
{
	listIndex.push_back(upIndex);
}

int startIndex = 0;
int nRet = 0;
while (!listIndex.empty())
{
	int nValue = 0;
	for (int i = startIndex; i < listIndex.front(); i++)
	{
		if (data[listIndex.front()] - data[i] > 0)
		{
			nValue += data[listIndex.front()] - data[i];
		}
	}
	nRet += nValue;
	startIndex = listIndex.front() + 1;
	listIndex.pop_front();
}

 其实想一想,我们要计算的其实就是当前较高的一个墙,和另一侧更高的墙之间的部分。 对于 '2' 号墙,不管它高度是 10 还是 20, 都不会影响 '7' 号墙和 '11' 号墙之间的雨水量。因为最左侧墙无限高,那么我们从右侧开始看,保持一个局部最大值 max,刚开始是 4, 那么往左移动时,只要小于4的都可以计算差值,一旦碰到大于 4 的,那么就不能继续计算差值了,此时需要更新 max 为 6,继续往左计算差值,再次会更新 max 为 9,直到游标为 0 结束。  时间复杂度为 O(n),空间复杂度 O(1)


int data[] = { 1, 2, 9, 4, 5, 5, 3, 6, 4, 3, 2, 4};
int nLen = sizeof(data) / sizeof(int);	
int rightMaxIndex = nLen - 1;
int rightMax = data[rightMaxIndex];
int nValue = 0;
for (int i = rightMaxIndex - 1; i >=0; i--)
{
    if (data[i] > rightMax)
    {
	rightMax = data[i];
	rightMaxIndex = i;
    }
    else
    {
	nValue += rightMax - data[i];
    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值