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

更新於:2020年6月11日

1K+ 次瀏覽

開啟你的職業生涯

完成課程獲得認證

開始學習
廣告