使用Python删除一个元素后查找最长1s子数组的程序
假设我们有一个名为nums的二进制数组,我们可以从中删除一个元素。我们必须找到在结果数组中只包含1的最长非空子数组的大小。如果没有这样的子数组,则返回0。
因此,如果输入类似于nums=[1,0,1,1,1,0,1,1,0],那么输出将是5,因为通过从位置5中删除0,我们可以得到一个子数组[1,1,1,1,1]有五个1。
示例
def solve(nums):
if 0 not in nums:
return len(nums)-1
if 1 not in nums:
return 0
a = []
cnt = 0
for i in nums:
if i == 0:
if cnt != 0:
a.append(cnt)
cnt = 0
a.append(i)
else:
cnt += 1
if cnt!=0:
a.append(cnt)
Max = 0
for i in range(len(a)):
if a[i] != 0:
continue
if a[i] == 0 and i == len(a)-1:
Max = max(Max,a[i-1])
elif a[i] == 0 and i == 0:
Max = max(Max,a[i+1])
elif a[i] == 0:
Max = max(Max,a[i+1]+a[i-1])
return Max
nums = [1,0,1,1,1,0,1,1,0]
print(solve(nums))输入
[1,0,1,1,1,0,1,1,0]输出结果
5