100天iOS数据结构与算法实战 Day08 – 栈的算法实战 TrappingRainWater

本文链接：https://blog.youkuaiyun.com/ios8988/article/details/89278846

看到这里是不是有点印象了，之前有关一个栈的算法。只不过这里的各种括号换成了bar，然后可以存储的水的坑就被每一对匹配bar夹着，所以我们只要找出每对匹配的bar并且计算相加即可。

栈不为空，current指向的高度大于栈顶高度，所以栈顶元素 height [ 0 ] 出栈，栈现在是空则break跳出while循环体，height [ 1 ]入栈，current 后移。

由于栈不为空且current 指向的高度大于栈顶高度，栈顶 height [ 2 ] 出栈。计算 height [ 3 ] 和新的栈顶之间的水也就是蓝色的水。计算完之后继续判断 current 和新的栈顶的关系。

由于 current 指向的高度大于栈顶 height [ 5 ] 的高度，将栈顶 height [ 5 ] 出栈，然后计算 current 指向的墙和新栈顶 height [ 4 ] 之间的水。

计算完之后继续判断 current 的指向和新栈顶的关系。此时 height [ 6 ] 不大于栈顶 height [ 4 ] ，所以将 height [ 6 ] 入栈。 current 后移。

由于 current 指向的高度大于栈顶高度，将栈顶 height [ 6 ] 出栈。计算和新的栈顶 height [ 4 ] 组成两个边界中的水。然后判断 current 和新的栈顶 height [ 4 ] 的关系，依旧是大于，所以把 height [ 4 ] 出栈。计算current 和新的栈顶 height [ 3 ] 之间的水。

注意：其实不停的出栈，可以看做是在找与 7 匹配的墙，也就是 3 。

然后判断 current 和新的栈顶 height [ 3 ] 的关系，依旧是大于，所以把 height [ 3 ] 出栈，栈空。将 current 指向的 height [ 7 ] 入栈。current 后移。

- (int)trapNSArray *)inputArray
{
if (inputArray.count == 0)
{
return 0;
}
DSStack *newStack = [[DSStack alloc] initWithSize:15];
int sumWater = 0, current = 0;
while (current < inputArray.count)
{
//1
while (![newStack isEmpty] && ([inputArray[current] intValue] > [inputArray[[[newStack peek] intValue]] intValue]))
{
//2
int top = [[newStack peek] intValue];
[newStack popLastObject];
//3
if ([newStack isEmpty])
{
break;
}
//4
int distance = current - [[newStack peek] intValue] - 1;
NSNumber *currentN = [NSNumber numberWithInt:[inputArray[current] intValue]];
NSNumber *peekN = [NSNumber numberWithInt:[inputArray[[[newStack peek] intValue]] intValue]];
NSNumber *minObj;
NSComparisonResult r = [currentN compare:peekN];
if (r == NSOrderedAscending)
{
minObj = currentN;
}
else if(r == NSOrderedSame)
{
minObj = currentN;
}
else if(r == NSOrderedDescending)
{
minObj = peekN;
}
//5
int bounded_height = [minObj intValue] - [inputArray[top] intValue];
//6
sumWater += distance * bounded_height;
}
//7
[newStack push:@(current++)];
}
return sumWater;
}