-
Notifications
You must be signed in to change notification settings - Fork 0
/
42-trappingRainWater.js
84 lines (72 loc) · 1.6 KB
/
42-trappingRainWater.js
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
/**
* @param {number[]} height
* @return {number}
*/
var trap = function (height) {
const trimmedHeights = trimHeights(height);
const maxIndex = trimmedHeights.reduce(
(acc, el, index) => (el > trimmedHeights[acc] ? index : acc),
0
);
const leftArray = trimmedHeights.slice(0, maxIndex + 1);
const rightArray = trimmedHeights
.slice(maxIndex, trimmedHeights.length)
.reverse();
const leftCount = leftArray.reduce(
(acc, el) => {
if (el > acc.leftHeight) {
return {
count: acc.count,
leftHeight: el,
};
}
return {
count: acc.count + (acc.leftHeight - el),
leftHeight: acc.leftHeight,
};
},
{
count: 0,
leftHeight: 0,
}
).count;
const rightCount = rightArray.reduce(
(acc, el) => {
if (el > acc.leftHeight) {
return {
count: acc.count,
leftHeight: el,
};
}
return {
count: acc.count + (acc.leftHeight - el),
leftHeight: acc.leftHeight,
};
},
{
count: 0,
leftHeight: 0,
}
).count;
return leftCount + rightCount;
};
const trimHeights = (heights) => {
const trimmedHeights = [...heights];
for (let i = 0; i < heights.length; i++) {
if (heights[i] === 0) {
trimmedHeights.shift();
} else {
break;
}
}
for (let i = heights.length - 1; i >= 0; i--) {
if (heights[i] === 0) {
trimmedHeights.pop();
} else {
break;
}
}
return trimmedHeights;
};
const height = [0, 1, 0, 2, 1, 0, 1, 3, 2, 1, 2, 1];
console.log(trap(height));