数据结构 | SORT | Merge Intervals

题目

Given a collection of intervals, merge all overlapping intervals.
Example 1:
Input: [[1,3],[2,6],[8,10],[15,18]]
Output: [[1,6],[8,10],[15,18]]
Explanation: Since intervals [1,3] and [2,6] overlaps, merge them into [1,6].
Example 2:
Input: [[1,4],[4,5], [3,7]]
Output: [[1,7]]
Explanation: Intervals [1,4], [4,5] and [3, 7] are considered overlapping.
【输入】第一行，数字n，表示区间的数量，n大于1小于2000；
第二行，n对区间；
【输出】 合并后的区间。
例如：
【输入】
4
1 3 2 6 8 10 15 18
【输出】
1 6 8 10 15 18

代码

#include <bits/stdc++.h>
using namespace std;

class MergeIntervals
{
public:
    vector<vector<int>> merge(vector<vector<int>> &intervals);
};

int main()
{
    int item, n;
    vector<vector<int> > intervals, ans;
    vector<int> tmp;
    cin >> n;
    for (int i = 0; i < n; i++) 
    {
        for (int j = 0; j < 2; j++)
        {
            cin >> item;
            tmp.push_back(item);
        }
        intervals.push_back(tmp);//注意这里要放进去vector<int>类型，不是int
        tmp.clear();//记得将tmp清空
    }
    MergeIntervals sol;
    ans = sol.merge(intervals);
    for (int i = 0; i < ans.size(); i++)
    {
        for (int j = 0; j < ans[i].size(); j++)
        {
            cout << ans[i][j] << " ";
        }
    }
    system("pause");
    return 0;
}

vector<vector<int> > MergeIntervals::merge(vector<vector<int> > &intervals)
{
    for (int i = 0; i < intervals.size() - 1; i++)
    {
        int n = intervals[i].size() - 1;//数据前移
        if (intervals[i][n] >= intervals[i + 1][0])//相邻的区间，前边的右侧小于等于后边的左侧
        {
            if (intervals[i][n] < intervals[i + 1][1])
            {
                intervals[i].erase(intervals[i].begin() + n);
                intervals[i + 1].erase(intervals[i + 1].begin());//删除掉两个边
            }
            else
            {
                intervals.erase(intervals.begin() + i + 1);
                i--;//这里整行删除后，数据会前移，i需要-1
            }
        }
    }
    return intervals;
}

一些问题

1. vector的数据erase后，会整体前移，如果是二维，就是在行中前移或整行前移。这里会有越界风险。
2. vector<vector<int> > 放入数据时，要放入的是vector类型，不是按下标放入int，输出的时候可以访问下标。