leetcode 228. 汇总区间(Summary Ranges)
题目描述:
给定一个无重复元素的有序整数数组,返回数组区间范围的汇总。
示例 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;
}
};