问题描述
Given an unsorted array of integers, find the length of the longest consecutive elements sequence.
For example,
Given [100, 4, 200, 1, 3, 2],
The longest consecutive elements sequence is [1, 2, 3, 4]. Return its length: 4.
Your algorithm should run in O(n) complexity.
解题思路
这道题要求求一个给定数组中的最长连续序列,并给定了O(n)复杂度限制,所以先将数组排序的方法不可行,我们可以这样想:先使用一个集合set存入所有的数字,然后遍历数组中的每个数字,如果其在集合中存在,那么将其移除,然后分别用两个变量pre和next算出其前一个数跟后一个数,如果pre在集合中,那么将pre移除集合,然后pre再自减1,直至pre不在集合之中,对next采用同样的方法,直至next和pre都不在集合中,则next-pre-1就是当前数字的最长连续序列。再用res存储这个序列的长度。以此方法遍历完数组中的所有元素后,所得到的最大的res即为最长序列长度。
代码展示
#include <iostream>
#include <stdlib.h>
#include <vector>
#include <set>
#include <string>
using namespace std;
class Solution {
public:
int longestConsecutive(vector<int>& nums){
int res = 0;
int length = nums.size();
set<int> s(nums.begin(), nums.end()); //定义一个集合s存储向量空间中所有的数
for(int i=0;i<length;i++) { //遍历数组
if (!s.count(nums[i])) continue; //对于val找出它在s中最长的连续段
s.erase(nums[i]); //找完后删除该连续段的数字,并记录其长度
int pre = nums[i] - 1, next = nums[i] + 1;
while (s.count(pre)) s.erase(pre--);
while (s.count(next)) s.erase(next++);
res = max(res, next - pre - 1); //用res储存最大的连续段长度
}
return res;
}
};
int main(){
int n;
vector<int> nums;
cout<<"请输入向量长度:";
cin>>n;
int num[n];
for(int i = 0;i<n;i++){
cin>>num[i];
nums.push_back(num[i]);
}
Solution solution;
int result=solution.longestConsecutive(nums);
cout<<result<<endl;
}
运行结果展示