侧边栏壁纸
博主头像
GabrielxD

列車は必ず次の駅へ。では舞台は?私たちは?

  • 累计撰写 471 篇文章
  • 累计创建 108 个标签
  • 累计收到 9 条评论

目 录CONTENT

文章目录

【有序集合】我的日程安排表 I

GabrielxD
2022-06-15 / 0 评论 / 0 点赞 / 35 阅读 / 731 字 / 正在检测是否收录...

题目

729. 我的日程安排表 I


实现一个 MyCalendar 类来存放你的日程安排。如果要添加的日程安排不会造成 重复预订 ,则可以存储这个新的日程安排。

当两个日程安排有一些时间上的交叉时(例如两个日程安排都在同一时间内),就会产生 重复预订 。

日程可以用一对整数 startend 表示,这里的时间是半开区间,即 [start, end), 实数 x 的范围为, start <= x < end

实现 MyCalendar 类:

  • MyCalendar() 初始化日历对象。
  • boolean book(int start, int end) 如果可以将日程安排成功添加到日历中而不会导致重复预订,返回 true 。否则,返回 false 并且不要将该日程安排添加到日历中。

示例:

输入:
["MyCalendar", "book", "book", "book"]
[[], [10, 20], [15, 25], [20, 30]]
输出:
[null, true, false, true]

解释:
MyCalendar myCalendar = new MyCalendar();
myCalendar.book(10, 20); // return True
myCalendar.book(15, 25); // return False ,这个日程安排不能添加到日历中,因为时间 15 已经被另一个日程安排预订了。
myCalendar.book(20, 30); // return True ,这个日程安排可以添加到日历中,因为第一个日程安排预订的每个时间都小于 20 ,且不包含时间 20 。

提示:

  • 0 <= start < end <= 10^9
  • 每个测试用例,调用 book 方法的次数最多不超过 1000 次。

解题

方法一:有序集合

思路

维护一个树集,存储长度为2的数组表示一个日程区间,按照区间的起始升序排序。

每次添加新日程安排时:

  • 利用树集的性质先找到在集合中应该排在当前日程(curr)后的第一个元素,如果该日程不为空且该日程起始比要添加日程的结束要大,则说明产生了重复预订,直接返回 false
  • 再找到应该排在当前日程(curr)前的第一个元素,如果该日程不为空且该日程结束比要添加日程的起始要大,则说明产生了重复预订,直接返回 false
  • 最后返回把 curr 加入集合(如果重复会返回 false)。

代码

class MyCalendar {
    private final TreeSet<Integer[]> set;

    public MyCalendar() {
        set = new TreeSet<>((a, b) -> a[0] - b[0]);
    }
    
    public boolean book(int start, int end) {
        Integer[] curr = {start, end};
        Integer[] ceiling = set.ceiling(curr);
        if (ceiling != null && ceiling[0] < end) return false;
        Integer[] floor = set.floor(curr);
        if (floor != null && floor[1] > start) return false;
        return set.add(curr);
    }
}

优化

class MyCalendar {
    private TreeSet<int[]> set;

    public MyCalendar() {
        set = new TreeSet<>((a, b) -> {
            if (a[1] <= b[0]) return -1;
            if (a[0] >= b[1]) return 1;
            return 0;
        });
    }
    
    public boolean book(int start, int end) {
        return set.add(new int[]{start, end});
    }
}
0

评论区