假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组 flowerbed 表示花坛,由若干 0 和 1 组成,其中 0 表示没种植花,1 表示种植了花。另有一个数 n ,能否在不打破种植规则的情况下种入 n 朵花?能则返回 true ,不能则返回 false。
解题思路:
从示例中可以看出,当列表中存在三个连续为0
的元素时,说明可以种一棵树;而还有一种情况是[0,0,1]
,[1,0,0]
这种形式,也可以种植一棵树,但是在第二种情况中没有前者分析的连续三个元素为0
的元素,这时我们可以把它变成我们想要的样子。具体代码如下:
def huatan(flower, n):
count = 0
flower = [0] + flower + [0]
print(flower)
for i in range(1, len(flower) - 1):
if flower[i-1] == 0 and flower[i] == 0 and flower[i+1] == 0:
flower[i] == 1
count += 1
if count >= n:
return True
return False