题目描述
近些年来,我国防沙治沙取得显著成果。某沙漠新种植N棵胡杨(编号1-N),排成一排。一个月后,有M棵胡杨未能成活。现可补种胡杨K棵,请问如何补种(只能补种,不能新种),可以得到最多的连续胡杨树?
输入描述:
N 总种植数量 1<=N<=100000 M 未成活胡杨数量 1<=M<=N M 个空格分隔的数,按编号从小到大排列 K 最多可以补种的数量 0<=K<=M
输出描述:
最多的连续胡杨棵树
示例1
输入
5
2
2 4
1
输出
3
说明
补种到2或4结果一样,最多的连续胡杨棵树都是3
示例2
输入
10
3
2 4 7
1
输出
6
说明
补种第7棵树,最多的连续胡杨棵树为6(5,6,7,8,9,10)
滑动窗口问题,保证窗口内有k个未成活,然后求最大窗口大小就行。左边少一个,右边就找一个新的
Python实现
#Python实现
while True:
try:
N = int(input())
M = int(input())
death = list(map(int,input().split()))
k = int(input())
nums = [1 for _ in range(N)]
for i in death:
nums[i - 1] = 0
left,right,res=0,0,0
for right in range(N):
k-=1-nums[right]
while k<0:
k+=1-nums[left]
left+=1
res = max(res,right-left+1)
print(res)
except:
break
如有问题,请大家提出指教