|
| 1 | +def max_ones_after_remove_one(nums: list[int]) -> int: |
| 2 | + """ |
| 3 | + Finds the maximum length of a consecutive subarray of 1s after removing exactly one element (0 or 1). |
| 4 | +
|
| 5 | + :param nums: A list of integers containing only 0s and 1s. |
| 6 | + :return: The maximum length of consecutive 1s achievable by deleting exactly one element from the list. |
| 7 | + """ |
| 8 | + left = 0 |
| 9 | + |
| 10 | + zero_position = -1 |
| 11 | + zero_counter = 0 |
| 12 | + max_counter = 0 |
| 13 | + |
| 14 | + for right in range(len(nums)): |
| 15 | + if nums[right] == 0: |
| 16 | + if zero_counter == 1: |
| 17 | + left = zero_position + 1 |
| 18 | + zero_counter -= 1 |
| 19 | + zero_counter += 1 |
| 20 | + zero_position = right |
| 21 | + max_counter = max(max_counter, right - left + 1 - zero_counter) |
| 22 | + |
| 23 | + return max_counter |
| 24 | + |
| 25 | + |
| 26 | +assert max_ones_after_remove_one([0, 0]) == 0, 'Test 1 failed' |
| 27 | +assert max_ones_after_remove_one([0, 1]) == 1, 'Test 2 failed' |
| 28 | +assert max_ones_after_remove_one([1, 0, 1]) == 2, 'Test 3 failed' |
| 29 | +assert max_ones_after_remove_one([1, 1, 1]) == 3, 'Test 4 failed' |
| 30 | +assert max_ones_after_remove_one([1, 1, 0, 1, 1]) == 4, 'Test 5 failed' |
| 31 | +assert max_ones_after_remove_one([1, 1, 1, 0, 1, 1, 0]) == 5, 'Test 6 failed' |
| 32 | +assert max_ones_after_remove_one([1, 1, 0, 1, 0, 1]) == 3, 'Test 7 failed' |
| 33 | +assert max_ones_after_remove_one([1, 1, 0, 0, 1, 0, 1]) == 2, 'Test 8 failed' |
| 34 | +assert max_ones_after_remove_one([1, 0, 1, 0, 1, 1, 1]) == 4, 'Test 9 failed' |
| 35 | +assert max_ones_after_remove_one([0, 0, 0, 0]) == 0, 'Test 10 failed' |
| 36 | +assert max_ones_after_remove_one([1, 0, 1, 1, 0]) == 3, 'Test 11 failed' |
0 commit comments