题目:
给你一个区间数组 intervals ,其中 intervals[i] = [starti, endi] ,且每个 starti 都 不同 。
区间 i 的 右侧区间 可以记作区间 j ,并满足 startj`` >= endi ,且 startj 最小化 。
返回一个由每个区间 i 的 右侧区间 在 intervals 中对应下标组成的数组。如果某个区间 i 不存在对应的 右侧区间 ,则下标 i 处的值设为 -1 。
算法:
func findRightInterval(intervals [][]int) []int {
for i := range intervals {
intervals[i] = append(intervals[i], i)
}
sort.Slice(intervals, func(i, j int) bool {
return intervals[i][0] < intervals[j][0]
})
ans := make([]int, len(intervals))
// 对每个i,查找intervals[j][0]大于等于intervals[i][1]的最小值j即可
// 思路是利用二分查找
for i := range intervals {
index := search(intervals, i)
ans[intervals[i][2]] = index
}
return ans
}
// 从intervals查找大于等于intervals[i][1]的最小的
func search(intervals [][]int, index int) int {
// 左右区间逼近思想,[left:right]最后可能是[len(intervals) - 1, intervals],所以不能设置right = len(intervals) - 1
left, right := index, len(intervals)
for left < right {
mid := (left + right) / 2
if intervals[mid][0] < intervals[index][1] {
left = mid + 1
} else {
right = mid
}
}
if left >= len(intervals){
return - 1
}
return intervals[left][2]
}