描述:
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.
分析:
如果允许 O(n log n) 的复杂度,那么可以先排序,可是本题要求 O(n)。由于序列里的元素是无序的,又要求 O(n),首先要想到用哈希表。
用一个哈希表visited[]记录每个元素是否使用,对每个元素,以该元素为中心,往左右扩张,直到不连续为止,记录下最长的长度。
在[100,4,200,1,3,2,101,99,97,98,102]的例子中:
(1)第一步:以100为中心
向右扩张:101,102
向左扩张:99,98,97
(2)第二步:以4为中心
向右扩张:无
向左扩张:3,2,1
(3)第三步:以200为中心
向右扩张:无
向左扩张:无
(4)1,3,2,101……..以被扩张,无需以他们为中心进行扩张
代码:
class Solution {
public:
int longestConsecutive(vector<int> &num) {
int i,j;
int CurLength = 0;
int MaxLength = 0;
int Len = num.size();
// Map中的元素是自动按key升序排序
unordered_map<int,bool> used;
//初始化
for(i = 0;i < Len;i++){
used[num[i]] = false;
}
for(i = 0;i < Len;i++){
//如果以扩张,跳过
if(used[num[i]]){
continue;
}
else{
//以num[i]为中心左右扩张,直到不连续为止
used[num[i]] = true;
CurLength = 1;
//向右扩张
for(j = num[i] + 1;used.find(j) != used.end();j++){
CurLength ++;
used[j] = true;
}
//向左扩张
for(j = num[i] - 1;used.find(j) != used.end();j--){
CurLength ++;
used[j] = true;
}
//更新最大长度
if(CurLength > MaxLength){
MaxLength = CurLength;
}
}//if
}//for
return MaxLength;
}
};