【LeetCode】436. 寻找右区间

题目

给你一个区间数组 intervals ,其中 intervals[i] = [starti, endi] ,且每个 starti 都 不同 。
区间 i 的 右侧区间 可以记作区间 j ,并满足 startj >= endi ,且 startj 最小化 。
返回一个由每个区间 i 的 右侧区间 在 intervals 中对应下标组成的数组。如果某个区间 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 * 104
intervals[i].length == 2
-106 <= starti <= endi <= 106
每个间隔的起点都 不相同

题解

扫描二维码关注公众号,回复: 14982271 查看本文章

排序+二分查找
存储数对的左值和对应的下标并排序
遍历intervals数组,对于每个右值使用二分查找找大于等于该右值的左值

class Solution {
    
    
public:
    vector<int> findRightInterval(vector<vector<int>>& intervals) {
    
    
        int n = intervals.size();
        if(n==1)
            return {
    
    -1};
            
        vector<int> res(n,-1);
        vector<pair<int,int>> left_nums;
        
        for(int i=0;i<n;i++)
        {
    
    
            left_nums.emplace_back(intervals[i][0],i);
        }
        sort(left_nums.begin(),left_nums.end());
        
        for(int i=0;i<n;i++)
        {
    
    
            auto bound = lower_bound(left_nums.cbegin(),left_nums.cend(),pair<int,int>(intervals[i][1],0));
            if(bound != left_nums.end())
                res[i] = bound->second;
        }
        return res;
    }
};

双指针

class Solution {
    
    
public:
    vector<int> findRightInterval(vector<vector<int>>& intervals) {
    
    
        int n = intervals.size();
        vector<pair<int,int>> left_vals;
        vector<pair<int,int>> right_vals;
        vector<int> res(n,-1);
        for(int i=0;i<n;i++)
        {
    
    
            left_vals.emplace_back(make_pair(intervals[i][0],i));
            right_vals.emplace_back(make_pair(intervals[i][1],i));
        }

        sort(left_vals.begin(),left_vals.end());
        sort(right_vals.begin(),right_vals.end());

        int i=0,j=0;
        while(i!=n && j!=n)
        {
    
    
            while(i!=n && left_vals[i].first < right_vals[j].first)
                i++;
            if(i<n)
                res[right_vals[j].second] = left_vals[i].second;
            j++;
            
        }
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_45972928/article/details/126370117