From 7d92af36a1fe2a57dab888aaf31ceefaea47fcef Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Khoa=20Hu=E1=BB=B3nh=20=28Daniel=29?= Date: Fri, 12 Apr 2024 23:54:39 +0700 Subject: [PATCH] Create main.go --- solution/42/main.go | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 solution/42/main.go diff --git a/solution/42/main.go b/solution/42/main.go new file mode 100644 index 0000000..ec76c1c --- /dev/null +++ b/solution/42/main.go @@ -0,0 +1,24 @@ +package main + +func trap(height []int) int { + n := len(height) + res := 0 + var st []int + + for r := 0; r < n; r++ { + for len(st) > 0 && height[st[len(st) - 1]] <= height[r] { + m := st[len(st) - 1] + st = st[:len(st) - 1] + + if len(st) > 0 { + l := st[len(st) - 1] + d := min(height[l], height[r]) + k := r - 1 - (l + 1) + 1 + res += k * (d - height[m]) + } + } + st = append(st, r) + } + + return res +}