Leetcode 2962. Count Subarrays Where Max Element Appears at Least K Times

  • [Leetcode 2962. Count Subarrays Where Max Element Appears at Least K Times](#Leetcode 2962. Count Subarrays Where Max Element Appears at Least K Times)
    • [1. 解题思路](#1. 解题思路)
    • [2. 代码实现](#2. 代码实现)

1. 解题思路

这一题思路上同样很直接,就是找到最大的元素所在的全部的位置坐标,然后分别考虑取 i i i到 i + k i+k i+k个最大元素时substring左右可以选择的起止点位置数目,相乘累加即可。

2. 代码实现

给出python代码实现如下:

python 复制代码
class Solution:
    def countSubarrays(self, nums: List[int], k: int) -> int:
        _max = max(nums)
        n = len(nums)
        flags = []
        for i, x in enumerate(nums):
            if x == _max:
                flags.append(i)
        m = len(flags)
        if m < k:
            return 0
        
        ans = 0
        flags.insert(0, -1)
        for i in range(m-k+1):
            ans += (flags[i+1]-flags[i]) * (n-flags[i+k])
        return ans

提交代码评测得到:耗时1040ms,占用内存30.7MB。

相关推荐
Espresso Macchiato1 个月前
Leetcode 3316. Find Maximum Removals From Source String
leetcode·动态规划·leetcode medium·leetcode 3316·leetcode双周赛141
Espresso Macchiato2 个月前
Leetcode 3306. Count of Substrings Containing Every Vowel and K Consonants II
leetcode·滑动窗口·leetcode medium·leetcode 3306·leetcode周赛417
Espresso Macchiato2 个月前
Leetcode 3301. Maximize the Total Height of Unique Towers
leetcode·leetcode medium·leetcode双周赛140·leetcode 3301
Espresso Macchiato2 个月前
Leetcode 3302. Find the Lexicographically Smallest Valid Sequence
leetcode medium·lcs·leetcode 3302·leetcode双周赛140·最大公共子序列
Espresso Macchiato3 个月前
Leetcode 3255. Find the Power of K-Size Subarrays II
leetcode·leetcode medium·leetcode 3255·leetcode 3254·leetcode周赛137
Espresso Macchiato4 个月前
Leetcode 3240. Minimum Number of Flips to Make Binary Grid Palindromic II
leetcode·leetcode medium·回文·leetcode 3240·leetcode双周赛136
Espresso Macchiato4 个月前
Leetcode 3234. Count the Number of Substrings With Dominant Ones
排列组合·leetcode medium·容斥原理·leetcode 3234·leetcode周赛408
Espresso Macchiato5 个月前
Leetcode 3201. Find the Maximum Length of Valid Subsequence I
leetcode medium·leetcode题解·leetcode 3201·leetcode周赛404
Espresso Macchiato5 个月前
Leetcode 3196. Maximize Total Cost of Alternating Subarrays
leetcode·动态规划·leetcode medium·leetcode周赛403·leetcode 3196
Espresso Macchiato5 个月前
Leetcode 3195. Find the Minimum Area to Cover All Ones I
leetcode·leetcode medium·leetcode题解·leetcode 3195·leetcode周赛403