https://leetcode.com/problems/data-stream-as-disjoint-intervals/
/** * Definition for an interval. * struct Interval { * int start; * int end; * Interval() : start(0), end(0) {} * Interval(int s, int e) : start(s), end(e) {} * }; */ class SummaryRanges { vector<Interval> vec; int find(int val) { int vlen = vec.size(); if (vlen == 0) { return -1; } int begin = 0; int end = vlen - 1; int mid; while (begin <= end) { mid = begin + (end - begin) / 2; // Here <= vs. > if (vec[mid].start > val) { end = mid - 1; } else { begin = mid + 1; } } // Here it‘s important to return end return end; } public: /** Initialize your data structure here. */ SummaryRanges() { } void addNum(int val) { int idx = find(val); if (idx < 0) { if (vec.size() > 0 && vec[0].start == val+1) { vec[0].start = val; } else { Interval inter(val, val); vec.insert(vec.begin(), inter); } } else { if (vec[idx].end == val-1) { vec[idx].end = val; } else if (vec[idx].end < val-1) { Interval inter(val, val); vec.insert(vec.begin()+idx+1, inter); idx++; } // check latter if (vec.size() > idx+1 && vec[idx+1].start == vec[idx].end + 1) { vec[idx].end = vec[idx+1].end; vec.erase(vec.begin()+idx+1); } } } vector<Interval> getIntervals() { return vec; } }; /** * Your SummaryRanges object will be instantiated and called as such: * SummaryRanges obj = new SummaryRanges(); * obj.addNum(val); * vector<Interval> param_2 = obj.getIntervals(); */
时间: 2024-10-05 20:26:17