题目描述
给定一个长度为 n 的数组 num 和滑动窗口的大小 size ,找出所有滑动窗口里数值的最大值。例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小 3,那么一共存在6 个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}; 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6 个:{[2,3,4],2,6,2,5,1}, {2,[3,4,2],6,2,5,1}, {2,3,[4,2,6],2,5,1}, {2,3,4,[2,6,2],5,1},{2,3,4,2,[6,2,5],1}, {2,3,4,2,6,[2,5,1]}。窗口大于数组长度或窗口长度为0 的时候,返回空。
输入示例
2,3,4,2,6,2,5,1
3
输出示例
[4,4,6,6,6,5]
分析:
就是给一组列表:[2,3,4,2,6,2,5,1],从索引为0开始每次切3个,分别是
1、 [2,3,4],其中最大的元素是:4
2、 [3,4,2],其中最大的元素是:4
3、 [4,2,6],其中最大的元素是:6
4、 [2,6,2],其中最大的元素是:7
5、 [6,2,5],其中最大的元素是:8
6、 [2,5,1],其中最大的元素是:5
最后将每个的最大元素返回回去: [4,4,6,6,6,5]
参考答案:
# 参考答案
def max_in_windows(nums, window_size):
if not nums or window_size <= 0 or window_size > len(nums):
return []
deque, res = [], []
for i in range(len(nums)):
while deque and nums[i] > nums[deque[-1]]:
deque.pop()
deque.append(i)
if deque[0] == i - window_size:
deque.pop(0)
if i >= window_size - 1:
res.append(nums[deque[0]])
return res
if __name__ == '__main__':
nums = list(map(int, input("请输入数组:").split(",")))
window_size = int(input("请输入滑动窗口大小:"))
res = max_in_windows(nums, window_size)
print("所有滑动窗口的最大值为:", res)
我写的(写的比较烂)
list_data = [1,2,3,4,5,6,]
window_size = 4
def max_in_windows(list_data, window_size):
ever_max_item = []
if window_size > len(list_data):
return []
for i in range(len(list_data)):
if len(list_data[i:i + window_size]) < window_size:
pass
else:
ever_max_item.append(max(list_data[i:i + window_size]))
return ever_max_item
print(max_in_windows(list_data, window_size))