LeetCode 436. 寻找右区间--二分+排序

436. 寻找右区间

给你一个区间数组 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 * 104
intervals[i].length == 2
-106 <= starti <= endi <= 106
每个间隔的起点都 不相同

题解

比较简单，直接排序后二分答案即可。

AC代码

class Solution {
public:
    struct Node
    {
        int l,r,id;
    };
    static int cmp(Node a1,Node a2)
    {
        if(a1.l<a2.l)return true;
        if(a1.l==a2.l)return a1.r<a2.r;
        return false;
    }
    vector<Node>q;
    int pos[20010];
    vector<int> findRightInterval(vector<vector<int>>& intervals) {
        for(int i=0;i<intervals.size();i++)
        {
            Node t;
            t.l=intervals[i][0],t.r=intervals[i][1],t.id=i;
            q.push_back(t);
        }
        sort(q.begin(),q.end(),cmp);
        memset(pos,-1,sizeof(pos));
        for(int i=0;i<q.size();i++)
        {
            int l=i,r=q.size()-1,fin=-1;
            while(l<=r)
            {
                int mid=(l+r)/2;
                if(q[mid].l>=q[i].r)
                {
                    r=mid-1;
                    fin=mid;
                }
                else l=mid+1;
            }
            if(fin!=-1)
            pos[q[i].id]=q[fin].id;
        }
        vector<int>res;        
        for(int i=0;i<q.size();i++)
        {
            res.push_back(pos[i]);
        }
        return res;
    }
};