巧用数学思维，轻松理解 LeetCode 228：汇总区间！

2023-12-18 18:22:25







## 前言

LeetCode 228：汇总区间是一道经典的数组操作题，考察了我们的算法思维和数据结构应用能力。题目要求我们对一个无重复元素的有序整数数组进行汇总，将相邻的元素合并成区间。例如，给定数组`[1, 2, 3, 4, 6, 7, 8, 9]`，我们需要将其汇总为`[[1, 4], [6, 9]]`。

## 算法思路

解决 LeetCode 228：汇总区间问题的关键在于巧用数学思维和算法技巧。我们可以使用一个变量`start`来记录区间的起始位置，当遇到下一个元素时，如果与`start`相邻，则将其纳入当前区间；否则，我们创建一个新的区间，将`start`和当前元素作为该区间的起始和结束位置。

具体算法步骤如下：

1. 初始化`start`变量，使其等于数组的第一个元素。
2. 遍历数组，依次处理每个元素。
3. 如果当前元素与`start`相邻，则将其纳入当前区间，并将`start`更新为当前元素。
4. 否则，创建一个新的区间，将`start`和当前元素作为该区间的起始和结束位置。
5. 重复步骤 2-4，直到遍历完整个数组。

## 代码实现

```python
def summaryRanges(nums):
    """
    :type nums: List[int]
    :rtype: List[str]
    """
    if not nums:
        return []

    result = []
    start = nums[0]
    for i in range(1, len(nums)):
        if nums[i] == nums[i - 1] + 1:
            continue
        if start == nums[i - 1]:
            result.append(str(start))
        else:
            result.append(str(start) + "->" + str(nums[i - 1]))
        start = nums[i]

    if start == nums[-1]:
        result.append(str(start))
    else:
        result.append(str(start) + "->" + str(nums[-1]))

    return result