leetcode605. 种花问题

假设你有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花卉不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给定一个花坛(表示为一个数组包含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的情况,然后再考虑如果数组头部有连续的两个0,那么可以在头部种一朵花,同理,如果尾部有连续的两个0,那么可以在尾部种一朵花,种完要将这个位置的0置为1,代表这个位置已经种花了,然后开始处理中间的情况,对中间的情况而言,如果存在连续的三个0,那么就可以在中间的那个位置上种花,种完置为1,这里的flag表示可以种植花的最大数量,最后得到的flag如果大于或等于n,即可以种植n朵花,反之则不行。有问题可以指出来,有更好的方法欢迎评论!

class Solution {
public:
    bool canPlaceFlowers(vector& flowerbed, int n) {
        if(flowerbed.size()==1 && flowerbed[0]==0)
            return n<=1? true:false;
        if(flowerbed.size()==1 && flowerbed[0]==1)
            return n==0? true:false;
        int nCount = flowerbed.size();
        int flag = 0;
        if(flowerbed[0] == 0 && flowerbed[1] == 0)
        {
            flowerbed[0] = 1;
            flag++;
        }
        for(int i = 1;i= n)
            return true;
        return false;
    }
};

你可能感兴趣的:(Leetcode数组)