1. 题目来源
2. 题目解析
被题目所蒙蔽了…
以为是最短路算法,或者是堆优化 dij 又有什么神奇的性质被我遗忘了?或者是最短路树?…
结果没读好题目,题目说明 query 的所有查询是不会有 相交但不包含 的关系的。也就是说,所有的查询,要么不想交,要么将包含一些小区间在里面。
那么思路就比较明确了,当遇见一个大的 query 区间,只需要将这个大 query 区间中的所有小区间全部弹出即可,因为步长是 1,所以有效的区间个数即为最短路长度。
- 时间复杂度: O ( ( n + q ) l o g n ) O((n+q)logn) O((n+q)logn)
- 空间复杂度: O ( n ) O(n) O(n)
class Solution {
public:
vector<int> shortestDistanceAfterQueries(int n, vector<vector<int>>& queries) {
set<pair<int, int>> S;
for (int i = 1; i < n; i ++ ) S.insert({i - 1, i});
vector<int> res;
for (auto q : queries) {
int l = q[0], r = q[1];
auto it = S.lower_bound({l, -1}); // 直接找到 小于 [l, ] 左端点的区间
// 如果找到了,且找到区间和查询区间同起点,且包含里面的小区间时
// it->first == l 这个条件是一个优化条件,如果同起点的话,可能该区间还没被删除过
// 但是不同起点的话,且还在我右边界内的点,肯定是被删除了的,就不用再进去判断了
// 但是不能将 it->second < r 这个条件删除。不然就会进入 if 判断,S 将 insert 一个区间包含 的 子区间,答案增多一个
if (it != S.end() && it->first == l && it->second < r) {
// 将这些小区间全部剔除
while (it != S.end() && it->first < r) it = S.erase(it);
// 将查询区间插入
S.insert({l, r});
}
res.push_back(S.size());
}
return res;
}
};
class Solution {
public:
vector<int> shortestDistanceAfterQueries(int n, vector<vector<int>>& queries) {
vector<int> d(n), ans(queries.size());
iota(d.begin(), d.end(), 0);
for (int i = 0; i < queries.size(); i++) {
auto l = upper_bound(d.begin(), d.end(), queries[i][0]);
auto r = lower_bound(d.begin(), d.end(), queries[i][1]);
if (l <= r) d.erase(l, r);
ans[i] = d.size() - 1;
}
return ans;
}
};