class Diff:def __init__(self,nums):n = len(nums)self.diff=[0]*(n)for i in range(1,n):self.diff[i]=nums[i]-nums[i-1]def increment(self,i,j,val,n):self.diff[i]+=valif j+1<n:self. diff[j+1]-=valdef result(self,n):res =[0]*nres[0]= self.diff[0]for i in range(1,n):res[i]=res[i-1]+ self.diff[i]return resclass Solution:def corpFlightBookings(self, bookings: List[List[int]], n: int) -> List[int]:diff = Diff([0]*n)for booking in bookings:first =booking[0]-1last = booking[1]-1val =booking[2]diff.increment(first,last,val,n)return diff.result(n)
求最后加法的结果
