[LeetCode] 128. 最长连续序列(数组)

128. 最长连续序列

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

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

示例:

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

解题思路： 此题的标签是并查集+数组，一开始解题的时候，想着如果用并查集的话，那么可以用hashmap存在并查集的关系，然后键存访问元素，值存访问元素所在的连续序列的第一个元素，在Union的时候会发现，需要对n个元素，两两进行Union，时间复杂度为$O(n^2)$，OJ TLE了。然后换了中非并查集的做法，发现标签为数组的题，确实有一定的技巧性。思路为先将所有的值存到hashmap中，然后访问数组，若元素在hashmap中存在，从该元素开始向两端搜索，直到元素不存在为止，假设向前搜索终止的值为pre，向后搜索终止的值为next，（注意搜索的过程中为减少后续的重复搜索，要将hashmap中搜索到的值erase掉），那么这段序列的长度为next-pre-1.不过算法的时间复杂度也不是正好O(n)，应该说是非常接近O(n)。

class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        unordered_set<int> st(nums.begin(), nums.end());
        int res = 0;
        for (auto num : nums) {
            if (st.count(num)) {
                st.erase(num);
                int pre = num - 1, next = num + 1;
                while (st.count(pre)) {
                    st.erase(pre);
                    --pre;
                }
                while (st.count(next)) {
                    st.erase(next);
                    ++next;
                }
                res = max(res, next - pre - 1);
            }
        }
        return res;
    }
};

—————————————

参考资料

https://www.cnblogs.com/grandyang/p/4276225.html