C++中能否放置花朵
假設我們有一塊長長的花壇,其中一些地塊種了花,一些是空的。現在有一個限制條件,花不能種在相鄰的地塊,否則它們會競爭水分,兩者都會枯死。所以如果我們有一個花壇,用一個包含0和1的陣列表示,0表示空,1表示已種花,並且給定一個數字n,我們需要檢查是否可以在不違反不相鄰種花規則的情況下種植n朵新花。
因此,如果輸入類似於flowerbed = [1,0,0,0,1],n = 1,則輸出為True
為了解決這個問題,我們將遵循以下步驟:
如果flowerbed的大小小於n,則:
返回false
如果flowerbed的大小為1,並且flowerbed[0]為0,並且n為1,則:
返回true
初始化i := 0,當i < flowerbed的大小時,更新(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,則:
返回true
如果n等於0,則:
返回true
返回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
廣告
資料結構
網路
關係資料庫管理系統 (RDBMS)
作業系統
Java
iOS
HTML
CSS
Android
Python
C語言程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP