滑动窗口-区间统计
2020-03-10 13:04:56
问题描述:
给定一个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] (区间 [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
问题求解:
暴力求解O(n ^ 2),显然是会爆掉的。
区间计数问题通常可以考虑使用滑动窗口来在O(n)的时间复杂度内完成求解。
shrink的策略:[start, end]中1的个数 > k的时候start需要右移,并且需要移动到下一个1的位置。
对于符合条件的[start, end]其中区间个数为end - start + 1 - cnt,可以理解为以end结尾的所有区间个数。当所有当前start对应的end都结束了之后,我们就需要移动start到下一个位置,重新开始计算。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | public long intervalStatistics( int [] nums, int k) { long res = 0; int n = nums.length; int start = 0; while (start < n && nums[start] != 0) start += 1; int cnt = 0; for ( int end = start; end < n; end += 1) { if (nums[end] == 1) cnt += 1; while (cnt > k) { cnt -= nums[start]; start += 1; } while (start < n && nums[start] != 0) { cnt -= nums[start]; start += 1; } if (nums[end] == 0) res += end - start + 1 - cnt; } return res; } |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 开发中对象命名的一点思考
· .NET Core内存结构体系(Windows环境)底层原理浅谈
· C# 深度学习:对抗生成网络(GAN)训练头像生成模型
· .NET 适配 HarmonyOS 进展
· .NET 进程 stackoverflow异常后,还可以接收 TCP 连接请求吗?
· 本地部署 DeepSeek:小白也能轻松搞定!
· 基于DeepSeek R1 满血版大模型的个人知识库,回答都源自对你专属文件的深度学习。
· 在缓慢中沉淀,在挑战中重生!2024个人总结!
· 如何给本地部署的DeepSeek投喂数据,让他更懂你
· 大人,时代变了! 赶快把自有业务的本地AI“模型”训练起来!