【LeetCode】1094 and 1109 （差分数组专题）

差分数组的思想可参考小而美的算法技巧：差分数组

1094. 拼车

解法：差分数组
开车经过的地点可以视为一个数组，各段旅程上下车可以理解为在子区间内进行加减操作。而差分数组的主要适⽤场景是频繁对原始数组的某个区间的元素进⾏增减。因此可以考虑使用差分数组进行求解。

class Solution:
    def carPooling(self, trips: List[List[int]], capacity: int) -> bool:
        diff = [0] * 1001
        diff[0] = capacity
        for trip in trips:
            diff[trip[1]] -= trip[0]
            diff[trip[2]] += trip[0]
        # print(diff)
        value = diff[0]
        if value < 0:
            return False
        for i in range(1, len(diff)):
            value = value + diff[i]
            # print(value)
            if value < 0:
                return False
        return True

1109. 航班预定统计

解法：差分数组
航班预定同样可以理解为在数组的子区间进行频繁增减操作。因此也可利用差分数组进行求解。

class Solution:
    def corpFlightBookings(self, bookings: List[List[int]], n: int) -> List[int]:
        nums = [0] * n
        for left, right, inc in bookings:
            nums[left - 1] += inc
            if right < n:
                nums[right] -= inc
    
        for i in range(1, n):
            nums[i] += nums[i - 1]
        
        return nums