Sync LeetCode submission Runtime - 128 ms (52.48%), Memory - 12.3 MB …

…(23.62%)
hucancode · Mar 29, 2024 · 0f30df5 · 0f30df5
1 parent 298934f
commit 0f30df5
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/leetcode/3213-count-subarrays-where-max-element-appears-at-least-k-times/solution.go b/leetcode/3213-count-subarrays-where-max-element-appears-at-least-k-times/solution.go
@@ -0,0 +1,17 @@
+func countSubarrays(nums []int, k int) int64 {
+    m := slices.Max(nums)
+    var q []int
+    var ret int64 = 0
+    for i, x := range nums {
+        if x == m {
+            q = append(q, i)
+        }
+        for len(q) > k {
+            q = q[1:]
+        }
+        if len(q) >= k {
+            ret += int64(q[0]) + 1
+        }
+    }
+    return ret
+}