题目
给定一个未排序的整数数组,找出最长连续序列的长度。
要求算法的时间复杂度为 O(n)。
示例:
输入: [100, 4, 200, 1, 3, 2]
输出: 4
解释: 最长连续序列是 [1, 2, 3, 4]。它的长度为 4。
思路
要求复杂度为 O(n),所以采用Map,以空间代价代替时间代价。Map负责存储<nums[i],目前该值所在的最长连续序列长度>。遍历数组,如果是重复的数则不做任何操作。否则,取(nums[i]+1)目前所在的最长连续序列长度,取(nums[i]-1)目前所在的最长连续序列长度,将两者求和再加1得出nums[i]目前该值所在的最长连续序列长度。之后更新map,需要注意的是,只有连续序列的两端才会对之后的数值产生影响,所以只需更新连续序列两端的map值即可。
代码
class Solution {
public int longestConsecutive(int[] nums) {
if(nums.length<=1){
return nums.length;
}
int res=0;
//要求复杂度为 O(n),所以采用Map,以空间代价代替时间代价。
//map存储<nums[i],目前该值所在的最长连续序列长度>
Map<Integer,Integer> map=new HashMap<Integer,Integer>();
for(int i=0;i<nums.length;i++){
if(map.get(nums[i])!=null){
continue;
}
//取(nums[i]+1)的value
int after=0;
if(map.get((nums[i]+1))==null){
}else{
after=map.get((nums[i]+1));
}
//取(nums[i]-1)的value
int before=0;
if(map.get((nums[i]-1))==null){
}else{
before=map.get((nums[i]-1));
}
//总和
int sum=before+1+after;
if(sum>res){
res=sum;
}
map.put(nums[i],sum);
//只有连续序列的两端才会对之后的数值产生影响,所以只需更新连续序列两端的map值即可。
map.put(nums[i]+after,sum);
map.put(nums[i]-before,sum);
}
return res;
}
}
注意
采用Map,以空间代价代替时间代价。