leetcode 最長連續序列 longest consecutive sequence

轉載請註明來自souldak,微博:@evagle
c++

題目(來自leetcode):算法

給你一個n個數的亂序序列,O(N)找出其中最長的連續序列的長度。數組

例如給你[100, 4, 200, 1, 3, 2],那麼最長的連續序列爲[1, 2, 3, 4],因此返回4。.net

思路:code

最簡單直接的想法是:將數組排序,而後掃一遍排好序的序列,從中找出最長的便可,這樣的話時間是O(nlogn)+O(n),顯然不符合題目要求,會超時。blog

那怎麼樣不排序就能作出來呢?排序

咱們先拋開算法,想一想本身會怎麼找,當數組很小,你能夠記住全部的數。例如上面的例子。咱們本身找的話,看到一個數例如4,咱們會下意識的去找有沒有3和5,有3那麼就會繼續找2,以此類推,直到找不到爲止,找過的數之後我就不會再去找了,例如1,2,3,4都在找4的時候一塊兒找過了,因此我後面只考慮100,和200,這樣每一個數其實我只找了一遍。 若是我可以在O(1)的時間判斷是否存在一個數,那麼我找最長的序列的時間就是O(N)的。如今好辦了,如何實如今O(1)的時間查找一個數?Answer:hash table。對,咱們只要將數存儲在hashmap或者hashset裏面就OK了。這裏只要用hashset就夠了,咱們並不須要鍵值對的映射。好了,僞代碼:element

1. 預處理:建hashset,將數組元素所有加進hashset裏頭leetcode

2. For each element e in array:rem

   2.1 if e in hashset:

           find e-1, e+1 in hashset until cannot go further anymore

           remove these elements from hashset

   2.2 if the sequence is longer all previous sequences ,update max length

另外從這一題學到的C++的東西:

hash_*系列例如hash_map,hash_set 等已經被desperate被廢棄了,C++11用unordered_map,unordered_set等來替代。

上代碼:

 

class Solution {
    public:
        int longestConsecutive(vector<int> &num) {

            unordered_set<int>::const_iterator it;
            unordered_set<int> set;
            for(int i=0;i<num.size();i++){
                if(set.find(num[i])==set.end()){
                    set.insert(num[i]);
                }
            }
            int maxc = 0;
            for(int i=0;i<num.size();i++){
                int c = 1 ;
                if(set.find(num[i])!=set.end()){
                    int cur = num[i]+1;
                    while(set.find(cur)!=set.end()){
                        set.erase(cur);
                        c++;
                        cur++;

                    }
                    cur = num[i]-1;
                    while(set.find(cur)!=set.end()){
                        set.erase(cur);
                        c++;
                        cur--;
                    }
                    if(maxc<c)
                        maxc=c;

                }
            }

            return maxc;
        }
};
相關文章
相關標籤/搜索