力扣: 合并区间 (scala)

1. 题目描述

给出一个区间的集合，请合并所有重叠的区间。

示例 1:

输入: [[1,3],[2,6],[8,10],[15,18]]
输出: [[1,6],[8,10],[15,18]]
解释: 区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6].
示例 2:

输入: [[1,4],[4,5]]
输出: [[1,5]]
解释: 区间 [1,4] 和 [4,5] 可被视为重叠区间。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/merge-intervals
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 思路
   ① 首先将输入的多个区间按照区间头大小升序排列,保证可以合并的区间都连续
   ② 如果 后一个区间的头 在 前一个区间内 (即: v2.head<= v1.last),则说明两个区间可以合并,反之,不可以合并
   ③ 用一个变量保存合并后的区间值,当变量和后一个区间不可合并时,将变量的值加入结果列表,然后将后一个区间赋值给变量
   ④ 初始化变量为排序后的区间集合第一个元素

3. 代码实现+注释

  def merge(intervals: Array[Array[Int]]): Array[Array[Int]] = {
    import scala.collection.mutable.ArrayBuffer
    if (intervals.length <= 1) return intervals
    //结果
    val res = new ArrayBuffer[Array[Int]]
    //排序
    val array = intervals.sortBy(_.head)
    //初始化中间变量
    var ints = array.head

    //遍历区间
    for (element <- array) {
      //element 相当于后一个区间,ints相当于前一个区间,这里判断element的区间头是否在ints区间内
      // 如果 element.head > ints.last , 那么说明后一个区间和前一个区间没有交集,那么就可以将合并的区间值加入结果
      if (element.head > ints.last) {
        //加入结果
        res.append(Array(ints.head, ints.last))
        //之后,我们开始从element开始,合并后面的区间
        ints = element
      } else {
        //如果element.head <= ints.last,则说明element和ints两个区间有交集,因为我们是排过序的,所以ints区间头
        //是小于等于element头的,合并后的区间尾则要取element和ints最大的尾数
        ints = Array(ints.head, Math.max(ints.last, element.last))
      }
    }
    //上面循环完毕后,少了一个区间
    //这里分两个情况说明: 即最后一个区间是否和前面的区间合并
    //例如: 未合并的情况 [2,6],[6,10],[15,18]
    // 1. 在循环内 [2,6],[6,10] ==> [2,10] 这个区间的合并是这句代码: (ints = Array(ints.head, Math.max(ints.last, element.last)))
    // 2. 下次循环, 此时ints是[2,10], element是[15,18],开始判断,没有交集,所以将ints加入结果,然后ints被赋值[15,18],之后退出循环
    // 3. 此时最后一个区间是没有加入结果的
    // 例如:合并的情况 [2,6],[6,10],[10,18]
    // 1. 在循环内 [2,6],[6,10] ==> [2,10] 这个区间的合并是这句代码: (ints = Array(ints.head, Math.max(ints.last, element.last)))
    // 2. 下次循环, 此时ints是[2,10], element是[10,18],开始判断,有交集,所以要合并,(ints = Array(ints.head, Math.max(ints.last, element.last)))
    // 3. 此时: ints被赋值[2,18],然后退出了循环, 合并后的区间是没有加入结果的
    //所以,在循环外将最后一个区间加上
    res.append(ints)
    res.toArray
  }

4. 初学算法,有不严谨错误的地方,感谢指出