Uriel's Corner

Research Associate @ Harvard University / Research Interests: Computer Vision, Biomedical Image Analysis, Machine Learning
posts - 0, comments - 50, trackbacks - 0, articles - 594
给出一列数nums,求这列数从左到右长度k的滑动窗口内的最大值依次是多少,维护一个优先队列,当计算到第i个数时,优先队列里面保存所有长度k以内值大于nums[i]的数的下标(因此可以保证nums[0]是窗口长度k以内的最大值)


 1 #239
 2 #Runtime: 1773 ms (Beats 16%)
 3 #Memory: 30.5 MB (Beats 48.38%)
 4 
 5 class Solution(object):
 6     def maxSlidingWindow(self, nums, k):
 7         """
 8         :type nums: List[int]
 9         :type k: int
10         :rtype: List[int]
11         """
12         wd_max = []
13         ans = []
14         for i in range(len(nums)):
15             if wd_max and wd_max[0] == i - k:
16                 wd_max.pop(0)
17             while wd_max and nums[wd_max[-1]] < nums[i]:
18                 wd_max.pop()
19             wd_max.append(i)
20             if i >= k - 1:
21                 ans.append(nums[wd_max[0]])
22         return ans

只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理