给定一个未排序的整数数组 nums ,找出数字连续的最长序列(不要求序列元素在原数组中连续)的长度。
请你设计并实现时间复杂度为 O(n) 的算法解决此问题。
这道题竟然可以用并查集,感觉很神奇哈。
就是把连续的数字视作在一个联通分量中,并查集中元素的父节点是连续数字中的最大数字。
然后遍历数组,找到长度最长的序列长度。
class UnionFind{
public:
unordered_map<int, int> mmap;
void merge(int s1, int s2) {
int r1 = find(s1);
int r2 = find(s2);
if (r1 > r2) {
mmap[r2] = r1;
} else if (r1 < r2) {
mmap[r1] = r2;
}
}
int find(int s) {
int root = s;
while(mmap[root]!=root) {
root = mmap[root];
}
while(mmap[s]!=s) {
int tmp = mmap[s];
mmap[tmp] = root;
s = mmap[s];
}
return root;
}
void add(int s) {
if (mmap.count(s) == 0) {
mmap[s]=s;
}
}
};
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
UnionFind uf;
int res = 0;
for (const auto& num : nums) {
uf.add(num);
}
for (const auto& num : nums) {
if (uf.mmap.count(num+1) != 0) {
uf.merge(num, num+1);
}
}
for (const auto& num : nums) {
int root = uf.find(num);
res = max(res, root - num+1);
}
return res;
}
};