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

1450. 在既定时间做作业的学生人数 - 力扣（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

---

解题思路

记s[i]为startTime的第i个元素，e[i]为endTime的第i个元素，q为queryTime

则：

$$\begin{array}{rl}ans& =\sum _{i=0}^{n-1}I(s[i]\le q\le e[i])\\ & =\sum _{i=0}^{n-1}I(q\ge s[i])-I(q>e[i])\\ & =\sum _{i=0}^{n-1}I(q\ge s[i])-I(q\ge e[i]+1)\\ & =\sum _{i=0}^{n-1}(\sum _{j=0}^{q}I(j=s[i])-I(j=e[i]+1))\\ & =\sum _{j=0}^q(\sum _{i=0}^{n-1}I(j=s[i])-I(j=e[i]+1))\\ & =\sum _{j=0}^{q}cnt[j]\end{array}$$

其中:

$$\begin{array}{rl}cnt[j]& =\sum _{i=0}^{n-1}I(j=s[i])-I(j=e[i]+1)\end{array}$$

遍历startTime数组，对cnt[startTime[i]]++, cnt[endTime[i]+1]--，可以得到cnt数组，对cnt累加从而得到结果，此即官方解答中的差分法.

另外：

$$\begin{array}{rl}cnt[j]& =\sum _{i=0}^{n-1}I(j=s[i])-I(j=e[i]+1)\\ & =I(j\in StartTime)-I(j-1\in EndTime)\end{array}$$

所以：

$$\begin{array}{rl}ans& =\sum _{j=0}^{q}cnt[j]\\ & =\sum _{j=0}^{q}I(j\in StartTime)-I(j-1\in EndTime)\\ & =upper\_bound(sortedStartTime,q)-lower\_bound(sortedEndTime,q)\end{array}$$

此即解答中的二分法。