描述
范围模块是跟踪数字范围的模块。 您的任务是以有效的方式设计和实现以下接口。
- addRange(int left,int right): 添加左闭右开[left,right)的区间,跟踪区间中的每个实数。 如果添加的区间里与已经跟踪的实数部分重合,那么就把区间内没有跟踪的实数也加进去。
- queryRange(int left,int right): 当且仅当当前[left,right)中的每个实数都被跟踪时,返回true。
- removeRange(int left,int right): 停止跟踪[left,right)区间内当前已经跟踪的每个实数。
- 一个左闭右开的区间 [left, right) 包含了 left <= x < right范围内所有的实数.
- 函数 addRange, queryRange, removeRange中参数的取值范围为0 < left < right < 10^9.
- 测试样例中调用addRange的次数最多为 1000.
- 测试样例中调用queryRange 的次数最多为5000.
- 测试样例中调用removeRange的次数最多为 1000.
在线评测地址:领扣题库官网
样例1
输入:
addRange(10,20)
removeRange(14,16)
queryRange(10,14)
queryRange(13,15)
queryRange(16,17)
输出: [true,false,true]
说明:
[10, 14)里的所有数字有已被跟踪
一些数字,例如:14, 14.03, 14.17 [13, 15)并没有被跟踪
尽管有remove的操作,区间[16, 17)中的16仍被跟踪
样例2
输入:
addRange(1,2)
queryRange(2,3)
addRange(11,20)
queryRange(15,20)
输出: [false,true]
代码
from bisect import bisect_left as bl, bisect_right as br
class Solution(object):
def __init__(self):
self.ivs = []
def addRange(self, left, right):
ivs = self.ivs
ilo, ihi = bl(ivs, left), br(ivs, right)
if ilo%2 == 1:
ilo -= 1
left = ivs[ilo]
if ihi%2 == 1:
right = ivs[ihi]
ihi += 1
self.ivs = ivs[:ilo] + [left, right] + ivs[ihi:]
def queryRange(self, left, right):
ivs = self.ivs
ilo = br(ivs, left)
return ilo%2 == 1 and ilo < len(ivs) and ivs[ilo-1] <= left < right <= ivs[ilo]
def removeRange(self, left, right):
ivs = self.ivs
ilo, ihi = bl(ivs, left), br(ivs, right)
new = []
if ilo%2 == 1:
ilo -= 1
new += [ivs[ilo], left]
if ihi%2 == 1:
new += [right, ivs[ihi]]
ihi += 1
self.ivs = ivs[:ilo] + new + ivs[ihi:]
更多题解参考:九章官网solution