1493. 删掉一个元素以后全为 1 的最长子数组

问题

给你一个二进制数组 nums ，你需要从中删掉一个元素。

请你在删掉元素的结果数组中，返回最长的且只包含 1 的非空子数组的长度。

如果不存在这样的子数组，请返回 0 。

提示 1：
输入：nums = [1,1,0,1]
输出：3
解释：删掉位置 2 的数后，[1,1,1] 包含 3 个 1 。

示例 2：
输入：nums = [0,1,1,1,0,1,1,0,1]
输出：5
解释：删掉位置 4 的数字后，[0,1,1,1,1,1,0,1] 的最长全 1 子数组为 [1,1,1,1,1] 。

示例 3：
输入：nums = [1,1,1]
输出：2
解释：你必须要删除一个元素。

分析

窗口长度会变，一般称作双指针，遍历左边界l，右边界r从l+1开始遍历，遇到大于1个0可以剪枝。可以使用zero_n来记录当前0的个数，大于1直接break，并且计算当前个数时，可以通过r-l+1-zero_n来计算。

代码

class Solution {
public:
    int longestSubarray(vector<int>& nums) {
        int res = 0;
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] == 0) {break;}
            if (i == nums.size()-1) {return nums.size()-1;}
        }
        for (int x = 0; x < nums.size(); x++) {
            int zero_n = 0;
            if (nums[x] == 0) {continue;}
            for (int i = x+1; i < nums.size() && zero_n <= 1; i++) {
                if (nums[i] == 0) {zero_n++;}
                res = max(res, i-x+1-zero_n);
                if (zero_n > 1) {
                    break;
                }
            }
        }
        return res;
    }
};