力扣 1004. 最大连续 1 的个数 III - 满足条件的 C 语言实现

2023-11-01 00:53:28

最近，我在刷力扣的题海时遇到了一个颇具挑战性的题目——力扣 1004：最大连续 1 的个数 III。这道题要求找出给定数组中满足特定条件的最大连续 1 的个数。具体来说，条件是允许在连续 1 中最多包含 K 个 0。

对于这道题，我的第一反应是使用滑动窗口的技巧。滑动窗口是一种常见的算法设计模式，特别适用于处理连续子数组问题。在我们的场景中，我们可以使用一个左指针和一个右指针来定义一个滑动窗口，并遍历数组以找到满足条件的最长连续 1。

下面是我实现的 C 语言代码：

#include <stdio.h>
#include <stdlib.h>

int maxConsecutiveOnes(int* nums, int numsSize, int k) {
    // 初始化滑动窗口的左右指针和最大长度
    int left = 0, right = 0, maxLength = 0;
    // 统计滑动窗口中 0 的个数
    int zeroCount = 0;

    while (right < numsSize) {
        // 如果当前元素为 0，增加 0 的计数
        if (nums[right] == 0) {
            zeroCount++;
        }

        // 如果 0 的个数超过 k，移动左指针直到 0 的个数不超过 k
        while (zeroCount > k) {
            if (nums[left] == 0) {
                zeroCount--;
            }
            left++;
        }

        // 更新最大长度
        maxLength = fmax(maxLength, right - left + 1);

        // 移动右指针
        right++;
    }

    return maxLength;
}