leetcode 1493. Longest Subarray of 1‘s After Deleting One Element(删除一个元素后‘1‘ 的最大长度)

请添加图片描述

数组nums仅含 0,1。要在数组中删除一个元素(全1也要删一个)。
返回删除一个元素后,最长的 全1 子数组的长度。

思路:

因为要求的是全1子数组的长度,
又要删除一个元素,如果一些1中间夹杂着一个0,把0删除后就可以把两段1的子数组接起来。
长度为由0隔开的两段子数组的长度之和。

那如果中间夹杂着多个0呢?
一样的处理,还是算由0隔开的两段子数组的长度,连续的0那么0隔开的子数组长度也为0.

所以两两相加(只能删除一个元素,所以只能两段)由0隔开的子数组长度,记下最大值。

有种特殊情况:没有0,全1,那么全1子数组的长度为n, 但是必须删除一个元素,所以最终长度为n-1.

    public int longestSubarray(int[] nums) {
    
    
        int n = nums.length;
        int pre0 = -1;
        int cur0 = findNextZero(0, nums);
        if (cur0 == n) return n - 1; //全1的情况

        int res = 0;
        while (cur0 != n) {
    
    
            int next0 = findNextZero(cur0 + 1, nums);
            int len = (cur0 - pre0 - 1) + (next0 - cur0 - 1); //两头的0去掉
            
            res = Math.max(res, len);
            
            pre0 = cur0;
            cur0 = next0;
        }
        return res;
    }

    int findNextZero(int from, int[] arr) {
    
    
        for (int i = from; i < arr.length; i++) {
    
    
            if (arr[i] == 0) {
    
    
                return i;
            }
        }
        return arr.length;
    }

猜你喜欢

转载自blog.csdn.net/level_code/article/details/131550676