1437. Check If All 1's Are at Least Length K Places Away

时间:2020-05-12 10:58:48   收藏:0   阅读:64

Given an array nums of 0s and 1s and an integer k, return True if all 1‘s are at least k places away from each other, otherwise return False.

 

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:

class Solution {
    public boolean kLengthApart(int[] nums, int k) {
        boolean res = false;
        List<Integer> list = new ArrayList();
        for(int i = 0; i < nums.length; i++){
            if(nums[i] == 1) list.add(i);
        }
        System.out.println(list.toString());
        //if(list.size() == 0) return k==0;
        int st = Integer.MAX_VALUE;
        for(int i = 1; i < list.size(); i++){
            st = Math.min(st, list.get(i) - list.get(i - 1));
            if(st - 1 < k) return false;
        }
        return true;
    }
}

求两1点间最小距离,注意如果没有1直接返回true

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

或者这样O(N)

原文:https://www.cnblogs.com/wentiliangkaihua/p/12874162.html

评论(0
© 2014 bubuko.com 版权所有 - 联系我们:wmxa8@hotmail.com
打开技术之扣,分享程序人生!