[LeetCode 1094] 差分数组的本质是动态规划

最新推荐文章于 2024-09-22 20:35:45 发布

元无心

最新推荐文章于 2024-09-22 20:35:45 发布

阅读量294

点赞数

分类专栏：算法初探文章标签： leetcode 动态规划算法

本文链接：https://blog.csdn.net/HermitSun/article/details/125447456

版权

算法初探专栏收录该内容

40 篇文章 0 订阅

订阅专栏

本文介绍了如何运用差分数组这一优化技巧来解决动态规划问题，特别是针对给定的一系列区间乘客上下车情况，判断一辆公交车是否能承载所有乘客。通过将区间操作合并为对差分数组的更新，减少了遍历开销，提高了算法效率。在给出的代码示例中，展示了如何用Python实现这个解决方案，检查每公里处的车上人数不超过给定的容量。

摘要由CSDN通过智能技术生成

差分数组其实就是动态规划的一个变体，因为对区间的修改最终还是会落实到每一个元素上。而且，这种修改一般是完全相同的，所以差分数组实际上是对状态修改进行了合并，从而优化了动态规划中 DP 数组的修改开销。

对于这道题来说，其实需要维护的就是在每一公里处车上的人数。并且，由于车是单向行驶的，当前的人数只依赖于上一公里时的人数，符合动态规划的模型。

朴素的想法是每次来一个区间，就遍历这个区间，但由于操作是相同的，可以对重复的操作进行合并，只保留头和尾即可。这也就是差分数组。

class Solution:
    def carPooling(self, trips: List[List[int]], capacity: int) -> bool:
        diff = [0] * 1001
        for num_passengers, fromi, toi in trips:
            diff[fromi] += num_passengers
            diff[toi + 1] -= num_passengers
        s = 0
        for n in diff:
            s += n
            if s > capacity:
                return False
        return True