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];
}
};