leetcode中的差分数组

航班预订统计

https://leetcode.cn/problems/corporate-flight-bookings/
这里有 n 个航班,它们分别从 1 到 n 进行编号。
有一份航班预订表 bookings ,表中第\(i\)条预订记录\(bookings[i] = [firsti, lasti, seatsi]\)意味着在从 \(first_i\)\(last_i\)(包含\(first_i\)\(last_i\))的 每个航班 上预订了\(seats_i\)个座位。
请你返回一个长度为 n 的数组 answer,里面的元素是每个航班预定的座位总数。

示例 1:
输入:bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
输出:[10,55,45,25,25]
解释:
航班编号        1   2   3   4   5
预订记录 1 :   10  10
预订记录 2 :       20  20
预订记录 3 :       25  25  25  25
总座位数:      10  55  45  25  25
因此,answer = [10,55,45,25,25]
示例 2:
输入:bookings = [[1,2,10],[2,2,15]], n = 2
输出:[10,25]
解释:
航班编号        1   2
预订记录 1 :   10  10
预订记录 2 :       15
总座位数:      10  25
因此,answer = [10,25]

提示:
\(1 <= n <= 2 * 10^4\)
\(1 <= bookings.length <= 2*10^4\)
\(bookings[i].length == 3\)
\(1 <= firsti <= lasti <= n\)
\(1 <= seatsi <= 10^4\)

class Solution {
public:
    vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
        int k=bookings.size();
        int sum[n+3];
        memset(sum,0,sizeof(sum));
        for(int i=0;i<k;i++){
            sum[bookings[i][0]]+=bookings[i][2];
            sum[bookings[i][1]+1]-=bookings[i][2];
        }
        for(int i=1;i<=n;i++){
            sum[i]+=sum[i-1];
           /// cout<<sum[i]<<" ";
        }
        //cout<<endl;
        vector<int> ans;
        for(int i=1;i<=n;i++){
            ans.push_back(sum[i]);
        }
        return ans;
        
    }
};

在既定时间做作业的学生人数

https://leetcode.cn/problems/number-of-students-doing-homework-at-a-given-time/description/
给你两个整数数组 startTime(开始时间)和 endTime(结束时间),并指定一个整数 queryTime 作为查询时间。

已知,第 i 名学生在 startTime[i] 时开始写作业并于 endTime[i] 时完成作业。

请返回在查询时间 queryTime 时正在做作业的学生人数。形式上,返回能够使 queryTime 处于区间 [startTime[i], endTime[i]](含)的学生人数。

示例 1:

输入:startTime = [1,2,3], endTime = [3,2,7], queryTime = 4
输出:1
解释:一共有 3 名学生。
第一名学生在时间 1 开始写作业,并于时间 3 完成作业,在时间 4 没有处于做作业的状态。
第二名学生在时间 2 开始写作业,并于时间 2 完成作业,在时间 4 没有处于做作业的状态。
第三名学生在时间 3 开始写作业,预计于时间 7 完成作业,这是是唯一一名在时间 4 时正在做作业的学生。
示例 2:

输入:startTime = [4], endTime = [4], queryTime = 4
输出:1
解释:在查询时间只有一名学生在做作业。
示例 3:

输入:startTime = [4], endTime = [4], queryTime = 5
输出:0
示例 4:

输入:startTime = [1,1,1,1], endTime = [1,3,2,4], queryTime = 7
输出:0
示例 5:

输入:startTime = [9,8,7,6,5,4,3,2,1], endTime = [10,10,10,10,10,10,10,10,10], queryTime = 5
输出:5

class Solution {
public:
    int busyStudent(vector<int>& startTime, vector<int>& endTime, int queryTime) {
        int n=startTime.size();
        int sum[1010];
        memset(sum,0,sizeof(sum));
        for(int i=0;i<n;i++){
            sum[startTime[i]]++;
            sum[endTime[i]+1]--;
        }
        for(int i=1;i<1000;i++){
            sum[i]+=sum[i-1];
        }
        return sum[queryTime];
    }
};
posted @ 2023-12-12 21:50  lipu123  阅读(20)  评论(0)    收藏  举报