From 1ec74c1574685b662e24f40416b9c5eea1a3c6b0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Khoa=20Hu=E1=BB=B3nh=20=28Daniel=29?= Date: Sun, 31 Mar 2024 23:44:06 +0700 Subject: [PATCH] Create main.go --- solution/2444/main.go | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 solution/2444/main.go diff --git a/solution/2444/main.go b/solution/2444/main.go new file mode 100644 index 0000000..77c4542 --- /dev/null +++ b/solution/2444/main.go @@ -0,0 +1,26 @@ +package main + +func countSubarrays(nums []int, minK int, maxK int) int64 { + var res int64 + left := 0 + pmin := -1 + pmax := -1 + + for right, num := range nums { + if num < minK || num > maxK { + left = right + 1 + pmin = -1 + pmax = -1 + } else { + if num == minK { + pmin = right + } + if num == maxK { + pmax = right + } + res += int64(max(0, min(pmin, pmax) - left + 1)) + } + } + + return res +}