之前面试碰到的一道算法题,题目大概是: 假设最左边有一个无限高的墙,右侧有一排不同高度的紧挨着的墙,求下雨后雨水注满在墙内的量。 例如在下图中,例如给定数组为 {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];
}
}