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
廣告
資料結構
網路
關係資料庫管理系統
作業系統
Java
iOS
HTML
CSS
Android
Python
C 程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP