diff --git a/cpp/2205/220520-CN.cpp b/cpp/2205/220520-CN.cpp new file mode 100644 index 0000000..f48a2cd --- /dev/null +++ b/cpp/2205/220520-CN.cpp @@ -0,0 +1,40 @@ +#include +#include +#include +#include + +/** + * 436. Find Right Interval + * You are given an array of intervals, where intervals[i] = [starti, endi] and each starti is unique. + * The right interval for an interval i is an interval j such that startj >= endi and startj is minimized. Note that i may equal j. + * Return an array of right interval indices for each interval i. If no right interval exists for interval i, then put -1 at index i. + */ + +class Solution { +public: + static std::vector findRightInterval(const std::vector>& intervals) { + std::vector> s; + for (int i = 0; i < intervals.size(); ++i) { + s.emplace_back(intervals[i][0], intervals[i][1], i); + } + std::sort(s.begin(), s.end()); + + std::vector ret(intervals.size()); + for (const auto& [f, t, id] : s) { + auto it = std::lower_bound(s.begin(), s.end(), std::tuple{t, -0x7FFFFFFF, -0x7FFFFFFF}); + if (it == s.end()) + ret[id] = -1; + else + ret[id] = std::get<2>(*it); + } + return ret; + } +}; + +int main() { + auto r = Solution::findRightInterval({{-100,-98},{-99,-97},{-98,-96},{-97,-95}}); + for (int i : r) { + std::printf("%d ", i); + } + return 0; +} diff --git a/cpp/2205/CMakeLists.txt b/cpp/2205/CMakeLists.txt index 365f717..440f46b 100644 --- a/cpp/2205/CMakeLists.txt +++ b/cpp/2205/CMakeLists.txt @@ -3,4 +3,4 @@ PROJECT(2205) SET(CMAKE_CXX_STANDARD 23) -ADD_EXECUTABLE(2205 220519.cpp) +ADD_EXECUTABLE(2205 220520-CN.cpp)