数组nums仅含 0,1。要在数组中删除一个元素(全1也要删一个)。
返回删除一个元素后,最长的 全1 子数组的长度。
思路:
因为要求的是全1子数组的长度,
又要删除一个元素,如果一些1中间夹杂着一个0,把0删除后就可以把两段1的子数组接起来。
长度为由0隔开的两段子数组的长度之和。
那如果中间夹杂着多个0呢?
一样的处理,还是算由0隔开的两段子数组的长度,连续的0那么0隔开的子数组长度也为0.
所以两两相加(只能删除一个元素,所以只能两段)由0隔开的子数组长度,记下最大值。
有种特殊情况:没有0,全1,那么全1子数组的长度为n, 但是必须删除一个元素,所以最终长度为n-1.
public int longestSubarray(int[] nums) {
int n = nums.length;
int pre0 = -1;
int cur0 = findNextZero(0, nums);
if (cur0 == n) return n - 1; //全1的情况
int res = 0;
while (cur0 != n) {
int next0 = findNextZero(cur0 + 1, nums);
int len = (cur0 - pre0 - 1) + (next0 - cur0 - 1); //两头的0去掉
res = Math.max(res, len);
pre0 = cur0;
cur0 = next0;
}
return res;
}
int findNextZero(int from, int[] arr) {
for (int i = from; i < arr.length; i++) {
if (arr[i] == 0) {
return i;
}
}
return arr.length;
}