128. 最长连续序列

题目

给定一个未排序的整数数组，找出最长连续序列的长度。

要求算法的时间复杂度为 O(n)。

示例

输入: [100, 4, 200, 1, 3, 2]
输出: 4
解释: 最长连续序列是 [1, 2, 3, 4]。它的长度为 4。

思路

• 使用哈希表记录数字所在区间的长度，在遍历数组的时候，如果当前数字n不在哈希表中，则查询n-1和n+1是否在哈希表中。则n处区间的长度为1 + ( n-1处区间长度) + (n+1处区间长度)，前提是n-1和n+1是存在在哈希表中的，然后通过n-1和n+1的长度就可以计算出区间两头的数字，将它们的区间长度更新。
• 因为每次都判断当前数字是否在哈希表中，如果遇到哈希表中的元素，就跳过，这样使得每次更新都只在意相连区间的两头的值，而不用管区间内部元素的值。

代码

class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        unordered_map<int, int> hash;
        int ans = 0;

        for ( auto& n : nums ) {
            if ( hash.count( n ) == 0 ) {
                int left = hash.count( n - 1 ) ? hash[n-1] : 0;
                int right = hash.count( n + 1 ) ? hash[n+1] : 0;

                hash[n] = 1 + left + right;
                
                hash[n-left] = hash[n];
                hash[n+right] = hash[n];

                ans = max( ans, hash[n] );
            }
        }

        return ans;
    }   
};