[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