C++ 中檢查所有 1 是否至少間隔 K 個位置


假設我們有一個包含 0 和 1 的陣列 nums 和一個整數 k,我們需要檢查所有 1 是否至少相隔 k 個位置,否則返回 False。

因此,如果輸入類似於 nums = [1,0,0,0,1,0,0,1],k = 2,則輸出將為 true,因為每個 1 都至少相隔 2 個位置。

為了解決這個問題,我們將遵循以下步驟:

  • last := -1

  • for 初始化 i := 0,當 i < nums 的大小,更新(i 增加 1),執行:

    • 如果 nums[i] 等於 1,則:

      • 如果 last 等於 -1 或 (i - last - 1) >= k,則:

        • last := i

      • 否則

        • 返回 false

  • 返回 true

示例

讓我們看看以下實現以獲得更好的理解:

線上演示

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
   bool kLengthApart(vector<int>& nums, int k) {
      int last = -1;
      for (int i = 0; i < nums.size(); i++) {
         if (nums[i] == 1) {
            if (last == -1 || (i - last - 1) >= k)
               last = i;
            else
               return false;
         }
      }
      return true;
   }
};
main(){
   Solution ob;
   vector<int> v = {1,0,0,0,1,0,0,1};
   cout << (ob.kLengthApart(v, 2));
}

輸入

{1,0,0,0,1,0,0,1}

輸出

1

更新於: 2020-11-17

137 次檢視

開啟你的 職業生涯

透過完成課程獲得認證

開始學習
廣告

© . All rights reserved.