leetcode 228. 汇总区间(Summary Ranges)

目录

• 题目描述：
• 示例 1:
• 示例 2:
• 解法：

题目描述：

给定一个无重复元素的有序整数数组，返回数组区间范围的汇总。

示例 1:

输入: [0,1,2,4,5,7]
输出: ["0->2","4->5","7"]
解释: 0,1,2 可组成一个连续的区间; 4,5 可组成一个连续的区间。

示例 2:

输入: [0,2,3,4,6,8,9]
输出: ["0","2->4","6","8->9"]
解释: 2,3,4 可组成一个连续的区间; 8,9 可组成一个连续的区间。

---

解法：

class Solution {
public:
    string itoa(long long num){
        bool neg = false;
        if(num < 0){
            neg = true;
            num = -num;
        }else if(num == 0){
            return "0";
        }
        string res = "";
        while(num != 0){
            res = char('0' + num%10) + res;
            num /= 10;
        }
        if(neg){
            return "-" + res;
        }else{
            return res;
        }
    }
    
    vector<string> summaryRanges(vector<int>& nums) {
        unordered_map<long long, long long> pre, pst;
        for(long long num : nums){
            if(pre.find(num-1) == pre.end()){
                pre[num] = num;
            }else{
                pre[num] = pre[num-1];
                pre.erase(num-1);
            }
            if(pst.find(num+1) == pst.end()){
                pst[num] = num;
            }else{
                pst[num] = pst[num+1];
                pst.erase(num+1);
            }
        }
        vector<string> res;
        vector<long long> lst;
        for(auto it : pre){
            lst.push_back(it.first);
        }
        sort(lst.begin(), lst.end());
        for(int num : lst){
            int a = pre[num], b = pst[num];
            if(a != b){
                res.push_back(itoa(a) + "->" + itoa(b));
            }else{
                res.push_back(itoa(a));
            }
        }
        return res;
    }
};