Letcode(4)种花问题

题目

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

  给你一个整数数组 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

解答

public:
    bool canPlaceFlowers(vector<int>& flowerbed, int n) {
        int canPlaceNum = 0;
        for (int i=0; i<flowerbed.size(); i++) {
            if (flowerbed.at(i)) {
                continue;
            }
            if (i == 0) {
                if (flowerbed.size() > 1) {
                    if (flowerbed.at(i) | flowerbed.at(i+1) ) {
                        continue;
                    } else {
                        flowerbed.at(i) = 1;
                        canPlaceNum++;
                    }
                } else {
                    flowerbed.at(i) = 1;
                    canPlaceNum++;   
                }
            } else if (i == flowerbed.size()-1) {
                  if (flowerbed.at(i) | flowerbed.at(i-1) ) {
                    continue;
                } else {
                    flowerbed.at(i) = 1;
                    canPlaceNum++;
                }
            } else {
                if ( flowerbed.at(i) | flowerbed.at(i-1) | flowerbed.at(i+1) ) {
                    continue;
                } else {
                    flowerbed.at(i) = 1;
                    canPlaceNum++;
                }
            }
        }

        return n>canPlaceNum?false:true;
    }
};