[LeetCode] 128. Longest Consecutive Sequence 解题思路

Given an unsorted array of integers, find the length of the longest consecutive elements sequence.

For example,
Given

[100, 4, 200, 1, 3, 2]

,
The longest consecutive elements sequence is

[1, 2, 3, 4]

. Return its length:

4

.

Your algorithm should run in O(n) complexity.

问题：给定一个无序数组，找出最长的连续序列，要求时间复杂度为 O(n) 。

一开始想到的是用先排序，再找结果，但是时间复杂度要求 O(n) ，使用排序会超时。

思索未果，再网上找到一个方案，借助 unordered_set 来实现。

将元素全部塞进 unordered_set 中。

取出 unordered_set 中的一个剩余元素 x ，找到 unordered_set 中 x 的全部前后相邻元素，并将 x 和相邻元素全部移除，此时能得到 x 的相邻长度。若 unordered_set 还有元素，则继续当前步骤。

在第二步中的所有相邻长度中，找出最大值便是原问题的解。

由于 unordered_set 是基于 hash_table 来实现的，所以每次插入、查找、删除都是 O(1)，而全部元素只会 插入、查找、删除 1 次，所以整体复杂度是 O(n)。

int longestConsecutive(vector<int>& nums) {

unordered_set<int> theSet;
for (int i = 0 ; i < nums.size(); i++) {
theSet.insert(nums[i]);
}

int longest = 0;
while (theSet.size() > 0 ) {

int tmp = *theSet.begin();
theSet.erase(tmp);

int cnt = 1;

int tmpR = tmp + 1;
while (theSet.count(tmpR)) {
cnt++;
theSet.erase(tmpR);
tmpR++;
}

int tmpL = tmp - 1;
while (theSet.count(tmpL)) {
cnt++;
theSet.erase(tmpL);
tmpL--;
}

longest = max( longest, cnt);

}

return longest;
}

参考资料：

[LeetCode] Longest Consecutive Sequence, 喜刷刷