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

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
解释：你必须要删除一个元素。

提示：

• 1 <= nums.length <= 10^5
• nums[i] 要么是 0 要么是 1 。

Solution 1

class Solution {
public:
    int longestSubarray(vector<int>& nums) {
        int k = 1; // remove k elements
        int result = 0;
        int fastIdx = 0, slowIdx = 0;
        int cnt0 = 0, cnt1 = 0;
        for (fastIdx = 0; fastIdx < nums.size(); fastIdx++) {
            int num = nums[fastIdx];
            if (num == 1) {
                cnt1++;
            } else {
                cnt0++;
                while (cnt0 > k) {
                    num = nums[slowIdx++];
                    if (num == 1) {
                        cnt1--;
                    } else {
                        cnt0--;
                    }
                }
            }
            result = max(result, cnt1);
        }
        if (fastIdx == nums.size() && slowIdx == 0) {
            result = nums.size() - 1;
        }
        return result;
    }
};