首页 > 代码库 > Longest Consecutive Sequence

Longest Consecutive Sequence

其实网上关于leetcode的题目的答案已经相当多了,这也是为啥我自己不把每道题目都贴出来的原因,觉得分析得没人好,代码也没别人的精简。不过,这道题目看到网上有不少做法跟实际要求的O(n)复杂度不太符合,所以特别粘贴出来,也正好记录下自己的一些想法。

 

说实话,在第二遍做这道题目的时候,我还是没能够把最为完善的解法想出来,能够想到的是利用hash表以序列值为key,1(只是简单做个标记)为val,把序列保存起来,利用hash表O(1)的插入,查询复杂度来满足题目复杂度要求。这道题目的解法其实不难,就是遍历num中的每个元素,然后分别以num[i] - 1和num[i] + 1的顺序扫描,看是否能构成连续的序列。这道题目有两个地方需要注意,看到别人使用set,map来做,虽然原理是对的,但是由于从容器实现角度来看,它们都是基于红黑树来实现的,其插入,查找复杂度为O(log n),故其实并不满足题目要求。此外,还有一点也很重要,就是每次扫描到连续序列的时候,必须把对应的元素从hash表中删除,这样才能够保证每个元素只被扫描一次,否则复杂度也并非O(n)(假如不删除表中元素提交的话,会time limit exceed)。注意到以上两点后,相信写代码实现就不难了。此外,我自己还犯过一个错,就是遍历元素的时候,一开始直接对hash表进行遍历,结果删除元素的时候会对迭代器产生影响而导致程序崩溃,后来改成对num的元素进行遍历就可以了,其实本来就该这么做,一时间没想通。。。好了,能AC的代码如下,可能写的比较丑陋,望轻拍~

class Solution {
public:
    int longestConsecutive(vector<int> &num) {
        if (num.size() <= 1)
            return num.size();
            
        unordered_map<int, int> table;
        for (int i = 0; i < num.size(); i++)
            table.insert(make_pair(num[i], 1));
            
        int maxLen = 0;
        int len;

        for (int i = 0; i < num.size(); i++)
        {
            len = 1;
            if (table[num[i]] == 1)
            {
                for (int j = num[i] - 1; ; j--)
                {
                    if (table.find(j) != table.end())
                    {
                        len++;
                        table.erase(j);
                    }
                    else
                        break;
                }
                for (int j = num[i] + 1; ; j++)
                {
                    if (table.find(j) != table.end())
                    {
                        len++;
                        table.erase(j);
                    }
                    else
                        break;
                }
                if (len > maxLen)
                    maxLen = len;
                
            }
        }
        
        return maxLen;
    }
};