【1450. 在既定时间做作业的学生人数】

最新推荐文章于 2024-10-05 12:07:50 发布

千北@

最新推荐文章于 2024-10-05 12:07:50 发布

阅读量119

点赞数

分类专栏： LeetCode 文章标签： c++ leetcode 算法数据结构

本文链接：https://blog.csdn.net/Sugar_wolf/article/details/126418893

版权

LeetCode 专栏收录该内容

397 篇文章 2 订阅

订阅专栏

来源：力扣（LeetCode）

描述：

给你两个整数数组 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

提示：

startTime.length == endTime.length
1 <= startTime.length <= 100
1 <= startTime[i] <= endTime[i] <= 1000
1 <= queryTime <= 1000

方法一：二分查找

对于每个学生的作业时间 [startTime[i],endTime[i]]，一定满足 startTime[i] ≤ endTime[i]。如果第 i 名学生在 queryTime 时正在作业，则一定满足 startTime[i] ≤ queryTime ≤ endTime[i]。设起始时间小于等于 queryTime 的学生集合为 lessStart，设结束时间小于 queryTime 的学生集合为 lessEnd，则根据上述推理可以知道 lessEnd ∈ lessStart，我们从 lessStart 去除 lessEnd 的子集部分即为符合条件的学生集合。因此我们通过二分查找找到始时间小于等于 queryTime 的学生人数，然后减去结束时间小于 queryTime 的学生人数，最终结果即为符合条件要求。

代码：

class Solution {
public: 
    int busyStudent(vector<int>& startTime, vector<int>& endTime, int queryTime) {
        sort(startTime.begin(), startTime.end());
        sort(endTime.begin(), endTime.end());
        int lessStart = upper_bound(startTime.begin(), startTime.end(), queryTime) - startTime.begin();
        int lessEnd = lower_bound(endTime.begin(), endTime.end(), queryTime) - endTime.begin();
        return lessStart - lessEnd;
    }
};

执行用时：4 ms, 在所有 C++ 提交中击败了54.33%的用户
内存消耗：10.5 MB, 在所有 C++ 提交中击败了70.00%的用户
复杂度分析
时间复杂度： O(nlogn)，其中 n 为数组的长度。排序需要的时间为 O(nlogn)，二分查找的时间复杂度为 O(logn)。
空间复杂度： O(logn)。排序需要的栈空间为 O(logn)。

方法二：差分数组

代码：

class Solution {
public:
    int busyStudent(vector<int>& startTime, vector<int>& endTime, int queryTime) {
        int n = startTime.size();
        int maxEndTime = *max_element(endTime.begin(), endTime.end());
        if (queryTime > maxEndTime) {
            return 0;
        }
        vector<int> cnt(maxEndTime + 2);
        for (int i = 0; i < n; i++) {
            cnt[startTime[i]]++;
            cnt[endTime[i] + 1]--;
        }
        int ans = 0;
        for (int i = 0; i <= queryTime; i++) {
            ans += cnt[i];
        }
        return ans;
    }
};

执行用时：0 ms, 在所有 C++ 提交中击败了100.00%的用户
内存消耗：10.8 MB, 在所有 C++ 提交中击败了5.67%的用户
复杂度分析
时间复杂度： O(n+queryTime)，其中 n 为数组的长度， queryTime 为给定的查找时间。首先需要遍历一遍数组，需要的时间为 O(n)，然后需要查分求和求出 queryTime 时间点正在作业的学生总数，需要的时间为 O(queryTime)，因此总的时间为 O(n+queryTime)。
空间复杂度：O(max(endTime))。

方法三：枚举

题目要求找到 queryTime 时正在做作业的学生人数，第 i 名学生的起始时间 startTime[i] 和完成时间 endTime[i] 如果满足 startTime[i] ≤ queryTime ≤ endTime[i]，则可知该名学生在 queryTime 时一定正在作业。我们遍历所有学生的起始时间和结束时间，统计符合上述条件的学生总数即可。

代码：

class Solution {
public:
    int busyStudent(vector<int>& startTime, vector<int>& endTime, int queryTime) {
        int n = startTime.size();
        int ans = 0;
        for (int i = 0; i < n; i++) {
            if (startTime[i] <= queryTime && endTime[i] >= queryTime) {
                ans++;
            }
        }
        return ans;
    }
};

执行用时：4 ms, 在所有 C++ 提交中击败了54.33%的用户
内存消耗：10.5 MB, 在所有 C++ 提交中击败了87.00%的用户
复杂度分析
时间复杂度： O(n)，其中 n 为数组的长度。只需遍历一遍数组即可。
空间复杂度： O(1)。
author：LeetCode-Solution