1450. 在既定时间做作业的学生人数
题目:给你两个整数数组 startTime(开始时间)和 endTime(结束时间),并指定一个整数 queryTime 作为查询时间。
已知,第 i 名学生在 startTime[i] 时开始写作业并于 endTime[i] 时完成作业。
请返回在查询时间 queryTime 时正在做作业的学生人数。形式上,返回能够使 queryTime 处于区间 [startTime[i], endTime[i]](含)的学生人数。
链接 https://leetcode.cn/problems/number-of-students-doing-homework-at-a-given-time
个人思路
- 按照题意就是当查询时间大于等于startTime,同时小于等于endTime就是表明该学生正在做作业。
class Solution:
def busyStudent(self, startTime: List[int], endTime: List[int], queryTime: int) -> int:
res = 0
for i,j in zip(startTime,endTime):
if i <= queryTime and j >= queryTime:
res += 1
return res
官方代码
class Solution:
def busyStudent(self, startTime: List[int], endTime: List[int], queryTime: int) -> int:
return sum(s <= queryTime <= e for s, e in zip(startTime, endTime))
时间复杂度:O(n),其中 n 为 数组的长度。只需遍历一遍数组即可。
空间复杂度:O(1)。
官方思路
- 差分数组
class Solution:
def busyStudent(self, startTime: List[int], endTime: List[int], queryTime: int) -> int:
maxEndTime = max(endTime)
if queryTime > maxEndTime:
return 0
cnt = [0] * (maxEndTime + 2)
for s, e in zip(startTime, endTime):
cnt[s] += 1
cnt[e + 1] -= 1
return sum(cnt[:queryTime + 1])
- 二分查找
对于每个学生的作业时间[startTime[i],endTime[i]],一定满足 [i]startTime[i]≤endTime[i]。如果第 i 名学生在 queryTime 时正在作业,则一定满足 startTime[i]≤queryTime≤endTime[i]。设起始时间小于等于 queryTime 的学生集合为 lessStart,设结束时间小于 queryTime 的学生集合为 lessEnd,则根据上述推理可以知道 lessEnd∈lessStart,我们从 lessStart 去除 lessEnd 的子集部分即为符合条件的学生集合。因此我们通过二分查找找到始时间小于等于 queryTime 的学生人数,然后减去结束时间小于 queryTime 的学生人数,最终结果即为符合条件要求。
class Solution:
def busyStudent(self, startTime: List[int], endTime: List[int], queryTime: int) -> int:
startTime.sort()
endTime.sort()
return bisect_right(startTime, queryTime) - bisect_left(endTime, queryTime)
复杂度分析
时间复杂度:O(nlogn),其中 n 为 数组的长度。排序需要的时间为 O(nlogn),二分查找的时间复杂度为 O(logn)。
空间复杂度:O(logn)。排序需要的栈空间为 O(logn)。
评论区大佬点评:如果queryTime是个很长的数组,但所有时间值都不大,方法二和方法三就都有意义,如果queryTime是个很长的数组,而且时间最大能到10^9,那么就只能方法三
作者:LeetCode-Solution
链接:https://leetcode.cn/problems/number-of-students-doing-homework-at-a-given-time/solution/zai-ji-ding-shi-jian-zuo-zuo-ye-de-xue-s-uv49/