算法竞赛之差分进阶——等差数列差分 python

目录


前置知识

给定区间 l, r ,让我们把数组中的 l, r 区间中的每一个数加上c,即 a l + c , a l + 1 + c , a l + 2 + c , a r + c;

怎么做?很简单,差分一下即可

还不会的小伙伴点此进入学习


进入正题

进阶一下:

给定区间 l, r ,把数组 l, r 区间中的数加上一个首项s、末项e、公差为d的等差数列,

即 a l + s , a l + 1 + s+d , a l + 2 + s+2d ······a r + e

怎么实现?先给出结论

al += s,
al + 1 += d - s
ar + 1 -=d + e
ar + 2 += e

再对a数组做两次前缀和,即得到ans

为何?

下面听我娓娓道来~


简单举个例子:
假设数组a=【0,0,0,0,0,0,0,0】
现要求对 a1 到 a5 这5个数字 分别加上以s为首项,d为公差,e为末项的等差数列,即a=【0,s,s+d,s+2d,s+3d,s+4d(e),0,0】
如何得到呢?我们先做一次差分试试:
diff1=【0,s,d,d,d,d,-e,0】什么也看不出来对吧。
再对差分数组做差分:
diff2=【0,s,d-s,0,0,0,-e-d,e】
哎,这不是一开始所进行的操作吗?
a1+=s
a2+=d-s
a6-=d+e
a7+=e
一切终成闭环


好了,实际运用的时候到了

实战演练

P4231 三步必杀 https://www.luogu.com.cn/problem/P4231



不了解异或运算可点此进入

题解code:

python 复制代码
n, m = map(int, input().split())
ans = [0] * (n + 3)
for i in range(m):
    l, r, s, e = map(int, input().split())
    d = int((e - s) / (r - l))
    ans[l] += s
    ans[l + 1] += d - s
    ans[r + 1] -= d + e
    ans[r + 2] += e    # 实现等差数列差分

for i in range(1, len(ans)):
    ans[i] += ans[i - 1]
for i in range(1, len(ans)):
    ans[i] += ans[i - 1]   # 两次前缀和

xor = 0
for i in ans:
    xor ^= i
print(f'{xor} {max(ans)}')

如果有更多问题或需要进一步的帮助,可以在评论区留言讨论哦!
如果喜欢的话,请给博主点个关注 谢谢

相关推荐
Wang ruoxi4 小时前
Pygame 小游戏——贪吃蛇
python·pygame
aWty_4 小时前
实分析入门(11)--Cantor三分集
学习·数学·算法·实变函数
兰令水4 小时前
leecodecode【二叉树递归+对称】【2026.6.1打卡-java版本】
算法
大数据魔法师9 小时前
Streamlit(二十三)- 教程(二)- 动态导航
python·web
心中有国也有家11 小时前
GE图引擎深度解析——CANN的计算图优化与执行引擎
人工智能·pytorch·python·学习·numpy
地平线开发者12 小时前
profiler debug 工具用法与高一致性策略
算法·自动驾驶
卷毛的技术笔记12 小时前
告别硬编码!Spring AI Alibaba 实现 AI Agent 智能工具调用(Tool Calling)
java·人工智能·后端·python·spring·ai编程
编程大师哥12 小时前
匿名函数 lambda + 高阶函数
java·python·算法
vb20081112 小时前
FastAPI APIRouter
开发语言·python
adrninistrat0r13 小时前
Java调用链MCP分析工具
java·python·ai编程