Leetcode 最长连续序列

题目链接：https://leetcode-cn.com/problems/longest-consecutive-sequence/

题目大意：

　　略。

分析：

　　注意有重复值，序列为空等情况。

代码如下：

class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        unordered_map< int, int > m1; // 存以 key 为首的最长连续序列的长度
        unordered_map< int, int > m2; // 存以 key 为尾的最长连续序列的长度
        unordered_set< int > vis; // 检查数是否出现过
        int ans = 0;
        
        for(int i = 0; i < nums.size(); ++i) {
            if(vis.find(nums[i]) != vis.end()) continue;
            vis.insert(nums[i]);
            
            m1[nums[i]] = m2[nums[i]] = 1;
            
            if(m2.find(nums[i] - 1) != m2.end()) { // 查一下是否存在以nums[i] - 1结尾的序列
                splice(nums[i] - 1, nums[i], m2, m1); // 拼接
            }
            if(m1.find(nums[i] + 1) != m1.end()) { // 查一下是否存在以nums[i] + 1开头的序列
                splice(nums[i], nums[i] + 1, m2, m1); // 拼接
            }
        }
        
        for(auto &x : m1) ans = max(ans, x.second);
        
        return ans;
    }
    
    // 拼接以B结尾的序列和以A开头的序列
    inline void splice(int B, int A, unordered_map< int, int > &mB, unordered_map< int, int > &mA) {
        mB[A + mA[A] - 1] = mA[B - mB[B] + 1] = mB[B] + mA[A];
        mA.erase(A);
        mB.erase(B);
    }
};