leetcode 56 合并区间 C语言实现

题目

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

示例 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] 可被视为重叠区间。

代码

先给区间排序，再合并。

void mergeSort(int **intervals, int l, int r, int **arr) {
    if (l >= r) return ;
    int mid = (l + r) / 2;
    mergeSort(intervals, l, mid, arr);
    mergeSort(intervals, mid + 1, r, arr);
    for (int i = l; i <= r; i++) {
        arr[i][0] = intervals[i][0];
        arr[i][1] = intervals[i][1];
    }
    int i = l, j = mid + 1, k = l;
    while (i <= mid && j <= r) {
        if (arr[i][0] <= arr[j][0]) {
            intervals[k][0] = arr[i][0];
            intervals[k++][1] = arr[i++][1];
        } else {
            intervals[k][0] = arr[j][0];
            intervals[k++][1] = arr[j++][1];
        }
    }
    while (i <= mid) {
        intervals[k][0] = arr[i][0];
        intervals[k++][1] = arr[i++][1];
    }
    while (j <= r) {
        intervals[k][0] = arr[j][0];
        intervals[k++][1] = arr[j++][1];
    }
}
int** merge(int** intervals, int intervalsSize, int* intervalsColSize, int* returnSize, int** returnColumnSizes){
    int **arr = (int **)malloc(sizeof(int *) * intervalsSize);
    for (int i = 0; i < intervalsSize; i++) {
        arr[i] = (int *)malloc(sizeof(int) * 2);
    }
    mergeSort(intervals, 0, intervalsSize - 1, arr);
    int **res = (int **)malloc(sizeof(int *) * intervalsSize);
    (*returnColumnSizes) = (int *)malloc(sizeof(int) * intervalsSize);
    (*returnSize) = 0;
    int i = 0, j = 0;
    while (j < intervalsSize) {
        while (j < intervalsSize - 1 && intervals[j][1] >= intervals[j + 1][0]) {
            if (intervals[j][1] > intervals[j + 1][1]) intervals[j + 1][1] = intervals[j][1];
            j++;
        }
        res[(*returnSize)] = (int *)malloc(sizeof(int) * 2);
        res[*returnSize][0] = intervals[i][0]; 
        res[*returnSize][1] = intervals[j][1];
        (*returnColumnSizes)[*returnSize] = 2;
        (*returnSize)++;
        i = j + 1;
        j++;
    }
    return res;
}