leetcode学习笔记1437 Check If All 1‘s Are at Least Length K Places Away

问题

Check If All 1’s Are at Least Length K Places Away

检查是否所有的1之间的间隔都至少是k.

Example 1:

Input: nums = [1,0,0,0,1,0,0,1], k = 2
Output: true
Explanation: Each of the 1s are at least 2 places away from each other.

Example 2:

Input: nums = [1,0,0,1,0,1], k = 2
Output: false
Explanation: The second 1 and third 1 are only one apart from each other.

Example 3:

Input: nums = [1,1,1,1,1], k = 0
Output: true

Example 4:

Input: nums = [0,1,0,1], k = 1
Output: true

Constraints:

• 1 <= nums.length <= 105
• 0 <= k <= nums.length
• nums[i] is 0 or 1

方法1

遍历,依次计算所有1之间的距离.

时间复杂度O(n).
空间复杂度O(1).

class Solution {
    public boolean kLengthApart(int[] nums, int k) {
        if(k == 0)
            return true;
        
        int left = -1;
        for(int i = 0; i < nums.length; i++){
            if(nums[i] == 0)
                continue;
            
            if(left < 0)
                left = i;
            else{
                if(i - left - 1 < k)
                    return false;
                else
                    left = i;
            }
        }
        return true;
    }
}