题目:最长连续序列
给定一个未排序的整数数组,找出最长连续序列的长度。
要求算法的时间复杂度为 O(n)。
示例:
输入: [100, 4, 200, 1, 3, 2]
输出: 4
解释: 最长连续序列是 [1, 2, 3, 4]。它的长度为 4。
答案:
1.排序
将数组排序,再找连续序列的最大长度
class Solution {
public int longestConsecutive(int[] nums) {
if (nums.length == 0) {
return 0;
}
Arrays.sort(nums);
int longestStreak = 1;
int currentStreak = 1;
for (int i = 1; i < nums.length; i++) {
if (nums[i] != nums[i-1]) {
if (nums[i] == nums[i-1]+1) {
currentStreak += 1;
}
else {
longestStreak = Math.max(longestStreak, currentStreak);
currentStreak = 1;
}
}
}
return Math.max(longestStreak, currentStreak);
}
}
时间复杂度:O(nlgn) 排序算法
空间复杂度:O(1)
2.HashSet
用HashSet将数字保存。同时,我们只对(当前数字 - 1)不在哈希表里的数字,作为连续序列的第一个数字去找对应的最长序列。
避免了重复查找,比如[100, 4, 200, 1, 3, 2],从3开始查找则查找了3 4,从2开始查查找了2 3 4,从1开始查查找了1 2 3 4,所以其实只从1(1-1=0不在哈希表中)开始查就可以了
public int longestConsecutive(int[] nums) {
HashSet<Integer> set = new HashSet<>();
for (int i = 0; i < nums.length; i++) {
set.add(nums[i]);
}
int max = 0;
for (int num : set) {
if (!set.contains(num - 1)) { //当前数字-1不在哈希表里,说明当前序列是连续序列的最小即第一个数字
int currentNum = num;
int count = 1;
while (set.contains(currentNum + 1)) { //计算连续序列的长度
count++;
currentNum += 1;
}
max = Math.max(max, count);
}
}
return max;
}
时间复杂度:O(n)。尽管在 for 循环中嵌套了一个 while 循环,时间复杂度看起来像是二次方级别的。但其实它是线性的算法。因为只有当 currentNum 遇到了一个序列的开始, while 循环才会被执行
空间复杂度:O(n) 哈希表