今日份题目:
给你一个二维整数数组 intervals ,其中 intervals[i] = [lefti, righti] 表示第 i 个区间开始于 lefti 、结束于 righti(包含两侧取值,闭区间)。区间的 长度 定义为区间中包含的整数数目,更正式地表达是 righti - lefti + 1 。
再给你一个整数数组 queries 。第 j 个查询的答案是满足 lefti <= queries[j] <= righti 的 长度最小区间 i 的长度 。如果不存在这样的区间,那么答案是 -1 。
以数组形式返回对应查询的所有答案。
示例1
输入:intervals = [[1,4],[2,4],[3,6],[4,4]], queries = [2,3,4,5]
输出:[3,3,1,4]
解释:查询处理如下:
- Query = 2 :区间 [2,4] 是包含 2 的最小区间,答案为 4 - 2 + 1 = 3 。
- Query = 3 :区间 [2,4] 是包含 3 的最小区间,答案为 4 - 2 + 1 = 3 。
- Query = 4 :区间 [4,4] 是包含 4 的最小区间,答案为 4 - 4 + 1 = 1 。
- Query = 5 :区间 [3,6] 是包含 5 的最小区间,答案为 6 - 3 + 1 = 4 。
示例2
输入:intervals = [[2,3],[2,5],[1,8],[20,25]], queries = [2,19,5,22]
输出:[2,-1,4,6]
解释:查询处理如下:
- Query = 2 :区间 [2,3] 是包含 2 的最小区间,答案为 3 - 2 + 1 = 2 。
- Query = 19:不存在包含 19 的区间,答案为 -1 。
- Query = 5 :区间 [2,5] 是包含 5 的最小区间,答案为 5 - 2 + 1 = 4 。
- Query = 22:区间 [20,25] 是包含 22 的最小区间,答案为 25 - 20 + 1 = 6 。
提示
1 <= intervals.length <= 105
1 <= queries.length <= 105
intervals[i].length == 2
1 <= lefti <= righti <= 107
1 <= queries[j] <= 107
题目思路
暴力算法思路:对每个数字,遍历所有区间找到满足的区间并对比得出最小区间根据公式输出结果。
优化算法思路:离线算法+优先队列。所谓离线算法,就是可以对区间进行排序。将所有满足left<=queries<=right的区间都放入队列中,后不断获取队首区间,如果队首区间的右端点right<queries[j],那么说明该区间不满足条件,从队列中出队;如果队首区间的右端点right>=queries[j],那么该区间为第j个查询的最小区间,终止过程。
代码
class Solution
{
public:
vector<int> minInterval(vector<vector<int>>& intervals, vector<int>& queries)
{
vector<int> index(queries.size());
iota(index.begin(), index.end(), 0);//产生从0开始的连续的值
sort(index.begin(), index.end(), [&](int i, int j) -> bool{return queries[i] < queries[j];});//按查询值从小到大排序
sort(intervals.begin(), intervals.end(), [](const vector<int> &it1, const vector<int> &it2) -> bool{return it1[0] < it2[0];});//根据第一个元素的大小,对区间按从小到大排序
priority_queue<vector<int>> pq;//优先队列
vector<int> result(queries.size(), -1);//存放最后结果,找不到符合的区间就是-1,所以初始化为-1
int i = 0;
for (auto j : index)
{
while (i < intervals.size() && intervals[i][0] <= queries[j]) //查找可行区间
{
int l = intervals[i][1] - intervals[i][0] + 1;//区间长度
//优先队列默认最大堆,所以用-l可以找到最小区间
pq.push({-l, intervals[i][0], intervals[i][1]});//优先队列插入区间长度和区间范围
i++;//遍历所有区间
}
//前边只是按照左端值判断区间是否可用,下面要排除不可用区间
while (!pq.empty() && pq.top()[2] < queries[j]) //区间不满足条件,出队
{
pq.pop();
}
if (!pq.empty()) //还有满足条件的区间
{
result[j] = -pq.top()[0];//区间长度l
}
}
return result;
}
};