笔试编程（5）--百度

时间：2020年7.20

一.滑动窗口最大值

题目描述

给定一个数组num,存在一个大小为k的滑动窗口，该窗口从数组的最左边移到最右边，只能在窗口看到k个数字，每次滑动向右移动一个位置，请返回最大滑动窗口里大数字。要求时间复杂度O（n）

• 输入：

[1,3,-1,-3,5,3,6,7],3

• 输出

[3,3,5,5,6,7]

分析思路

1.窗口最大值遍历数组a到len(a)-k+1，每组最大值用max(nums[i:i+k])求解

2.输入

利用切片

lis=input()
k=lis.split(',')[-1]
a=lis[0:len(lis)-(len(k)+1)]
k=int(k)
from ast import literal_eval
a=literal_eval(a)

3.输出

ans=maxpool1d_cpu(a,k)# ans接收的是列表
print(str(ans))

完整代码

def maxpool1d_cpu(nums,k):
    result=[]
    for i in range(len(nums)-k+1):
        result.append(max(nums[i:i+k]))
    return result
lis=input()
k=lis.split(',')[-1]
a=lis[0:len(lis)-(len(k)+1)]
k=int(k)
from ast import literal_eval
a=literal_eval(a)
print(a,k)
ans=maxpool1d_cpu(a,k)# ans接收的是列表
print(type(ans))
print(str(ans))

输出结果

二.统计幸运对个数