贪心:leetcode605(简单)——种花问题

题目

在这里插入图片描述

题解

思路:官方的题解主要是数学规律的查找,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

猜你喜欢

转载自blog.csdn.net/memory_MM_forever/article/details/121564859
今日推荐