leedcode 128. 最长连续序列(哈希)

题目描述

难度:中等

给定一个未排序的整数数组 nums ,找出数字连续的最长序列(不要求序列元素在原数组中连续)的长度。

请你设计并实现时间复杂度为 O(n) 的算法解决此问题

示例

示例 1:

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

输入:nums = [0,3,7,2,5,8,4,6,0,1]
输出:9
 

提示:

0 <= nums.length <= 105
-109 <= nums[i] <= 109

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/longest-consecutive-sequence
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解

  • 哈希 map
  • 看到需要 o(n) 的复杂度,以为只是遍历一遍,没有思路。想了五六分钟,想到就算遍历两遍,也是 o(n) 的复杂度,思路代码都想出来了。
  • 比如  nums = [100,4,200,1,3,2]  第一遍遍历,mp[100]=1,mp[4]=1,mp[200]=1,mp[1]=1,mp[3]=1,mp[2]=1。第二次遍历,mp[100]=1,看 100 以前和以后还有没有连续的数,mp[100]=1,把 mp[100]标记为2。mp[101]==0 不存在,mp[99]==0 不存在 ,所以此时最大连续为 1。遍历到 4 时,看 4以前和以后还有没有连续的数,mp[5]==0 不存在,再看 4 以前的数,mp[3]=1,mp[2]=1,mp[1]=1,并把它们标记为 2,mp[1]=2,mp[2]=2,mp[3]=2,mp[4]=2,所以此时最大连续为 4。200 同理。遍历到 1 时,由于 mp[1]=2,说明已经遍历过第二遍了,不用再遍历了。后面的也同理。所以,每个数只被遍历 2 此,时间复杂度为 o(2n),即为  o(n)。
class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        unordered_map<int,int>mp;
        int ans=0;
        int n = nums.size();
        for(int i=0;i<n;i++){//nums中有的数都标记为 1
            mp[nums[i]]=1;
        }
        for(int i=0;i<n;i++){//再遍历一遍nums
            if(mp[nums[i]]==1){
                int count=0;        
                for(int j=nums[i];j<1000000005;j++){//往右最大到多少
                    if(mp[j]==1){//如果第二遍没遍历过且存在
                        mp[j]=2;//标记第二遍已经遍历过
                        count++;
                    }else{
                        break;
                    }
                }
                for(int j=nums[i]-1;j>-1000000005;j--){//往左最小到多少
                    if(mp[j]==1){//如果第二遍没遍历过且存在
                        mp[j]=2;//标记第二遍已经遍历过
                        count++;
                    }else{
                        break;
                    }           
                }
            ans=max(ans,count);
            }
        } 
        return ans;      
    }
};

  • 官方也是哈希,用的是 set,代码更简洁 
  • 第一次遍历用 set 标记,再去重,第二次遍历,每次只检查这个数是否是连续序列的第一个数,也就是这个数 x 在 set 中且它的前驱数 x-1不在 set 中。如果这个数是连续序列的第一个数,那么向后遍历有多少个数在数组中,即是否有一个 x, x+1, x+2,……, x+y 的连续序列,其长度为 y+1。
class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        unordered_set<int> num_set;
        for (const int& num : nums) {
            num_set.insert(num);//放入 set 中
        }

        int longestStreak = 0;

        for (const int& num : num_set) {
            if (!num_set.count(num - 1)) {//检查这个数是否是连续序列的第一个数
                int currentNum = num;
                int currentStreak = 1;//此时长度

                while (num_set.count(currentNum + 1)) {//向后遍历有多少个连续数
                    currentNum += 1;
                    currentStreak += 1;//长度+1
                }

                longestStreak = max(longestStreak, currentStreak);
            }
        }

        return longestStreak;           
    }
};
原文地址:https://www.cnblogs.com/caiyishuai/p/15314342.html