1
2

输入：flowerbed = [1,0,0,0,1], n = 1
输出：true

1
2

输入：flowerbed = [1,0,0,0,1], n = 2
输出：false

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

// @lc code=start
impl Solution {
    /// ## 解题思路
    /// - 贪心法
    /// 1. 从左到右遍历数组;
    /// 2. 如果当前位置为1, 则表明当前位置已经有花了, 跳2格到下下一个位置;
    /// 3. 否则当前位置为0.
    /// 4. 由于不可能有位置相邻的1, 且当前面每次遇到1时, 必须跳两格, 所以此时只需判断后面是否为0;
    /// 5. 如果f[i+1] == 0 或者 i=len-1, 则可以在当前位置种花, 种完后跳2格;
    /// 6. 如果f[i+1] == 1, 则其下一格f[i+2]必定不能种, 需要往后跳3格;
    pub fn can_place_flowers(flowerbed: Vec<i32>, n: i32) -> bool {
        if n <= 0 {
            return true;
        }
        let mut n = n;
        let mut i = 0;
        while i < flowerbed.len() && n > 0{
            if flowerbed[i] == 1 {
                i += 2;
            }  else if i == flowerbed.len() -1 || flowerbed[i+1] == 0 {
                n -= 1;
                i += 2;
            } else {
                i += 3;
            }
        }

        return n<=0;
    }
}
// @lc code=end

struct Solution;

#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn test() {
        assert_eq!(Solution::can_place_flowers(vec![1, 0, 0], 1), true);
        assert_eq!(Solution::can_place_flowers(vec![1, 0, 0, 0, 0, 1], 2), false);
        assert_eq!(Solution::can_place_flowers(vec![1, 0, 0, 0, 0, 1], 2), false);
    }
}