题目要求
Given a set of intervals, for each of the interval i, check if there exists an interval j whose start point is bigger than or equal to the end point of the interval i, which can be called that j is on the "right" of i.
For any interval i, you need to store the minimum interval j's index, which means that the interval j has the minimum start point to build the"right"relationship for interval i. If the interval j doesn't exist, store -1 for the interval i. Finally, you need output the stored value of each interval as an array.
Note:
You may assume the interval's end point is always bigger than its start point.
You may assume none of these intervals have the same start point.
Example 1:
Input: [[1,2] ]
Output: [-1]
Explanation: There is only one interval in the collection, so it outputs -1.
Example 2:
Input: [[3,4], [2,3], [1,2] ]
Output: [-1, 0, 1]
Explanation: There is no satisfied "right" interval for [3,4].
For [2,3], the interval [3,4] has minimum-"right" start point;
For [1,2], the interval [2,3] has minimum-"right" start point.
Example 3:
Input: [[1,4], [2,3], [3,4] ]
Output: [-1, 2, -1]
Explanation: There is no satisfied "right" interval for [1,4] and [3,4].
For [2,3], the interval [3,4] has minimum-"right" start point.
NOTE: input types have been changed on April 15, 2019. Please reset to default code definition to get new method signature.
假设一个二维的整数数组中每一行表示一个区间,每一行的第一个值表示区间的左边界,第二个值表示区间的右边界。现在要求返回一个整数数组,用来记录每一个边界右侧最邻近的区间。
如 [[1,4], [2,3], [3,4] ]
表示三个区间,[1,4]不存在最邻近右区间,因此 [1,4] 的最邻近右区间的位置为 -1。[2,3]最邻近右区间为 [3,4],因此返回 2,[3,4] 也不存在最临近右区间,因此也返回 -1。最终函数返回数组[-1,2,-1]。
思路 1:二分法
如果我们将区间按照左边界进行排序,则针对每一个区间的右边界,只要找到左边界比这个值大的最小左边界所在的区间即可。这里不能够直接对原来的二维数组进行排序,因为会丢失每一个区间的原始下标位置。代码中采用了内部类 Node 来记录每一个区间的左边界以及每一个区间的原始下标,并对 Node 进行排序和二分法查找。代码如下:
public int[] findRightInterval(int[][] intervals) {int[] result = new int[intervals.length];
Arrays.fill(result, -1);
Node[] leftIndex = new Node[intervals.length];
for(int i = 0 ; i<intervals.length ; i++) {Node n = new Node();
n.index = i;
n.value = intervals[i][0];
leftIndex[i] = n;
}
Arrays.sort(leftIndex);
for(int i = 0 ; i<intervals.length ; i++) {int rightIndex = intervals[i][1];
int left = 0, right = intervals.length-1;
while(left <=right) {int mid = (left + right) / 2;
Node tmp = leftIndex[mid];
if(tmp.value > rightIndex) {right = mid - 1;}else {left = mid + 1;}
}
if(leftIndex[right].value == rightIndex) {result[i] = leftIndex[right].index;
}else if(right<intervals.length-1) {result[i] = leftIndex[right+1].index;
}
}
return result;
}
public static class Node implements Comparable<Node>{
int value;
int index;
@Override
public int compareTo(Node o) {
// TODO Auto-generated method stub
return this.value - o.value;
}
}
思路二:桶排序
换一种思路,有没有办法将所有的区间用一维数组的方式展开,一维数组的每一个下标上的值,都记录了该位置所在的右侧第一个区间。具体流程如下:
- 假设输入为
[3,4], [2,3], [1,2]
的三个区间,展开来后我们知道这里的三个区间横跨了 [1,4] 这么一个区间 - 我们可以用长度为 4 的一维数组 bucket 来表示这个完整的区间,其中每一个下标代表的位置都以左边界作为偏移值,如数组的 0 下标其实代表位置 1,数组的 1 下标代表位置 2。
- 先根据所有区间的左值更新区间数组,如 [3,4] 代表着区间数组中位置为 3 -1= 2 的位置的第一个右侧区间为[3,4], 因此 bucket[2]=0, 同理 bucket[1]=0, bucket[0]=1
- 开始查询时,如果当前桶下标上没有记录右侧的第一个区间,则不停的向右遍历,直到找到第一个值。如果没找到,则说明其右侧不存在区间了。反过来,也可以从右往左遍历 bucket 数组,每遇到一个没有填充的位置,就将右侧的值赋予当前位置。
代码如下:
public int[] findRightInterval(int[][] intervals) {int[] result = new int[intervals.length];
int min = intervals[0][0], max = intervals[0][1];
for(int i = 1 ; i<intervals.length ; i++) {min = Math.min(min, intervals[i][0]);
max = Math.max(max, intervals[i][1]);
}
int[] buckets = new int[max - min + 1];
Arrays.fill(buckets, -1);
for(int i = 0 ; i<intervals.length ; i++) {buckets[intervals[i][0] - min] = i;
}
for(int i = buckets.length-2 ; i>=0 ; i--) {if(buckets[i] == -1) buckets[i] = buckets[i+1];
}
for(int i = 0 ; i<intervals.length ; i++) {result[i] = buckets[intervals[i][1] - min];
}
return result;
}
这里的核心思路在于,如何理解 bucket 数组,这个 bucket 数组本质上是将所有的区间以最左边界和最右边界展开,数组的每一个下标对应着区间中的相对位置,并记录了这个下标右侧的第一个区间的位置。