航班预订统计算法详解编程语言

这里有 n 个航班,它们分别从 1 到 n 进行编号。

我们这儿有一份航班预订表,表中第 i 条预订记录 bookings[i] = [i, j, k] 意味着我们在从 i 到 j 的每个航班上预订了 k 个座位。

请你返回一个长度为 n 的数组 answer,按航班编号顺序返回每个航班上预订的座位数。

解:这道题用暴力的方法很简单,用一个set按航班号为key存储就好了

下面这种方法,只用了O(N)的复杂度就解决了

首先我们将对一个区间的操作转化为左、右两个端点上的操作

如对于 bookings = [1, 3, 10] n = 5

我们正常解法为在 1-3 位置上各加1如图所示

 航班预订统计算法详解编程语言

 

 

转换为使用端点表示为:从1开始+10, 从4开始不加,那我们就标记位置1为10,位置4 为 -10如图:

 航班预订统计算法详解编程语言

 

 

在统计时 记res[i] = res[i] + res[i – 1] (i > 0)

 航班预订统计算法详解编程语言

 

 

那么对于 多个区间,我们只需要叠加起来即可

航班预订统计算法详解编程语言

 

 

class Solution { 
public: 
    vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) { 
        int length =bookings.size(); 
        //tag记录第i个相比i-1的变量 
        vector<int> tag(n,0); 
        int l=0,r=0,seat=0; 
        for(int index=0;index<length;index++) 
        { 
            //作为下标,减掉1从0开始 
            l=bookings[index][0]-1; 
            r=bookings[index][1]-1; 
            seat=bookings[index][2]; 
            tag[l]+=seat; 
            if(r+1<n) 
            { 
                tag[r+1]-=seat; 
            } 
        } 
 
        //这里之后tag表示的就不是差值而是每次相加后的实际值 
        //第一个值不算差值么,这个是我写代码时疑惑的地方,想明白后其实第一个值就是差值自身 
        for(int index=1;index<n;index++) 
        { 
            tag[index]+=tag[index-1]; 
        } 
        return tag; 
    } 
};

 

原创文章,作者:ItWorker,如若转载,请注明出处:https://blog.ytso.com/17563.html

(0)
上一篇 2021年7月19日
下一篇 2021年7月19日

相关推荐

发表回复

登录后才能评论