题目
给你一个区间数组 intervals ,其中 intervals[i] = [starti, endi] ,且每个 starti 都 不同 。
区间 i 的 右侧区间 可以记作区间 j ,并满足 startj >= endi ,且 startj 最小化 。
返回一个由每个区间 i 的 右侧区间 的最小起始位置组成的数组。如果某个区间 i 不存在对应的 右侧区间 ,则下标 i 处的值设为 -1 。
示例 1:
输入:intervals = [[1,2]]
输出:[-1]
解释:集合中只有一个区间,所以输出-1。
示例 2:
输入:intervals = [[3,4],[2,3],[1,2]]
输出:[-1,0,1]
解释:对于 [3,4] ,没有满足条件的“右侧”区间。
对于 [2,3] ,区间[3,4]具有最小的“右”起点;
对于 [1,2] ,区间[2,3]具有最小的“右”起点。
示例 3:
输入:intervals = [[1,4],[2,3],[3,4]]
输出:[-1,2,-1]
解释:对于区间 [1,4] 和 [3,4] ,没有满足条件的“右侧”区间。
对于 [2,3] ,区间 [3,4] 有最小的“右”起点。
提示:
1 <= intervals.length <= 2 * 10^4
intervals[i].length == 2
-10^6 <= starti <= endi <= 10 ^6
每个间隔的起点都 不相同
来源:力扣(LeetCode)
解题思路
这个题目的意思大致是,对于intervals中的每一个元素i,在所有元素的第一个位置上找到大于或等于i[1]的元素的下标。通常的做法就是给整个intervals按照第一个元素排个序,然后双层遍历找到每个元素的符合条件的下标(即,刚刚好大于或者等于)。
class Solution:def findRightInterval(self, intervals: List[List[int]]) -> List[int]:if len(intervals)==1:return [-1]ans=[-1]*len(intervals)temp=list(enumerate(intervals))temp.sort(key=lambda x:x[1])for i in range(len(intervals)):if temp[i][1][0]==temp[i][1][1]:ans[temp[i][0]]=temp[i][0]breakfor j in range(i+1,len(intervals)):if temp[j][1][0]>=temp[i][1][1]:ans[temp[i][0]]=temp[j][0]breakif j==len(intervals)-1 and temp[j][1][0]<temp[i][1][1]:ans[temp[i][0]]=-1return ans
非常明显,这种时间复杂度达到O(n^2)的暴力算法通常耗时非常久。在前面我们对intervals数组的操作中有一步是排序,对于此我们可以有效利用二分查找来解决这个问题。
class Solution:def findRightInterval(self, intervals: List[List[int]]) -> List[int]:temp=[i[0] for i in intervals]temp=list(enumerate(temp))temp.sort(key=lambda x:x[1])TEMP=[i[1] for i in temp]ans=[]for i in intervals:index=bisect.bisect_left(TEMP,i[1])if index==len(temp):ans.append(-1)else:ans.append(temp[index][0])return ans