LeetCode - 128. Longest Consecutive Sequence (哈希表)
题目链接
题目
解析
第一种方法:
-
使用一个
HashSet来存储对应的值，一开始先将所有的值都加入set； -
遍历数组的每一个元素，每次去检查当前元素
num的前一个元素num - 1是不是在set中，如果是，说明num不是最长长度的起点，跳过； -
如果不是，则在
set集合中不断的每次+1，(也就是不断检查num + 1、num + 2、num + 3、num + 4.....在不在set中)，并记录查到的元素个数(也就是长度)，然后更新结果(记录最大长度)res即可； -
时间复杂度: 虽然有两重循环，但是每个元素最多访问两次，所以时间复杂度为
O(N)；
看两个例子:
import java.io.*;
import java.util.*;
class Solution {
public int longestConsecutive(int[] nums) {
if(nums == null || nums.length == 0)
return 0;
HashSet<Integer>set = new HashSet<>();
for(int num : nums)
set.add(num);
int res = 0;
for(int num : nums){
if(!set.contains(num - 1)){
int len = 1, temp = num + 1;
while(set.contains(temp++))
len++;
res = Math.max(res, len);
}
}
return res;
}
public static void main(String[] args){
PrintStream out = System.out;
int[] nums = {1, 2, 3, 6, 4, 5, 7};
out.println(new Solution().
longestConsecutive(nums)
);
}
}
第二种解法:
-
用一个
HashMap记录<key , value > =<数组的值，这个值如果作为边界(左/右)时的最大长度>。整个过程和动态规划有点类似； -
遍历数组的每一个元素，先得到
num - 1和num + 1在map中对应的value，如果两个value都为空，说明此时num两边都没有相邻的元素，所以put(num, 1)，表示num作为左/右边界的最大长度为1； -
如果
map.get(num - 1) == null && map.get(num + 1) != null，说明此时num可以作为右边界，而此时不但要更新num的value，也要更新nums[num - map.get(num - 1)]的value，这个value就是map.get(num - 1) + 1，所以说这个过程有点类似动态规划； -
同理，如果
map.get(num - 1) != null && map.get(num + 1) == null，说明此时num可以作为左边界，而此时不但要更新num的value，也要更新nums[num + map.get(num + 1)]的value，这个value就是map.get(num + 1) + 1。 -
如果
map.get(num - 1) != null && map.get(num + 1) != null，则此时同时可以作为左右边界，说明它是连接两边的桥梁，所以要同时更新num、nums[num - map.get(num - 1)]、nums[num + map.get(num + 1)]的值为map.get(num - 1) + map.get(num + 1) + 1；
看一个例子:
import java.io.*;
import java.util.*;
class Solution {
public int longestConsecutive(int[] nums) {
if(nums == null || nums.length == 0)
return 0;
HashMap<Integer, Integer> map = new HashMap<>();
for(int num : nums){
if(map.containsKey(num)) // reduplicative numbers
continue;
Integer L = map.get(num - 1);
Integer R = map.get(num + 1);
if(L == null && R == null)
map.put(num, 1);
else if(L == null && R != null){
map.put(num, R + 1);
map.put(num + R, R + 1);
}else if(L != null && R == null){
map.put(num, L + 1);
map.put(num - L, L + 1);
}else {
map.put(num, L + R + 1);
map.put(num - L, L + R + 1);
map.put(num + R, L + R + 1);
}
}
int res = 0;
for(Map.Entry<Integer, Integer>entry : map.entrySet())
res = Math.max(res, entry.getValue());
return res;
}
public static void main(String[] args){
PrintStream out = System.out;
int[] nums = {1, 2, 3, 6, 4, 5, 7};
out.println(new Solution().
longestConsecutive(nums)
);
}
}
C++代码:
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
unordered_set<int> set(nums.begin(), nums.end());
int res = 0;
for(int num : nums){
if(!set.count(num - 1)){
int len = 1, temp = num + 1;
while(set.count(temp++))
len++;
res = max(res, len);
}
}
return res;
}
};
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
unordered_map<int, int>mp;
for(int num : nums){
if(mp.count(num))
continue;
auto it_l = mp.find(num - 1);
auto it_r = mp.find(num + 1);
int l = it_l != mp.end() ? it_l->second : 0;
int r = it_r != mp.end() ? it_r->second : 0;
if(l > 0 && r > 0)
mp[num] = mp[num - l] = mp[num + r] = l + r + 1;
else if(l > 0)
mp[num] = mp[num - l] = l + 1;
else if(r > 0)
mp[num] = mp[num + r] = r + 1;
else
mp[num] = 1;
}
int res = 0;
for(const auto & kv : mp)
res = max(res, kv.second);
return res;
}
};