题目
给你一个区间数组 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]
break
for j in range(i+1,len(intervals)):
if temp[j][1][0]>=temp[i][1][1]:
ans[temp[i][0]]=temp[j][0]
break
if j==len(intervals)-1 and temp[j][1][0]<temp[i][1][1]:
ans[temp[i][0]]=-1
return 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
|