Leetcode 128. Longest Consecutive Sequence 最长连续子序列 DFS，并查集

最新推荐文章于 2021-08-17 17:01:18 发布

Xurui_Luo

最新推荐文章于 2021-08-17 17:01:18 发布

阅读量501

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/Xurui_Luo/article/details/117936787

版权

算法专栏收录该内容

137 篇文章

订阅专栏

这篇博客介绍了如何利用深度优先搜索（DFS）和并查集（Disjoint Set）两种数据结构来解决寻找给定整数数组中最长连续子集的问题。在DFS实现中，通过记忆化搜索优化了搜索效率，而在并查集中，通过查找和合并操作高效地确定了最长连续序列。两种方法都以O(n)的时间复杂度完成任务。

摘要由CSDN通过智能技术生成

DFS

每访问一个元素i，都直接去看i +1, i + 2… i + n直到这些元素不存在，通过记忆化搜索的方式，保证只会搜n次

class Solution {
public:
    unordered_map<int, int> dic;
    int dfs(int x){
        if(!dic.count(x)) return 0;
        if(dic[x] != 0) return dic[x];
        return dic[x] = dfs(x + 1) + 1;
    }
    int longestConsecutive(vector<int>& nums) {
        if(nums.size() == 0 ) return 0;
        int res = 1;
        for(auto e: nums){
            dic[e] = 0;
        }
        for(auto e: nums){
            res = max(res, dfs(e));
        }
        return res;
    }
};

并查集

class Solution {
public:
    unordered_map<int, int> par, count;
    int find(int v){
        return par[v] == v?v:par[v] = find(par[v]);
    }
    int merge(int u, int v){
        u = find(u);
        v = find(v);
        if(u != v){
            par[v] = par[u];
            count[u] += count[v];
        }
        return count[u];
    }
    int longestConsecutive(vector<int>& nums) {
        if(nums.size() == 0 ) return 0;
        int res = 1;
        for(auto e: nums){
            par[e] = e;
            count[e] = 1;
        }
        for(auto e: nums){
            if (count.count(e + 1) ) 
                res = max(res, merge(e + 1, e));
        }
        return res;
    }
};