Arrays and Strings: Max Consecutive Ones III

Author: jhonnymoreira

src/arrays-and-strings/max-consecutive-ones-iii.ts

@@ -0,0 +1,20 @@
+export function longestOnes(nums: number[], k: number): number {
+  const n = nums.length;
+  let leftIndex = 0;
+  let rightIndex = 0;
+
+  while (rightIndex < n) {
+    if (nums[rightIndex] === 0) {
+      k--;
+    }
+
+    if (k < 0) {
+      k += 1 - nums[leftIndex]!;
+      leftIndex++;
+    }
+
+    rightIndex++;
+  }
+
+  return rightIndex - leftIndex;
+}

tests/arrays-and-strings/max-consecutive-ones-iii.test.ts

@@ -0,0 +1,15 @@
+import { longestOnes } from '@/arrays-and-strings/max-consecutive-ones-iii.js';
+
+describe('Arrays and Strings: Max Consecutive Ones III', () => {
+  test.each([
+    { nums: [1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 0], k: 2, output: 6 },
+    {
+      nums: [0, 0, 1, 1, 0, 0, 1, 1, 1, 0, 1, 1, 0, 0, 0, 1, 1, 1, 1],
+      k: 3,
+      output: 10,
+    },
+    { nums: [0, 1, 1], k: 0, output: 2 },
+  ])('longestOnes($nums, $k) === $output', ({ nums, k, output }) => {
+    expect(longestOnes(nums, k)).toStrictEqual(output);
+  });
+});