分析
首先由于start唯一,我们可以根据start还原原intervals的位置,所以应该建立一个dict来记录 然后抽取出start来进行排序
随后按intervals顺序遍历interval,抽取他的end 如果end大于最大的start,就返回-1 否则则通过bisect_left找到第一个大于等于start的位置idx 这个是排序后的idx,对应唯一的start,也对应唯一的原intervals的idx,把这个idx存入ans即可
ac code
class Solution:
def findRightInterval(self, intervals: List[List[int]]) -> List[int]:
sorted_interval = sorted([interval[0] for interval in intervals])
n = len(intervals)
start2idx = defaultdict(int)
for i in range(n):
start2idx[intervals[i][0]] = i
ans = []
for interval in intervals:
end = interval[1]
if end > sorted_interval[-1]:
ans.append(-1)
continue
idx = bisect_left(sorted_interval, end)
ans.append(start2idx[sorted_interval[idx]])
return ans
总结
通过dict来记录start和intervals对应的位置,来找回 根据start排序,通过bisect_left找到end合适的插入位置 通过sorted_idx -> start -> original_idx的映射关系还原
|