-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.cpp
58 lines (50 loc) · 1.59 KB
/
main.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
class Solution {
public:
vector<int> findRightInterval(vector<vector<int>>& intervals) {
vector<pair<int, int>> startIdx;
for (int i = 0; i < intervals.size(); ++i) {
startIdx.push_back({intervals[i][0], i});
}
sort(startIdx.begin(), startIdx.end());
vector<int> result;
for (vector<int>& interval : intervals) {
auto it = lower_bound(
startIdx.begin(), startIdx.end(), interval[1],
[](pair<int, int>& e, int t) -> bool { return e.first < t; });
if (it == startIdx.end()) {
result.push_back(-1);
} else {
result.push_back(it->second);
}
}
return result;
}
};
class Solution {
public:
vector<int> findRightInterval(vector<vector<int>>& intervals) {
vector<array<int, 2>> startIdx;
for (int i = 0; i < intervals.size(); ++i) {
startIdx.push_back({intervals[i][0], i});
}
sort(startIdx.begin(), startIdx.end());
vector<int> result;
for (vector<int>& interval : intervals) {
int l = -1, r = startIdx.size();
while (r - l > 1) {
int m = l + (r - l) / 2;
if (startIdx[m][0] < interval[1]) {
l = m;
} else {
r = m;
}
}
if (r == startIdx.size()) {
result.push_back(-1);
} else {
result.push_back(startIdx[r][1]);
}
}
return result;
}
};