1667. 区间统计
CAT 专属题目
中文English
给定一个01数组 arr
和 一个整数 k
, 统计有多少区间符合如下条件:
- 区间的两个端点都为 0 (允许区间长度为1)
- 区间内 1 的个数不多于
k
样例
样例 1:
输入: arr = [0, 0, 1, 0, 1, 1, 0], k = 1
输出: 7
解释: [0, 0], [1, 1], [3, 3], [6, 6], [0, 1], [0, 3], [1, 3]
样例 2:
输入: arr = [1, 1, 1, 0, 0, 1], k = 2
输出: 3
解释: [3, 3], [4, 4], [3, 4]
注意事项
arr
的大小不超过 10^5
public class Solution {
/**
* @param arr: the 01 array
* @param k: the limit
* @return: the sum of the interval
*/
public long intervalStatistics(int[] arr, int k) {
if (arr == null || arr.length == 0) return 0;
int left = 0, right = 0;
long numOfOne = 0, sum = 0;
while (right < arr.length){
if (arr[right] == 1){
numOfOne++;
right++;
continue;
}
if (arr[left] == 1){
numOfOne--;
left++;
continue;
}
while (numOfOne > k){
if (arr[left] == 1){
numOfOne--;
}
left++;
}
sum += right - left + 1 - numOfOne;
right++;
}
return sum;
}
}