|
| 1 | +package leetcode |
| 2 | + |
| 3 | +import ( |
| 4 | + "github.com/halfrost/LeetCode-Go/structures" |
| 5 | +) |
| 6 | + |
| 7 | +// Interval define |
| 8 | +type Interval = structures.Interval |
| 9 | + |
| 10 | +// SummaryRanges define |
| 11 | +type SummaryRanges struct { |
| 12 | + intervals []Interval |
| 13 | +} |
| 14 | + |
| 15 | +// Constructor352 define |
| 16 | +func Constructor352() SummaryRanges { |
| 17 | + return SummaryRanges{} |
| 18 | +} |
| 19 | + |
| 20 | +// AddNum define |
| 21 | +func (sr *SummaryRanges) AddNum(val int) { |
| 22 | + if sr.intervals == nil { |
| 23 | + sr.intervals = []Interval{ |
| 24 | + Interval{ |
| 25 | + Start: val, |
| 26 | + End: val, |
| 27 | + }, |
| 28 | + } |
| 29 | + return |
| 30 | + } |
| 31 | + |
| 32 | + low, high := 0, len(sr.intervals)-1 |
| 33 | + for low <= high { |
| 34 | + mid := low + (high-low)>>1 |
| 35 | + if sr.intervals[mid].Start <= val && val <= sr.intervals[mid].End { |
| 36 | + return |
| 37 | + } else if val < sr.intervals[mid].Start { |
| 38 | + high = mid - 1 |
| 39 | + } else { |
| 40 | + low = mid + 1 |
| 41 | + } |
| 42 | + } |
| 43 | + |
| 44 | + if low == 0 { |
| 45 | + if sr.intervals[0].Start-1 == val { |
| 46 | + sr.intervals[0].Start-- |
| 47 | + return |
| 48 | + } |
| 49 | + ni := Interval{Start: val, End: val} |
| 50 | + sr.intervals = append(sr.intervals, ni) |
| 51 | + copy(sr.intervals[1:], sr.intervals) |
| 52 | + sr.intervals[0] = ni |
| 53 | + return |
| 54 | + } |
| 55 | + |
| 56 | + if low == len(sr.intervals) { |
| 57 | + if sr.intervals[low-1].End+1 == val { |
| 58 | + sr.intervals[low-1].End++ |
| 59 | + return |
| 60 | + } |
| 61 | + sr.intervals = append(sr.intervals, Interval{Start: val, End: val}) |
| 62 | + return |
| 63 | + } |
| 64 | + |
| 65 | + if sr.intervals[low-1].End+1 < val && val < sr.intervals[low].Start-1 { |
| 66 | + sr.intervals = append(sr.intervals, Interval{}) |
| 67 | + copy(sr.intervals[low+1:], sr.intervals[low:]) |
| 68 | + sr.intervals[low] = Interval{Start: val, End: val} |
| 69 | + return |
| 70 | + } |
| 71 | + |
| 72 | + if sr.intervals[low-1].End == val-1 && val+1 == sr.intervals[low].Start { |
| 73 | + sr.intervals[low-1].End = sr.intervals[low].End |
| 74 | + n := len(sr.intervals) |
| 75 | + copy(sr.intervals[low:], sr.intervals[low+1:]) |
| 76 | + sr.intervals = sr.intervals[:n-1] |
| 77 | + return |
| 78 | + } |
| 79 | + |
| 80 | + if sr.intervals[low-1].End == val-1 { |
| 81 | + sr.intervals[low-1].End++ |
| 82 | + } else { |
| 83 | + sr.intervals[low].Start-- |
| 84 | + } |
| 85 | +} |
| 86 | + |
| 87 | +// GetIntervals define |
| 88 | +func (sr *SummaryRanges) GetIntervals() [][]int { |
| 89 | + intervals := [][]int{} |
| 90 | + for _, interval := range sr.intervals { |
| 91 | + intervals = append(intervals, []int{interval.Start, interval.End}) |
| 92 | + } |
| 93 | + return intervals |
| 94 | +} |
| 95 | + |
| 96 | +/** |
| 97 | + * Your SummaryRanges object will be instantiated and called as such: |
| 98 | + * obj := Constructor(); |
| 99 | + * obj.AddNum(val); |
| 100 | + * param_2 := obj.GetIntervals(); |
| 101 | + */ |
0 commit comments