给定一个未排序的整数数组 nums ,找出数字连续的最长序列(不要求序列元素在原数组中连续)的长度。
请你设计并实现时间复杂度为 O(n) 的算法解决此问题。
示例 1:
输入:nums = [100,4,200,1,3,2]
输出:4
解释:最长数字连续序列是 [1, 2, 3, 4]。它的长度为 4。
示例 2:
输入:nums = [0,3,7,2,5,8,4,6,0,1]
输出:9
核心思路:
如果当前数字 num 是一个序列的开始(即 num - 1 不在哈希表中),那么从 num 开始逐个检查后续数字 num + 1, num + 2, … 是否在哈希表中存在,直到不再连续。
#include <iostream>
#include <vector>
#include <unordered_set>
#include <algorithm>
using namespace std;
class Solution {
public:
int longestConsecutive(vector<int>& nums)
{
if (nums.empty())
return 0;
// 快速查找且唯一
unordered_set<int> numSet(nums.begin(), nums.end());
int longestStreak = 0;
for (const int& num : nums)
{
// 保证连续性
if (!numSet.count(num - 1))
{
int currNum = num;
int currentStreak = 1;
// 寻找最长序列
while (numSet.count(currNum+1))
{
currNum += 1;
currentStreak += 1;
}
// 更新结果
longestStreak = max(longestStreak, currentStreak);
}
}
return longestStreak;
}
};
int main()
{
Solution solution;
vector<int> nums1 = { 100, 4, 200, 1, 3, 2 };
vector<int> nums2 = { 0, 3, 7, 2, 5, 8, 4, 6, 0, 1 };
cout << "The longest consecutive sequence length in nums1: "
<< solution.longestConsecutive(nums1) << endl;
cout << "The longest consecutive sequence length in nums2: "
<< solution.longestConsecutive(nums2) << endl;
return 0;
}
题目来源:LeetCode