题目描述

假设有一个很长的花坛，一部分地块种植了花，另一部分却没有。可是，花不能种植在相邻的地块上，它们会争夺水源，两者都会死去。

给你一个整数数组 flowerbed 表示花坛，由若干 0 和 1 组成，其中 0 表示没种植花，1 表示种植了花。另有一个数 n ，能否在不打破种植规则的情况下种入 n 朵花？能则返回 true ，不能则返回 false。

 

示例 1：

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

示例 2：

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

提示：

  • 1 <= flowerbed.length <= 2 * 104
  • flowerbed[i] 为 0 或 1
  • flowerbed 中不存在相邻的两朵花
  • 0 <= n <= flowerbed.length

链接：https://leetcode-cn.com/problems/can-place-flowers

题解

class Solution {
    public boolean canPlaceFlowers(int[] flowerbed, int n) {
        int left = -1, num = 0;
        for (int i = 0; i < flowerbed.length; i++) {
            if (flowerbed[i] == 1) {
                if (left < 0) {
                    num += i / 2;
                } else {
                    int tmp = i - left - 1;
                    num += tmp % 2 == 1 ? tmp / 2 : tmp / 2 - 1;
                }
                if (num >= n) {
                    return true;
                }
                left = i;
            }
        }
        if (left >= 0) {
            num += (flowerbed.length - left - 1) / 2;
        } else {
            num += (flowerbed.length - left) / 2;
        }
        return num >= n;
    }
}

返回