滑动窗口-区间统计

2020-03-10 13:04:56

问题描述：

给定一个01数组 arr 和 一个整数 k, 统计有多少区间符合如下条件:

1. 区间的两个端点都为 0 (允许区间长度为1)
2. 区间内 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] (区间 [i，j] 表示下标 i（包括）和下标 j（包括）之间的元素)

样例 2:

输入: arr = [1, 1, 1, 0, 0, 1], k = 2
输出: 3
解释: [3, 3], [4, 4], [3, 4] (区间 [i，j] 表示下标 i（包括）和下标 j（包括）之间的元素)

注意事项

arr 的大小不超过 10^5

问题求解：