能否在 C++ 中放置鲜花
假设我们有一个长长的花坛,其中一些地块已种花,而另一些地块是空的。现在有一个限制,花不能种植在相邻的地块中,它们会争夺水,并且都会死亡。因此,如果我们有一个花坛,用包含 0 和 1 的数组表示,0 表示空,1 表示已填满,并且还给出了一个数字 n,我们必须检查是否可以在其中种植 n 朵新花而不违反无相邻花规则。
因此,如果输入为 flowerbed = [1,0,0,0,1], n = 1,则输出将为 True
为了解决这个问题,我们将遵循以下步骤 −
如果花坛的大小 < n,则 −
返回 false
如果花坛大小等于 1,且 flowerbed[0] 等于 0,且 n 等于 1,则 −
返回 true
初始化 i := 0,当 i < 花坛大小时,更新(将 i 增加 1),执行 −
如果 n > 0,则 −
如果 i 与 0 相同,则 −
如果 flowerbed[i] 与 0 相同且 flowerbed[1] 与 0 相同,则 −
flowerbed[0] := 1
(将 n 减少 1)
否则,当 i 与 flowerbed - 1 的大小相同时,则 −
如果 flowerbed[i] 与 0 相同,且 flowerbed[i - 1] 不等于 1,则 −
flowerbed[i] := 1
(将 n 减少 1)
否则,当 flowerbed[i] 等于 0 且 flowerbed[i + 1] 等于 0 且 flowerbed[i - 1] 等于 0 时,则 −
flowerbed[i] := 1
(将 n 减少 1)
如果 n 与 0 相同,则 −
return true
如果 n 与 0 相同,则 −
return true
return false
示例
让我们看下面的实现,以便更好地理解 −
#include <bits/stdc++.h> using namespace std; class Solution { public: bool canPlaceFlowers(vector<int>& flowerbed, int n) { if (flowerbed.size() < n) return false; if (flowerbed.size() == 1 && flowerbed[0] == 0 && n == 1) return true; for (int i = 0; i < flowerbed.size(); i++) { if (n > 0) { if (i == 0) { if (flowerbed[i] == 0 && flowerbed[1] == 0) { flowerbed[0] = 1; n--; } } else if (i == flowerbed.size() - 1) { if (flowerbed[i] == 0 && flowerbed[i - 1] != 1) { flowerbed[i] = 1; n--; } } else if (flowerbed[i] == 0 && flowerbed[i + 1] == 0 && flowerbed[i - 1] == 0) { flowerbed[i] = 1; n--; } } if (n == 0) { return true; } } if (n == 0) { return true; } return false; } }; main(){ Solution ob; vector<int> v = {1,0,0,0,1}; cout << (ob.canPlaceFlowers(v, 1)); }
输入
{1,0,0,0,1}, 1
输出
1