题目
题解
思路:官方的题解主要是数学规律的查找,prev为上一朵种植的花的位置:
1.如果位置 i 为1且前面都为0,可以种 i//2 朵花;如前面有 1 , 可以种 (i-prev-2)//2 朵花;
如果遍历的过程中count>=n,就直接结束,不需要再计算
2.最后一个区间为最后的1的右侧区间,可以种(m-prev-1)//2 朵花;
3.特殊情况:整个数组都是0,可以种(m+1)//2 朵花;
class Solution:
def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:
m = len(flowerbed)
count = 0
prev = -1
for i in range(0, m):
if flowerbed[i] == 1:
if prev < 0:
count += i//2
else:
count += (i-prev-2)//2
if count >= n:
return True
prev = i
if prev < 0:
count += (m+1)//2
else:
count += (m-prev-1)//2
return count >= n