[leetcode] 128. Longest Consecutive Sequence

Given an unsorted array of integers, find the length of the longest consecutive elements sequence.

For example,
Given [100, 4, 200, 1, 3, 2],
The longest consecutive elements sequence is [1, 2, 3, 4]. Return its length: 4.

Your algorithm should run in O(n) complexity.

这道题是找未排序数组中最长连续序列的长度,题目难度为Hard。


题目限定时间复杂度O(n),所以不能sort之后通过遍历求结果,考虑到Hash Table的插入、查找等操作平均时间复杂度理论上能达到O(1),所以这里用Hash Table来解决问题。遍历数组,将遍历到的数字从Hash Table中删除,然后以此数字为准向左右逐个减1加1查看相邻数字是否在Hash Table中,在Hash Table中的数字也从Hash Table中删除,直至相邻数字不在Hash Table中,这样就找到了一个连续序列,拿此序列长度和最大序列长度比较并更新最大序列长度。通过以上方法遍历完数组即可得到最长连续序列长度。具体代码:

class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        unordered_set<int> hash(nums.begin(), nums.end());
        int ret = 1;
        
        for(auto num:nums) {
            if(hash.find(num) == hash.end()) continue;
            int cnt = 1;
            int curNum = num;
            hash.erase(num);
            while(hash.find(curNum-1) != hash.end()) {
                ++cnt;
                hash.erase(--curNum);
            }
            curNum = num;
            while(hash.find(curNum+1) != hash.end()) {
                ++cnt;
                hash.erase(++curNum);
            }
            ret = max(ret, cnt);
        }
        
        return ret;
    }
};

阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/TstsUgeg/article/details/51536708
文章标签: leetcode
个人分类: leetcode
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭