leetcode436.寻找右区间

题目大意

给定一组区间,对于每一个区间 i,检查是否存在一个区间 j,它的起始点大于或等于区间 i 的终点,这可以称为 j 在 i 的“右侧”。

对于任何区间,你需要存储的满足条件的区间 j 的最小索引,这意味着区间 j 有最小的起始点可以使其成为“右侧”区间。如果区间 j 不存在,则将区间 i 存储为 -1。最后,你需要输出一个值为存储的区间值的数组。

注意:

  • 你可以假设区间的终点总是大于它的起始点。
  • 你可以假定这些区间都不具有相同的起始点。

示例 1:

输入: [ [1,2] ]
输出: [-1]
**解释**:集合中只有一个区间,所以输出-1

示例 2:

输入: [ [3,4], [2,3], [1,2] ]
输出: [-1, 0, 1]
解释:对于[3,4],没有满足条件的“右侧”区间。
对于[2,3],区间[3,4]具有最小的“右”起点;
对于[1,2],区间[2,3]具有最小的“右”起点。

解题思路

这道题求的是,对于每一个区间(x,y),在区间数组中找到第一个大于等于y(即大于等于y,且最接近y)的区间的索引。
很容易想到另一个题目:给定有序数组和target,找到第一个大于等于target的数字的索引。
因此我们将原区间数组按照区间端点升序排列,同时记录该区间在原排列中的索引。
因此,对于任意一个区间(x,y),y等价于target,所有区间的x等价于有序数组,在其中二分查找即可。

class Solution {
private:
	int findLower(vector<pair<int, int>> & nums, int target){
		int left = 0, right = nums.size() - 1, mid = 0;
		while (left <= right){
			mid = left + (right - left) / 2;
			if (nums[mid].first < target)
				left = mid + 1;
			else
				right = mid - 1;
		}
		return left == nums.size() ? -1 : nums[left].second;
	}
public:
    vector<int> findRightInterval(vector<vector<int>>& intervals) {
    	if (intervals.size() == 1)
    		return {-1};

    	vector<pair<int, int>> nums;
    	for (int i = 0; i < intervals.size(); ++i){
    		nums.push_back({intervals[i][0], i});
    	}

    	sort(nums.begin(), nums.end());

    	vector<int> res(intervals.size(), -1);

    	int tmp = -1, target = 0;

    	for(int i = 0; i < intervals.size(); ++i){
    		target = intervals[i][1];
    		tmp = findLower(nums, target);
    		res[i] = tmp;
    	}

    	return res;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_41092190/article/details/106516931