1493. Longest Subarray of 1's After Deleting One Element

1493. Longest Subarray of 1's After Deleting One Element (Medium)

Given a binary array nums, you should delete one element from it.

Return the size of the longest non-empty subarray containing only 1's in the resulting array.

Return 0 if there is no such subarray.

 

Example 1:

Input: nums = [1,1,0,1]
Output: 3
Explanation: After deleting the number in position 2, [1,1,1] contains 3 numbers with value of 1's.

Example 2:

Input: nums = [0,1,1,1,0,1,1,0,1]
Output: 5
Explanation: After deleting the number in position 4, [0,1,1,1,1,1,0,1] longest subarray with value of 1's is [1,1,1,1,1].

Example 3:

Input: nums = [1,1,1]
Output: 2
Explanation: You must delete one element.

Example 4:

Input: nums = [1,1,0,0,1,1,1,0,1]
Output: 4

Example 5:

Input: nums = [0,0,0]
Output: 0

 

Constraints:

• 1 <= nums.length <= 10^5
• nums[i] is either 0 or 1.

Related Topics:
Array

Solution 1.

prev2 and prev are the indexes of the non-one values we've seen most recently during scanning.

prev2              prev            i
  0       1 1 1     0      1 1 1   0 

If the array only contains 1, then return N - 1. Otherwise, the answer is the maximum of i - prev2 - 2.

// OJ: https://leetcode.com/problems/longest-subarray-of-1s-after-deleting-one-element/
// Author: github.com/lzl124631x
// Time: O(N)
// Space: O(1)
class Solution {
public:
    int longestSubarray(vector<int>& A) {
        int N = A.size(), prev2 = -1, prev = -1, ans = 0;
        for (int i = 0; i <= N; ++i) {
            if (i < N && A[i] == 1) continue;
            if (i == N && prev == -1) return N - 1;
            if (prev != -1) ans = max(ans, i - prev2 - 2);
            prev2 = prev;
            prev = i;
        }
        return ans;
    }
};

Solution 2. Sliding Window

Sliding window [i, j) with at most 2 zeros.

1. Keep extending the right edge until the window becomes invalid, i.e. more than one zero, or can't be extended any more.
2. Now cnt can be either 0, 1 or 2. The corresponding size after deleting one element is j - i - max(cnt, 1).
3. Then we shrink the left edge until cnt becomes valid, i.e. cnt <= 1.
4. Back to step 1 until the window reaches the end.

// OJ: https://leetcode.com/problems/longest-subarray-of-1s-after-deleting-one-element/
// Author: github.com/lzl124631x
// Time: O(N)
// Space: O(1)
class Solution {
public:
    int longestSubarray(vector<int>& A) {
        int i = 0, j = 0, N = A.size(), cnt = 0, ans = 0;
        while (j < N) {
            while (j < N && cnt <= 1) cnt += A[j++] == 0;
            ans = max(ans, j - i - max(cnt, 1));
            while (cnt > 1) cnt -= A[i++] == 0;
        }
        return ans;
    }
};

Solution 3. Sliding Window

Sliding window [i, j) with at most 1 zero.

1. Keep extending the right edge until it reaches the end or one more extension will include 2 zeros.
2. j - i - 1 is the corresponding answer for this window.
3. Shrink the left edge until cnt == 0.
4. Back to step 1 until the window reaches the end.

// OJ: https://leetcode.com/problems/longest-subarray-of-1s-after-deleting-one-element/
// Author: github.com/lzl124631x
// Time: O(N)
// Space: O(1)
class Solution {
public:
    int longestSubarray(vector<int>& A) {
        int i = 0, j = 0, N = A.size(), cnt = 0, ans = 0;
        while (j < N) {
            while (j < N && cnt + (A[j] == 0) <= 1) cnt += A[j++] == 0;
            ans = max(ans, j - i - 1);
            while (cnt == 1) cnt -= A[i++] == 0;
        }
        return ans;
    }
};

Solution 4. Sliding Window

Sliding window [i, j] with at most 1 zero.

1. Keep extending the right edge.
2. Once the cnt > 1, shrink the left edge.
3. The size for this window with one deletion is j - i.
4. Back to step 1 until it reaches the end.

This solution is simpler than the previous one because we don't need to think about when to stop the extension of the right edge.

// OJ: https://leetcode.com/problems/longest-subarray-of-1s-after-deleting-one-element/
// Author: github.com/lzl124631x
// Time: O(N)
// Space: O(1)
class Solution {
public:
    int longestSubarray(vector<int>& A) {
        int i = 0, j = 0, N = A.size(), cnt = 0, ans = 0;
        for (; j < N; ++j) {
            cnt += A[j] == 0;
            while (cnt > 1) cnt -= A[i++] == 0;
            ans = max(ans, j - i);
        }
        return ans;
    }
};

Solution 5. Sliding Window

Sliding window [i - ans + 1, i] with at most 1 zero that never shrinks.

// OJ: https://leetcode.com/problems/longest-subarray-of-1s-after-deleting-one-element/
// Author: github.com/lzl124631x
// Time: O(N)
// Space: O(1)
class Solution {
public:
    int longestSubarray(vector<int>& A) {
        int N = A.size(), cnt = 0, ans = 0;
        for (int i = 0; i < N; ++i) {
            cnt += A[i] == 0;
            if (cnt > 1) cnt -= A[i - ans] == 0;
            else ++ans;
        }
        return ans - 1;
    }
};

Sliding window [i, j] with at most 1 zero taht never shrinks.

// OJ: https://leetcode.com/problems/longest-subarray-of-1s-after-deleting-one-element/
// Author: github.com/lzl124631x
// Time: O(N)
// Space: O(1)
class Solution {
public:
    int longestSubarray(vector<int>& A) {
        int i = 0, j = 0, N = A.size(), cnt = 0;
        for (; j < N; ++j) {
            cnt += A[j] == 0;
            if (cnt > 1) cnt -= A[i++] == 0;
        }
        return j - i - 1;
    }
};