1、LeetCode----703. 数据流中的第K大元素
https://leetcode-cn.com/problems/kth-largest-element-in-a-stream/
基本思路
#首先保证一个数组里面的个数为k
#通过每次找出数组中的最小的数与新加入的数进行比较
#当新加入的数大于最小的数时将其替换为最小的数,
#最终要求的数为数组中最小的数,返回此数
#最后一个case通不过,由于复杂度大了,当代大神改进此算法
class KthLargest:
def __init__(self, k: int, nums: List[int]):
self.k = k
self.nums = nums
if len(self.nums) > self.k:
while len(self.nums) > self.k:
self.nums.remove(min(self.nums))
if not len(self.nums):
while len(self.nums)!=self.k:
self.nums.append(-100000)
def add(self, val: int) -> int:
min_num = min(self.nums)
if len(self.nums) < self.k:
self.nums.append(val)
elif val > min_num:
self.nums.remove(min_num)
self.nums.append(val)
return min(self.nums)
2、LeetCode----239. 滑动窗口最大值
https://leetcode.com/problems/top-k-frequent-words/description/
基本思路
#这一题显示是困难,但其实使用python解并不是很难
#首先我们考虑的是遍历整个数组,将内容进行切片处理,每次求解出切片里面最大的元素添加到answer中
#最终返回answer即可,当然需要考虑到一些特殊的情况,比如说数组为空,或者k=1的情况。
class Solution:
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
if not nums:
return
if k == 1:
return nums
answer = []
for i in range(len(nums)):
if i+k > len(nums):
break
temp = nums[i:i+k]
answer.append(max(temp))
return answer