最长连续序列
给定一个未排序的整数数组 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
思路
一开始想的是直接把序列排序,然后遍历一遍元素把与前一个元素的差值不为1的下标记录下来,最后作差来比较长度,提交之后发现元素会重复,那就行不通了。
看了题解,写的很精炼,主要思路就是用set来记录元素之后,在遍历过程中确认每个元素是否作为序列的开头,统计序列们长度取最大值。
多少记一下吧。
代码
被重复元素干掉的菜鸡思路
//有重复元素 寄
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
sort(nums.begin(),nums.end());
vector<int> index;
if(nums.size()<2)
return nums.size();
for(int i=1;i<nums.size();i++){
if(nums[i]-nums[i-1]!=1 ){
index.emplace_back(i);
}else if(nums[i]-nums[i-1]==1 && i==nums.size()-1){
index.emplace_back(i+1);
}
}
vector<int> length;
length.emplace_back(index[0]);
for(int i=0;i<index.size()-1;i++){
length.emplace_back(index[i+1]-index[i]);
}
if(length.size()==1){
return length[0];
}
sort(length.begin(),length.end());
return length[length.size()-1];
}
};
官解:
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
unordered_set<int> num_set;
for (const int& num : nums) {
num_set.insert(num);
}
int longestStreak = 0;
for (const int& num : num_set) {
if (!num_set.count(num - 1)) {
int currentNum = num;
int currentStreak = 1;
while (num_set.count(currentNum + 1)) {
currentNum += 1;
currentStreak += 1;
}
longestStreak = max(longestStreak, currentStreak);
}
}
return longestStreak;
}
};
用find也行:
//find写法
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
unordered_set<int> num_set;
for (auto i : nums) {
num_set.insert(i);
}
int max_cnt=0;
for(auto &i:nums){
if (num_set.find(i-1)==num_set.end()) {
int cur=i;
int cnt=1;
while(num_set.find(cur+1)!=num_set.end()){
cur++;
cnt++;
}
max_cnt=max(max_cnt,cnt);
}
}
return max_cnt;
}
};