LeetCode-Longest Consecutive Sequence

算法分析与设计，第18周博客

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*log(n))。不符合题目的要求，所以，排序方法是不行的。时间复杂度要求是O(n)，这也就要求在常数级别的遍历中就把答案找出来。那么这样就必须保存下一些信息，也就是，对于每个n[i]，都能够记录下与之相关的最长连续序列长度，那么对于剩下的数，我们只需要查看是否有与n[i]相连接的，如果有，就把两者的长度拼起来，否则，设为1.

总结如下：对于每个数p，找出p-1和p+1的 最长连续序列长度，设分别为left和right，那么p的最长连续序列长度就是left+1+right；另外，需要更新左右两端的最长连续序列长度，把它们的值都设为left+1+right.

这里选择了使用无序字典来存储信息，在c++里，无序字典是根据hash值来进行存储和查找的，平均时间复杂度是O(1)，所以代码的总体时间复杂度也是O(n)，代码如下：

class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        int n = nums.size();
        if (n == 0)
            return n;
        unordered_map<int, int> map;
        int longest = 1;
        for (int i = 0; i < n; ++i) {
            if (map.find(nums[i]) != map.end())
                continue;
            int left = map.find(nums[i]-1) == map.end() ? 0 : map.find(nums[i]-1)->second;
            int right = map.find(nums[i]+1) == map.end() ? 0 : map.find(nums[i]+1)->second;
            int cur = left+1+right;
            map[nums[i]] = cur;
            map[nums[i]-left] = cur;
            map[nums[i]+right] = cur;
            longest = max(longest, cur);
        }
        return longest;
    }
};