/
042_Trapping_Rain_Water.java
117 lines (111 loc) · 3.33 KB
/
042_Trapping_Rain_Water.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
class Solution {
class Bar {
int h, p;
Bar(int h, int p) {
this.h = h;
this.p = p;
}
}
public int trap(int[] height) {
int ans = 0;
Stack<Bar> stack = new Stack<Bar>();
for (int i = 0; i < height.length; i ++) {
while (!stack.isEmpty() && height[i] >= stack.peek().h) {
Bar tmp = stack.pop();
if (!stack.isEmpty()) {
int h = Math.min(height[i] - tmp.h, stack.peek().h - tmp.h);
ans += h * (i - stack.peek().p - 1);
}
}
stack.push(new Bar(height[i], i));
}
return ans;
}
}
class Solution {
public int trap(int[] height) {
Stack<Integer> stack = new Stack<Integer>();
int ans = 0;
for (int i = 0; i < height.length; i ++) {
while (!stack.isEmpty() && height[i] >= height[stack.peek()]) {
int prev = stack.pop();
if (!stack.isEmpty()) {
int tmp = Math.min(height[i] - height[prev],
height[stack.peek()] - height[prev]);
ans += (i - stack.peek() - 1) * tmp;
}
}
stack.push(i);
}
return ans;
}
}
class Solution {
public int trap(int[] height) {
int ans = 0;
int l = 0, r = height.length - 1;
int leftMax = 0, rightMax = 0;
while (l < r) {
if (height[l] < height[r]) {
if (height[l] > leftMax)
leftMax = height[l];
else ans += leftMax - height[l];
l ++;
} else {
if (height[r] > rightMax)
rightMax = height[r];
else ans += rightMax - height[r];
r --;
}
}
return ans;
}
}
// new stack
class Solution {
public int trap(int[] height) {
Stack<Integer> stack = new Stack<>();
int ans = 0;
for (int i = 0; i < height.length; i ++) {
while (!stack.isEmpty() && height[i] >= height[stack.peek()]) {
int cur = height[stack.pop()];
if (!stack.isEmpty())
ans += (i - stack.peek() - 1) * (Math.min(height[i], height[stack.peek()]) - cur);
}
stack.push(i);
}
return ans;
}
}
// new two scan
class Solution {
public int trap(int[] height) {
int[] water = new int[height.length];
int bar = 0, ans = 0;
for (int i = 0; i < height.length; i ++) {
bar = Math.max(bar, height[i]);
water[i] = bar;
}
bar = 0;
for (int i = height.length - 1; i >= 0; i --) {
bar = Math.max(bar, height[i]);
water[i] = Math.min(water[i], bar);
ans += water[i] - height[i];
}
return ans;
}
}
// one scan two pointer
class Solution {
public int trap(int[] height) {
int ans = 0, l = 0, r = height.length - 1;
int barL = 0, barR = 0;
while (l <= r) {
barL = Math.max(barL, height[l]);
barR = Math.max(barR, height[r]);
if (barL < barR) ans += barL - height[l ++];
else ans += barR - height[r --];
}
return ans;
}
}