605. 种花问题

地址：

力扣https://leetcode-cn.com/problems/can-place-flowers/

题目：

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/can-place-flowers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

考虑计算当前能新种植的最大数，最后和 n 比较，如果最大数 < n ，那么种不下

对于一个3元素组合，可能存在 1 或 0 的新种，比如：[1,0,0] or [0,1,0] or [0,0,1]

最容易想到的就是看当前元素值：

1. [i] == 1，可能的位置 [1,0,0]，所以 下标 +2

2. [i] == 0，可能的位置 [0,0,0]，[0,0,1]，所以下标 +2

        如果[i+1] == 1，[0,1,0] 不可能，所以下标 +1, +2 

3. 最后处理末尾元素可能不是 3元组

        [0]，[0,0] 需要考虑 [i-1] 是否为0，如果是0，那么[1]，[1,0]

        [0,1] 不用考虑，因为前面两种情况 1 的位置已经处理了

方法一、遍历

bool canPlaceFlowers(int* flowerbed, int flowerbedSize, int n){
	int maxSize = 0;
	
	if(flowerbedSize == 1)
	{
		if(flowerbed[0] == 0)
			maxSize++;
			
		if(maxSize < n)
			return false;
		else
			return true;	
	}
	
	for(int i=0; i<flowerbedSize; )
	{
		// [i] == 1
		if(flowerbed[i] == 1)
		{
			i+=2;
			continue;
		}
		
		// [i] == 0
		if(flowerbed[i] == 0)
		{
			if(i+1 < flowerbedSize)
			{
				if(flowerbed[i+1] == 0)
				{
					maxSize++;
					i+=2;
					continue;
				}
				i++; 
				i+=2; // new start
				
				continue;
			}
			else if(flowerbed[i-1] != 1)
			{
				maxSize++;
				i++;
			}
		}
	}
	
	if(maxSize < n)
		return false;
	else
		return true;
}

查看更多刷题笔记