力扣 128. 最长连续序列

题目链接

128. 最长连续序列

题目描述

思路

    字典存储当前序列的头和长度，新的值来到后检查字典中是否存在值加一
    再存一个尾和长度，新来值查看有没有存在值
    更新头尾，以及判断有没有能连续起来的

代码

class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        unordered_map<int,int> head,tail;
        for(auto &num:nums){
            if(head[num])continue;
            if(tail[num])continue;
            if(head[num+1]!=0){
                head[num]=head[num+1]+1;
                tail[head[num]+num-1]+=1;
            }
            if(tail[num-1]!=0){
                tail[num]=tail[num-1]+1;
                head[num-tail[num]+1]+=1;
            }
            if(tail[num]&&head[num]){
                head[num-tail[num]+1]+=(head[num]-1);
                tail[head[num]+num-1]=head[num-tail[num]+1];
            }
            if(!head[num])head[num]=1;
            if(!tail[num])tail[num]=1;
            // cout<<"head"<<endl;
            // for(auto &x:head){
            //     cout<<x.first<<" "<<x.second<<endl;
            // }
            // cout<<"tail"<<endl;
            // for(auto &x:tail){
            //     cout<<x.first<<" "<<x.second<<endl;
            // }
        }
        int max=0;
        for(auto &x:head){
            if(x.second>max)max=x.second;
        }
        return max;
    }
};

结果

优化

看的大佬的代码，很妙，直接遍历哈希表就可以判断的

class Solution {
public:
//参考题解：最长连续序列 | 哈希 | 代码简洁易懂 【c++/java版】
    int longestConsecutive(vector<int>& nums) {
        int res = 0;
        unordered_set<int> uset;
        //将数组中的值加入到哈希表中，便于查找，顺带降重，提高查找效率
        for(auto num:nums)
            uset.insert(num);
        //遍历哈希表
        for(auto x:uset){
            //从连续序列的第一个数开始查找
            if(!uset.count(x - 1)){ //如果x前一位数不存在，则说明x是连续序列的第一个数
                int y = x;
                //这里直接在哈希表中查找y+1，当y+1不存在时，说明正好查找到了y
                while(uset.count(y+1)){
                    y++;
                }
                res = max(res,y-x+1);
            }
        }
        return res;
    }
};