题目描述
给定一个长度为 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 = 4def 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:passelse:ever_max_item.append(max(list_data[i:i + window_size]))return ever_max_itemprint(max_in_windows(list_data, window_size))