雨水灌墙算法实现

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