LeetCode刷题插分数组技巧

最新推荐文章于 2023-09-28 13:05:25 发布

原创最新推荐文章于 2023-09-28 13:05:25 发布 · 288 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#hashmap #leetcode

刷题专栏收录该内容

23 篇文章

订阅专栏

本文探讨了LeetCode上的航班预订统计问题，介绍了两种解决方法：一是使用hashmap辅助的暴力遍历叠加解法，但存在超时问题；二是利用差分数组技巧，通过多次对区间进行加减操作，实现更高效的解决方案。

LeetCode刷题插分数组技巧

具体方法参考labuladong的论那些小而美的算法技巧：差分数组/前缀和

1109. 航班预订统计

方法一: 使用hashmap来辅助的暴力遍历叠加解法（超时）

class Solution(object):
    # 用hash辅助的暴力法, 但是会超时
    def corpFlightBookings(self, bookings, n):
        """
        :type bookings: List[List[int]]
        :type n: int
        :rtype: List[int]
        """
        if n == 0:
            return []

        hashmap = dict()
        for book in bookings:
            for i in range(book[0] - 1, book[1]):
                hashmap[i] = hashmap.get(i, 0) + book[2]
        
        res = []
        for k in range(n):
            res.append(hashmap.get(k, 0))

        return res

方法二: 利用插分数组, 多次对区间进行加减

import copy
class Difference:
    def __init__(self, n):
        self.n = n
        self.nums = [0] * n
        self.diff = copy.copy(self.nums)
        
        for i in range(1, n):
            self.diff[i] = self.nums[i] - self.nums[i-1]


    def increase(self, i, j, k):
        self.diff[i] += k
        if j + 1 < self.n:
            self.diff[j + 1] -= k

    def result(self):
        res = copy.copy(self.diff)
        for i in range(1, self.n):
            res[i] += res[i - 1]
        return res

    
class Solution(object):
    # 用hash辅助的暴力法, 但是会超时
    def corpFlightBookings(self, bookings, n):
        df = Difference(n)
        for book in bookings:
            df.increase(book[0]-1, book[1]-1, book[2])
        return df.result()