航班预订统计(差分数组)

Posted 2021-09-15 秦枫-_-

解题思路：
差分数组对应的概念是前缀和数组，对于数组 [1,2,2,4][1,2,2,4]，其差分数组为 [1,1,0,2][1,1,0,2]，差分数组的第 ii 个数即为原数组的第 i-1i−1 个元素和第 ii 个元素的差值，也就是说我们对差分数组求前缀和即可得到原数组。

ans数组刚开始记录的是每个数和他前一个数的差值即表示为差分数组,那么我们可以举例[1,2,10]，ans[0]=10,ans[1]=0，ans[2]=-10，表示的就是此时idx=1跟他前一个数idx=0相差为0，即num[0]=num[1]=10，idx=2，跟idx=1相差为-10,即num[1]=10，num[2]=0

class Solution {
    public int[] corpFlightBookings(int[][] bookings, int n) {
    int []ans=new int[n];
    for(int []ch:bookings){
        ans[ch[0]-1]+=ch[2];
        if(ch[1]<n)ans[ch[1]]-=ch[2];//ans此时记录的是相邻两个数的差，也即差分数组
    }
    for(int i=1;i<n;i++){
        ans[i]+=ans[i-1];
    }
    return ans;
    }
}