436.寻找右区间

61 阅读1分钟

题目:
给你一个区间数组 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]
}