forked from javadev/LeetCode-in-Java
-
Notifications
You must be signed in to change notification settings - Fork 0
/
MyCalendar.java
49 lines (39 loc) · 1.39 KB
/
MyCalendar.java
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
package g0701_0800.s0729_my_calendar_i;
// #Medium #Binary_Search #Design #Ordered_Set #Segment_Tree #Programming_Skills_II_Day_20
// #2022_03_25_Time_17_ms_(97.23%)_Space_43_MB_(87.36%)
import java.util.TreeSet;
@SuppressWarnings("java:S1210")
public class MyCalendar {
static class Meeting implements Comparable<Meeting> {
public final int start;
public final int end;
public Meeting(int start, int end) {
this.start = start;
this.end = end;
}
@Override
public int compareTo(Meeting anotherMeeting) {
return this.start - anotherMeeting.start;
}
}
private final TreeSet<Meeting> meetings;
public MyCalendar() {
meetings = new TreeSet<>();
}
public boolean book(int start, int end) {
Meeting meetingToBook = new Meeting(start, end);
Meeting prevMeeting = meetings.floor(meetingToBook);
Meeting nextMeeting = meetings.ceiling(meetingToBook);
if ((prevMeeting == null || prevMeeting.end <= meetingToBook.start)
&& (nextMeeting == null || meetingToBook.end <= nextMeeting.start)) {
meetings.add(meetingToBook);
return true;
}
return false;
}
}
/*
* Your MyCalendar object will be instantiated and called as such:
* MyCalendar obj = new MyCalendar();
* boolean param_1 = obj.book(start,end);
*/