-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSolution128.cpp
More file actions
34 lines (32 loc) · 1.13 KB
/
Solution128.cpp
File metadata and controls
34 lines (32 loc) · 1.13 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
//
// Solution128.cpp
// Algorithm
//
// Created by Pancf on 2020/12/20.
// Copyright © 2020 Pancf. All rights reserved.
//
#include "Solution128.hpp"
#include <unordered_map>
#include <algorithm>
int Solution128::longestConsecutive(std::vector<int> &nums)
{
if (!nums.size()) return 0;
int size = (int)nums.size();
std::unordered_map<int, int> mp; // key is nums[i], value is the length of sequence to which key belongs
int res = 0;
for (int i = 0; i < size; ++i) {
auto key = nums[i];
if (mp.find(key) == mp.end()) {
int left = mp.find(key - 1) != mp.end() ? mp[key - 1] : 0; // left neighbor has `left` length
int right = mp.find(key + 1) != mp.end() ? mp[key + 1] : 0; // right neighbor has `right` length
int sum = left + right + 1;
res = std::max(sum, res);
mp[key] = sum;
// Why only update mp[key - left] and mp[key + right]?
// Because only mp[key - left] and mp[key + right] has the possibility to calculate new sum
mp[key - left] = sum;
mp[key + right] = sum;
}
}
return res;
}