0

0

使用差分数组高效解决“通过操作使数组全为零”问题

心靈之曲

心靈之曲

发布时间:2026-03-18 12:57:23

|

635人浏览过

|

来源于php中文网

原创

本文介绍如何用差分数组优化滑动窗口模拟法,以 o(n) 时间复杂度判断能否通过若干次长度为 k 的子数组减 1 操作,将整数数组全部变为 0。

本文介绍如何用差分数组优化滑动窗口模拟法,以 o(n) 时间复杂度判断能否通过若干次长度为 k 的子数组减 1 操作,将整数数组全部变为 0。

在 LeetCode 题目「Apply Operations to Make All Array Elements Equal to Zero」中,核心约束是:每次只能选择长度恰好为 k 的连续子数组,将其所有元素减 1;操作可执行任意次;目标是让整个数组变为全 0。直观的贪心策略是——从左到右处理,一旦遇到非零值 nums[i],就必须用以 i 为起点的长度为 k 的子数组来“抵消”它(即执行 nums[i] 次减 1 操作)。否则,nums[i] 将永远无法被后续操作覆盖(因操作不能向左延伸),导致失败。

但若按原始思路暴力模拟(如对每个位置反复取窗口最小值、批量减法、再赋回原数组),时间复杂度会退化至 O(nk),在大规模输入下必然超时。关键瓶颈在于:重复计算窗口内累计减量,且无法快速反映历史操作对当前位置的影响

✅ 正确解法是引入 差分数组(Difference Array) 实现 O(1) 区间更新 + O(1) 单点查询:

  • 定义 diff[0..n](长度为 n+1),其中 diff[i] 表示从位置 i 开始的“净减量变化”;
  • 维护一个运行变量 curr 表示当前位置 i 已被前面所有操作影响的总减量(即 diff 的前缀和);
  • 遍历 i ∈ [0, n):
    • 更新 curr += diff[i],得到当前实际已减去的总量;
    • 计算当前位置真实剩余值:real = nums[i] + curr(注意:curr 为负值,代表已减去量);
    • 若 real < 0 → 过度减损,非法;
    • 若 real > 0 → 必须从此处启动 real 次新操作:
      • 检查是否越界:i + k > n ⇒ 无法覆盖,返回 False;
      • 否则,在差分数组中标记:diff[i] -= real(起始减量),diff[i + k] += real(终止补偿);

该算法单次遍历完成,时间复杂度 O(n),空间复杂度 O(n),彻底规避了窗口内 min 查找与数组拷贝开销。

Buildt.ai
Buildt.ai

AI驱动的软件开发平台,可以自动生成代码片段、代码分析及其他自动化任务

下载

以下是完整、可直接提交的 Python 实现:

class Solution:
    def checkArray(self, nums: List[int], k: int) -> bool:
        n = len(nums)
        diff = [0] * (n + 1)  # 差分数组,索引 0~n
        curr = 0  # 当前累计减量(即 diff[0..i] 前缀和)

        for i in range(n):
            curr += diff[i]           # 应用位置 i 的差分更新
            real = nums[i] + curr    # 当前真实剩余值

            if real < 0:
                return False         # 被减成负数,非法
            if real > 0:
                if i + k > n:        # 无法放置长度为 k 的操作区间
                    return False
                diff[i] -= real      # 在 i 处开始 real 次减操作
                diff[i + k] += real  # 在 i+k 处撤销影响

        return True

⚠️ 注意事项:

  • 差分数组 diff 长度必须为 n+1,以安全支持 diff[n] 赋值(避免索引越界);
  • real = nums[i] + curr 中 curr 为负值(表示已减量),因此 real 是剩余待处理值;
  • 所有判断(越界、负值)必须在应用当前 real 操作前完成,确保逻辑原子性;
  • 无需显式检查最终数组是否为 0 —— 贪心+差分保证:若全程合法,则结尾必全为 0。

总结:本题是差分数组的经典应用场景。它将“多次区间减法”的累积效应压缩为前缀和维护,把原本 O(nk) 的模拟降维至线性。掌握该模式,可高效解决一类“区间增/减 + 贪心决策”的构造性数组问题。

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
页面置换算法
页面置换算法

页面置换算法是操作系统中用来决定在内存中哪些页面应该被换出以便为新的页面提供空间的算法。本专题为大家提供页面置换算法的相关文章,大家可以免费体验。

508

2023.08.14

抖漫入口地址合集
抖漫入口地址合集

本专题整合了抖漫入口地址相关合集,阅读专题下面的文章了解更多详细地址。

17

2026.03.17

多环境下的 Nginx 安装、结构与运维实战
多环境下的 Nginx 安装、结构与运维实战

本专题聚焦多环境下Nginx实战,详解开发、测试及生产环境的差异化安装策略与目录结构规划。深入剖析配置模块化设计、灰度发布流程及跨环境同步机制。结合监控告警、故障排查与自动化运维工具,提供全链路管理方案,助力团队构建灵活、高可用的Nginx服务体系,从容应对复杂业务场景挑战。

1

2026.03.17

PS 批量添加图片
PS 批量添加图片

本专题整合了PS批量添加图片教程合集,阅读专题下面的文章了解更多详细操作。

3

2026.03.17

Nginx 基础架构:从安装配置到系统化管理
Nginx 基础架构:从安装配置到系统化管理

本专题深入解析Nginx基础架构,涵盖从源码编译与包管理安装,到核心配置文件优化及虚拟主机部署。进一步探讨日志轮转、性能调优、高可用集群构建及自动化运维策略,助力管理员实现从单一服务搭建到企业级系统化管理的全面升级,确保Web服务高效、稳定运行。

4

2026.03.17

mulerun骡子快跑入口地址汇总
mulerun骡子快跑入口地址汇总

本专题整合了mulerun入口地址合集,阅读专题下面的文章了解更多详细内容。

64

2026.03.17

源码编译安装Nginx详解:模块选择、依赖准备与常见错误排查
源码编译安装Nginx详解:模块选择、依赖准备与常见错误排查

本专题详解Nginx源码编译全流程:从GCC、OpenSSL等依赖准备,到按需定制HTTP/SSL/流媒体模块的configure参数策略。深入剖析“缺少库文件”、“配置选项冲突”及“权限错误”等常见报错,提供精准排查思路与解决方案。助您掌握灵活构建高性能、定制化Nginx的核心技能,满足复杂生产环境需求。

1

2026.03.17

Linux环境安装Nginx全流程:apt、yum与源码编译方式深度实操
Linux环境安装Nginx全流程:apt、yum与源码编译方式深度实操

本专题深度实操Linux下Nginx三大安装方式:apt/yum包管理器快速部署,适合新手与标准化运维;源码编译灵活定制模块,满足高性能与特殊需求场景。内容涵盖环境准备、依赖安装、配置优化及平滑升级策略,对比各方案优劣,助您根据业务场景选择最佳实践,构建稳定高效的Web服务基石。

6

2026.03.17

c++ 字符处理
c++ 字符处理

本专题整合了c++字符处理教程、字符串处理函数相关内容,阅读专题下面的文章了解更多详细内容。

7

2026.03.17

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号